test/api/testlib.c

   1 /* $Id: testlib.c,v 1.5 2004-11-19 10:27:15 heikki Exp $
   2    Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
   3    Index Data Aps
   4
   5 This file is part of the Zebra server.
   6
   7 Zebra is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 2, or (at your option) any later
  10 version.
  11
  12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with Zebra; see the file LICENSE.zebra.  If not, write to the
  19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  20 02111-1307, USA.
  21 */
  22
  23 /** testlib - utilities for the api tests */
  24
  25 #include <assert.h>
  26 #include <yaz/pquery.h>
  27 #include <idzebra/api.h>
  28 #include "testlib.h"
  29
  30 /** start_log: open a log file */
  31 /*    FIXME - parse command line arguments to set log levels etc */
  32 int log_level=0; /* not static, t*.c may use it */
  33
  34 void start_log(int argc, char **argv)
  35 {
  36     char logname[2048];
  37     if (!argv)
  38         return;
  39     if (!argv[0])
  40         return;
  41     sprintf(logname, "%s.log", argv[0]);
  42     yaz_log_init_file(logname);
  43     log_level = yaz_log_mask_str_x(argv[0],0);
  44     yaz_log_init_level(YLOG_DEFAULT_LEVEL | log_level);
  45     yaz_log(log_level,"starting %s",argv[0]);
  46 }
  47
  48 /**
  49  * start_up : do common start things, and a zebra_start
  50  *    - nmem_init
  51  *    - build the name of logfile from argv[0], and open it
  52  *      if no argv passed, do not open a log
  53  *    - read zebra.cfg from env var srcdir if it exists; otherwise current dir
  54  *      default to zebra.cfg, if no name is given
  55  */
  56 ZebraService start_up(char *cfgname, int argc, char **argv)
  57 {
  58     nmem_init();
  59     start_log(argc, argv);
  60     return start_service(cfgname);
  61 }
  62
  63 /** start_service - do a zebra_start with a decent config name */
  64 ZebraService start_service(char *cfgname)
  65 {
  66     char cfg[256];
  67     char *srcdir = getenv("srcdir");
  68     ZebraService zs;
  69     if (!srcdir || ! *srcdir)
  70         srcdir=".";
  71     if (!cfgname || ! *cfgname )
  72         cfgname="zebra.cfg";
  73
  74     sprintf(cfg, "%.200s/%s",srcdir, cfgname);
  75     zs=zebra_start(cfg);
  76     if (!zs)
  77     {
  78         printf("zebra_start failed, probably because missing config file \n"
  79                "check %s\n", cfg);
  80         exit(9);
  81     }
  82     return zs;
  83 }
  84
  85
  86 /** close_down closes down the zebra, logfile, nmem, xmalloc etc. logs an OK */
  87 int close_down(ZebraHandle zh, ZebraService zs, int retcode)
  88 {
  89     if (zh)
  90         zebra_close(zh);
  91     if (zs)
  92         zebra_stop(zs);
  93
  94     if (retcode)
  95         yaz_log(log_level,"========= Exiting with return code %d", retcode);
  96     else
  97         yaz_log(log_level,"========= All tests OK");
  98     nmem_exit();
  99     xmalloc_trav("x");
 100     return retcode;
 101 }
 102
 103 /** inits the database and inserts test data */
 104
 105 void init_data( ZebraHandle zh, const char **recs)
 106 {
 107     int i;
 108     char *addinfo;
 109     assert(zh);
 110     zebra_select_database(zh, "Default");
 111     yaz_log(log_level,"going to call init");
 112     i=zebra_init(zh);
 113     yaz_log(log_level,"init returned %d",i);
 114     if (i)
 115     {
 116         printf("init failed with %d\n",i);
 117         zebra_result(zh, &i, &addinfo);
 118         printf("  Error %d   %s\n",i,addinfo);
 119         exit(1);
 120     }
 121     if (recs)
 122     {
 123         zebra_begin_trans (zh, 1);
 124         for (i = 0; recs[i]; i++)
 125             zebra_add_record (zh, recs[i], strlen(recs[i]));
 126         zebra_end_trans (zh);
 127         zebra_commit (zh);
 128     }
 129
 130 }
 131
 132
 133
 134 int do_query(int lineno, ZebraHandle zh, char *query, int exphits)
 135 {
 136     ODR odr;
 137     YAZ_PQF_Parser parser;
 138     Z_RPNQuery *rpn;
 139     const char *setname="rsetname";
 140     int hits;
 141     int rc;
 142
 143
 144     yaz_log(log_level,"======================================");
 145     yaz_log(log_level,"qry[%d]: %s", lineno, query);
 146     odr=odr_createmem (ODR_DECODE);
 147
 148     parser = yaz_pqf_create();
 149     rpn = yaz_pqf_parse(parser, odr, query);
 150     if (!rpn) {
 151         printf("Error: Parse failed \n%s\n",query);
 152         exit(1);
 153     }
 154     rc=zebra_search_RPN (zh, odr, rpn, setname, &hits);
 155     if (rc) {
 156         printf("Error: search returned %d \n%s\n",rc,query);
 157         exit (1);
 158     }
 159
 160     if (hits != exphits) {
 161         printf("Error: search returned %d hits instead of %d\n",
 162                 hits, exphits);
 163         exit (1);
 164     }
 165     yaz_pqf_destroy(parser);
 166     odr_destroy (odr);
 167     return hits;
 168 }
 169
 170
 171 /**
 172  * makes a query, checks number of hits, and for the first hit, that
 173  * it contains the given string, and that it gets the right score
 174  */
 175 void ranking_query(int lineno, ZebraHandle zh, char *query,
 176           int exphits, char *firstrec, int firstscore )
 177 {
 178     ZebraRetrievalRecord retrievalRecord[10];
 179     ODR odr_output = odr_createmem (ODR_ENCODE);
 180     const char *setname="rsetname";
 181     int hits;
 182     int rc;
 183     int i;
 184
 185     hits=do_query(lineno, zh, query, exphits);
 186
 187     for (i = 0; i<10; i++)
 188         retrievalRecord[i].position = i+1;
 189
 190     rc=zebra_records_retrieve (zh, odr_output, setname, 0,
 191                      VAL_TEXT_XML, hits, retrievalRecord);
 192
 193     if (rc) {
 194         printf("Error: retrieve returned %d \n%s\n",rc,query);
 195         exit (1);
 196     }
 197
 198     if (!strstr(retrievalRecord[0].buf, firstrec))
 199     {
 200         printf("Error: Got the wrong record first\n");
 201         printf("Expected '%s' but got \n",firstrec);
 202         printf("%.*s\n",retrievalRecord[0].len,retrievalRecord[0].buf);
 203         exit(1);
 204     }
 205
 206     if (retrievalRecord[0].score != firstscore)
 207     {
 208         printf("Error: first rec got score %d instead of %d\n",
 209                 retrievalRecord[0].score, firstscore);
 210         exit(1);
 211     }
 212     odr_destroy (odr_output);
 213 }
 214