4055c5afe8d4a360e1091ed12d79bd1634a8b906
[idzebra-moved-to-github.git] / test / api / t4.c
1 /* $Id: t4.c,v 1.19 2006-03-31 15:58:05 adam Exp $
2    Copyright (C) 1995-2005
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 /* t4 - insert a small pile of records, search and fetch them */
24
25 #include "testlib.h"
26
27 const char *myrec[] = {
28         "<gils>\n"
29         "  <title>My title</title>\n"
30         "</gils>\n",
31         0};
32         
33 #define NUMBER_TO_FETCH_MAX 1000
34
35 static void tst(int argc, char **argv)
36 {
37     int i;
38     int number_to_be_inserted = 5;
39     int number_to_fetch = 5;
40
41     ZebraService zs = tl_start_up(0, argc, argv);
42     ZebraHandle zh = zebra_open(zs, 0);
43
44     YAZ_CHECK(tl_init_data(zh, myrec));
45
46     YAZ_CHECK(zebra_begin_trans (zh, 1) == ZEBRA_OK);
47
48     for (i = 0; i< number_to_be_inserted-1; i++)
49     {  /* -1 since already inserted one in init_data */
50         zebra_add_record(zh, myrec[0], strlen(myrec[0]));
51     }
52     YAZ_CHECK(zebra_end_trans(zh) == ZEBRA_OK);
53
54     zebra_close(zh);
55     zebra_stop(zs);
56
57     zs = tl_zebra_start("");
58     zh = zebra_open(zs, 0);
59     YAZ_CHECK(zebra_select_database(zh, "Default") == ZEBRA_OK);
60     zebra_set_resource(zh, "sortmax", "3"); /* make small sort boundary */
61
62     for (i = 0; i<2; i++)
63     {
64         ZEBRA_RES ret;
65         ZebraRetrievalRecord retrievalRecord[NUMBER_TO_FETCH_MAX];
66         char setname[20];
67         int j;
68         ODR odr_input = odr_createmem(ODR_DECODE);    
69         ODR odr_output = odr_createmem(ODR_DECODE);    
70         YAZ_PQF_Parser parser = yaz_pqf_create();
71         Z_RPNQuery *query = yaz_pqf_parse(parser, odr_input, "@attr 1=4 my");
72         zint hits;
73         
74         sprintf(setname, "s%d", i+1);
75         ret = zebra_search_RPN(zh, odr_input, query, setname, &hits);
76         if (ret != ZEBRA_OK)
77         {
78             int code = zebra_errCode(zh);
79             yaz_log(YLOG_WARN, "Unexpected error code=%d", code);
80             exit(1);
81         }
82         if (hits != number_to_be_inserted)
83         {
84             yaz_log(YLOG_WARN, "Unexpected hit count " ZINT_FORMAT 
85                     "(should be %d)", hits, number_to_be_inserted);
86             exit(1);
87         }
88
89         yaz_pqf_destroy(parser);
90
91         odr_destroy(odr_input);
92
93         YAZ_CHECK(zebra_begin_trans(zh, 1) == ZEBRA_OK);
94
95         for (j = 0; j < number_to_fetch; j++)
96             retrievalRecord[j].position = j+1;
97
98         ret = zebra_records_retrieve(zh, odr_output, setname, 0,
99                                      VAL_TEXT_XML, number_to_fetch,
100                                      retrievalRecord);
101         if (ret != ZEBRA_OK)
102         {
103             int code = zebra_errCode(zh);
104             yaz_log(YLOG_FATAL, "zebra_records_retrieve returned error %d",
105                     code);
106             exit(1);
107         }
108         
109         for (j = 0; j < number_to_fetch; j++)
110         {
111             if (!retrievalRecord[j].buf)
112             {
113                 yaz_log(YLOG_FATAL, "No record buf at position %d", j);
114                 exit(1);
115             }
116             if (!retrievalRecord[j].len)
117             {
118                 yaz_log(YLOG_FATAL, "No record len at position %d", j);
119                 exit(1);
120             }
121         }
122         odr_destroy(odr_output);
123
124         YAZ_CHECK(zebra_end_trans(zh) == ZEBRA_OK);
125     }
126     zebra_commit(zh);
127     YAZ_CHECK(tl_close_down(zh, zs));
128 }
129
130 TL_MAIN
131