Test case for YAZ-822
[yaz-moved-to-github.git] / test / test_rpn2cql.c
1 /* This file is part of the YAZ toolkit.
2  * Copyright (C) Index Data
3  * See the file LICENSE for details.
4  */
5 #if HAVE_CONFIG_H
6 #include <config.h>
7 #endif
8
9 #include <stdlib.h>
10 #include <stdio.h>
11
12 #include <yaz/test.h>
13 #include <yaz/log.h>
14 #include <yaz/rpn2cql.h>
15 #include <yaz/wrbuf.h>
16 #include <yaz/pquery.h>
17
18 static int compare2(cql_transform_t ct, const char *pqf, const char *cql,
19                     int expected_error)
20 {
21     int ret = 0;
22     ODR odr = odr_createmem(ODR_ENCODE);
23     WRBUF w = wrbuf_alloc();
24     Z_RPNQuery *q = p_query_rpn(odr, pqf);
25
26     if (q)
27     {
28         int r = cql_transform_rpn2cql_wrbuf(ct, w, q);
29
30         if (r != 0)
31         {
32             const char *addinfo = 0;
33             int err = cql_transform_error(ct, &addinfo);
34             /* transform error */
35             yaz_log(YLOG_LOG, "%s -> Error %d", pqf, r);
36             if (err == 0)
37                 ;
38             else if (err == expected_error)
39             {
40                 if (addinfo && cql && !strcmp(addinfo, cql))
41                     ret = 1;
42                 else if (!addinfo && !cql)
43                     ret = 1;
44             }
45         }
46         else if (r == 0)
47         {
48             yaz_log(YLOG_LOG, "%s -> %s", pqf, wrbuf_cstr(w));
49             if (!expected_error)
50                 ret = 1;
51             else if (cql && !strcmp(wrbuf_cstr(w), cql))
52             {
53                 ret = 1;
54             }
55             else
56             {
57                 yaz_log(YLOG_WARN, " expected: %s", cql ? cql : "null");
58                 yaz_log(YLOG_WARN, " got:      %s", wrbuf_cstr(w));
59             }
60         }
61     }
62     wrbuf_destroy(w);
63     odr_destroy(odr);
64     return ret;
65 }
66
67 static int compare(cql_transform_t ct, const char *pqf, const char *cql)
68 {
69     return compare2(ct, pqf, cql, 0);
70 }
71
72 static void tst1(void)
73 {
74     cql_transform_t ct = cql_transform_create();
75
76     YAZ_CHECK(compare(ct, "abc", "abc"));
77     YAZ_CHECK(compare(ct, "\"a b c\"", "\"a b c\""));
78     YAZ_CHECK(compare(ct, "@and a b", "a and b"));
79     YAZ_CHECK(compare(ct, "@or a @and b c", "a or (b and c)"));
80     YAZ_CHECK(compare(ct, "@or @and a b @and c d", "(a and b) or (c and d)"));
81     YAZ_CHECK(compare(ct, "@or @or a b @or c d", "(a or b) or (c or d)"));
82     YAZ_CHECK(compare(ct, "@and @and a b @and c d", "(a and b) and (c and d)"));
83
84     YAZ_CHECK(compare(ct, "@attr 1=field abc", "field=abc"));
85     YAZ_CHECK(compare2(ct, "@attr 1=4 abc", 0, 114)); /* should fail */
86
87     cql_transform_define_pattern(ct, "index.title", "1=4");
88     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "title=abc"));
89
90     cql_transform_define_pattern(ct, "index.foo", "1=bar");
91     YAZ_CHECK(compare(ct, "@attr 1=bar abc", "foo=abc"));
92
93     cql_transform_close(ct);
94 }
95
96 static void tst2(void)
97 {
98     WRBUF w = wrbuf_alloc();
99     cql_transform_t ct = 0;
100     const char *srcdir = getenv("srcdir");
101     if (srcdir)
102     {
103         wrbuf_puts(w, srcdir);
104         wrbuf_puts(w, "/");
105     }
106     wrbuf_puts(w, "../etc/pqf.properties");
107
108     ct = cql_transform_open_fname(wrbuf_cstr(w));
109     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "dc.title=abc"));
110     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=108 abc", "dc.title=/exact abc"));
111     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 3=1 @attr 6=1 abc", "dc.title=abc"));
112     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=1 @attr 6=1 abc", "dc.title=abc"));
113     YAZ_CHECK(compare(ct, "@attr 1=1016 abc", "abc"));
114     /* Date tests */
115     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=1 1980", "dc.date<1980"));
116     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=2 1980", "dc.date<=1980"));
117     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=3 1980", "dc.date=1980"));
118     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=4 1980", "dc.date>=1980"));
119     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=5 1980", "dc.date>1980"));
120     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=4 234 @attr 1=30 @attr 2=2 1990", "dc.date>=234 and dc.date<=1990"));
121
122     /* Truncation */
123     YAZ_CHECK(compare(ct, "@attr 5=1 water", "water*"));
124     YAZ_CHECK(compare(ct, "@attr 5=2 water", "*water"));
125     YAZ_CHECK(compare(ct, "@attr 5=3 water", "*water*"));
126     YAZ_CHECK(compare(ct, "@attr 5=100 water", "water"));
127     YAZ_CHECK(compare(ct, "@attr 5=102 water", "water"));
128     YAZ_CHECK(compare(ct, "@attr 5=104 water", "water"));
129
130     YAZ_CHECK(compare(ct, "@attr 5=102 wat.*er", "wat*er"));
131     YAZ_CHECK(compare(ct, "@attr 5=104 wat?er", "wat*er"));
132
133     YAZ_CHECK(compare(ct, "@attr 5=102 wat.er", "wat?er"));
134     YAZ_CHECK(compare(ct, "@attr 5=104 wat#er", "wat?er"));
135     YAZ_CHECK(compare(ct, "@attr 5=102 wat?er", "wat\\?er"));
136     YAZ_CHECK(compare(ct, "@attr 5=104 wat*er", "wat\\*er"));
137     YAZ_CHECK(compare(ct, "@attr 5=102 wat#er", "wat#er"));
138
139     /* \. is 'eaten' by PQF parser */
140     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\.er", "wat?er"));
141
142     /* Escape sequences */
143     /* note: escape sequences that survive after PQF parse below */
144     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\?er", "wat\\?er"));
145     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\?er", "wat\\?er"));
146
147     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\*er", "wat\\*er"));
148     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\*er", "wat\\*er"));
149
150     YAZ_CHECK(compare(ct, "wat\\\\#er", "wat#er"));
151     YAZ_CHECK(compare(ct, "@attr 5=100 wat\\\\#er", "wat#er"));
152     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\#er", "wat#er"));
153     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\#er", "wat#er"));
154     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\.er", "wat.er"));
155     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\.er", "wat.er"));
156
157     /* Quoting */
158     YAZ_CHECK(compare(ct, "@attr 5=100 \"\"", "\"\""));
159     YAZ_CHECK(compare(ct, "@attr 5=1 \"\"", "\"*\""));
160     YAZ_CHECK(compare(ct, "@attr 5=2 \"\"", "\"*\""));
161     YAZ_CHECK(compare(ct, "@attr 5=3 \"\"", "\"**\""));
162     YAZ_CHECK(compare(ct, "@attr 5=102 \"\"", "\"\""));
163     YAZ_CHECK(compare(ct, "@attr 5=104 \"\"", "\"\""));
164
165     YAZ_CHECK(compare(ct, "@attr 5=1 \"water basket\"", "\"water basket*\""));
166     YAZ_CHECK(compare(ct, "@attr 5=2 \"water basket\"", "\"*water basket\""));
167     YAZ_CHECK(compare(ct, "@attr 5=3 \"water basket\"", "\"*water basket*\""));
168
169     /* Other */
170     YAZ_CHECK(compare(ct, "@attr 2=103 @attr 1=_ALLRECORDS 1", "cql.allRecords=1"));
171     YAZ_CHECK(compare2(ct, "@attr 1=500 abc", 0, 114));
172     YAZ_CHECK(compare2(ct, "@attr 5=99 x", "99", 120));
173     cql_transform_close(ct);
174     wrbuf_destroy(w);
175 }
176
177 int main (int argc, char **argv)
178 {
179     YAZ_CHECK_INIT(argc, argv);
180     YAZ_CHECK_LOG();
181     tst1();
182     tst2();
183     YAZ_CHECK_TERM;
184 }
185
186 /*
187  * Local variables:
188  * c-basic-offset: 4
189  * c-file-style: "Stroustrup"
190  * indent-tabs-mode: nil
191  * End:
192  * vim: shiftwidth=4 tabstop=8 expandtab
193  */
194