1 /* This file is part of the YAZ toolkit.
2 * Copyright (C) Index Data
3 * See the file LICENSE for details.
14 #include <yaz/rpn2solr.h>
15 #include <yaz/wrbuf.h>
16 #include <yaz/pquery.h>
18 static int compare(solr_transform_t ct, const char *pqf, const char *solr)
21 ODR odr = odr_createmem(ODR_ENCODE);
22 WRBUF w = wrbuf_alloc();
23 Z_RPNQuery *q = p_query_rpn(odr, pqf);
27 int r = solr_transform_rpn2solr_wrbuf(ct, w, q);
32 yaz_log(YLOG_LOG, "%s -> Error %d", pqf, r);
33 if (!solr) /* also expected error? */
38 yaz_log(YLOG_LOG, "%s -> %s", pqf, wrbuf_cstr(w));
39 if (solr && !strcmp(wrbuf_cstr(w), solr))
43 yaz_log(YLOG_WARN, " expected: %s", solr ? solr : "null");
44 yaz_log(YLOG_WARN, " got: %s", wrbuf_cstr(w));
53 static void tst1(void)
55 solr_transform_t ct = solr_transform_create();
57 YAZ_CHECK(compare(ct, "@or a @and b c", "a OR (b AND c)"));
58 YAZ_CHECK(compare(ct, "abc", "abc"));
59 YAZ_CHECK(compare(ct, "\"a b c\"", "\"a b c\""));
60 YAZ_CHECK(compare(ct, "@not a b", "a AND NOT b"));
61 YAZ_CHECK(compare(ct, "@and @or a b c", "(a OR b) AND c"));
62 YAZ_CHECK(compare(ct, "@and a b", "a AND b"));
63 YAZ_CHECK(compare(ct, "@or a b", "a OR b"));
64 YAZ_CHECK(compare(ct, "@or a @and b c", "a OR (b AND c)"));
65 YAZ_CHECK(compare(ct, "@or @and a b @and c d", "(a AND b) OR (c AND d)"));
66 YAZ_CHECK(compare(ct, "@or @or a b @or c d", "(a OR b) OR (c OR d)"));
67 YAZ_CHECK(compare(ct, "@or @or @or a b @or c d @or e f", "((a OR b) OR (c OR d)) OR (e OR f)"));
68 YAZ_CHECK(compare(ct, "@and @and a b @and c d", "(a AND b) AND (c AND d)"));
69 YAZ_CHECK(compare(ct, "@attr 1=field abc", "field:abc"));
70 YAZ_CHECK(compare(ct, "@attr 1=field \"a b c\"", "field:\"a b c\""));
71 YAZ_CHECK(compare(ct, "@attr 1=4 abc", 0)); /* should fail */
73 solr_transform_define_pattern(ct, "index.title", "1=4");
74 YAZ_CHECK(compare(ct, "@attr 1=4 abc", "title:abc"));
76 solr_transform_define_pattern(ct, "index.foo", "1=bar");
77 YAZ_CHECK(compare(ct, "@attr 1=bar abc", "foo:abc"));
80 YAZ_CHECK(compare(ct, "@attr 5=1 water", "water*"));
81 YAZ_CHECK(compare(ct, "@attr 5=2 water", "*water"));
82 YAZ_CHECK(compare(ct, "@attr 5=3 water", "*water*"));
83 YAZ_CHECK(compare(ct, "@attr 5=100 water", "water"));
84 YAZ_CHECK(compare(ct, "@attr 5=101 water", 0));
85 YAZ_CHECK(compare(ct, "@attr 5=104 w#ter", "w?ter"));
86 YAZ_CHECK(compare(ct, "@attr 5=104 w#t?r", "w?t*r"));
87 YAZ_CHECK(compare(ct, "@attr 5=104 w#te?", "w?te*"));
88 YAZ_CHECK(compare(ct, "@attr 5=104 w\\#te?", "w?te*")); /* PQF eats # */
89 YAZ_CHECK(compare(ct, "@attr 5=104 w\\\\#te?", "w#te*"));
91 /* reserved characters */
92 YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\\\\\",
94 YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\\\\\\\\"",
95 "\\\"" "\\\\" "\\\""));
96 YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\", "\\\"\\\\"));
98 YAZ_CHECK(compare(ct, "@attr 5=104 \\{:\\}", "\\{\\:\\}"));
100 YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\\\\\\\\"",
101 "\\\"" "\\\\" "\\\""));
103 YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=1 1980", "date:[* TO 1980}"));
104 YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=2 1980", "date:[* TO 1980]"));
105 YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=3 1980", "date:1980"));
106 YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=4 1980", "date:[1980 TO *]"));
107 YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=5 1980", "date:{1980 TO *]"));
108 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=2 234 @attr 1=date @attr 2=4 1990", "date:[1990 TO 234]"));
109 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=2 345 @attr 1=date @attr 2=4 234", "date:[234 TO 345]"));
110 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=5 234 @attr 1=titl @attr 2=2 1990", "date:{234 TO *] AND titl:[* TO 1990]"));
111 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=5 234 @attr 1=date @attr 2=2 1990", "date:{234 TO 1990]"));
112 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=4 234 @attr 1=date @attr 2=2 1990", "date:[234 TO 1990]"));
113 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=5 234 @attr 1=date @attr 2=1 1990", "date:{234 TO 1990}"));
114 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=4 234 @attr 1=date @attr 2=2 1990", "date:[234 TO 1990]"));
115 YAZ_CHECK(compare(ct, "@or @attr 1=date @attr 2=4 234 @attr 1=date @attr 2=2 1990", "date:[234 TO *] OR date:[* TO 1990]"));
116 YAZ_CHECK(compare(ct, "@or @attr 1=date @attr 2=2 234 @attr 1=date @attr 2=4 1990", "date:[* TO 234] OR date:[1990 TO *]"));
117 YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=1 1980", "date:[* TO 1980}"));
118 YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=2 1980", "date:[* TO 1980]"));
119 YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=2 234 @attr 1=date @attr 2=4 1990", "date:[1990 TO 234]"));
121 solr_transform_close(ct);
124 static void tst2(void)
126 WRBUF w = wrbuf_alloc();
127 solr_transform_t ct = 0;
128 const char *srcdir = getenv("srcdir");
131 wrbuf_puts(w, srcdir);
134 wrbuf_puts(w, "../etc/pqf.properties");
136 ct = solr_transform_open_fname(wrbuf_cstr(w));
137 YAZ_CHECK(compare(ct, "@attr 1=4 abc", "dc.title:abc"));
139 YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=108 abc", "dc.title:abc"));
142 YAZ_CHECK(compare(ct, "@attr 1=4 @attr 3=1 @attr 6=1 abc", "dc.title:abc"));
143 YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=1 @attr 6=1 abc", "dc.title:abc"));
145 YAZ_CHECK(compare(ct, "@attr 1=1016 abc", "abc"));
147 YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=1 1980", "dc.date:[* TO 1980}"));
148 YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=2 1980", "dc.date:[* TO 1980]"));
149 YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=3 1980", "dc.date:1980"));
150 YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=4 1980", "dc.date:[1980 TO *]"));
151 YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=5 1980", "dc.date:{1980 TO *]"));
152 YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=2 234 @attr 1=30 @attr 2=4 1990", "dc.date:[1990 TO 234]"));
153 YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=2 345 @attr 1=30 @attr 2=4 234", "dc.date:[234 TO 345]"));
154 YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=5 234 @attr 1=4 @attr 2=2 1990", "dc.date:{234 TO *] AND dc.title:[* TO 1990]"));
155 YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=5 234 @attr 1=30 @attr 2=2 1990", "dc.date:{234 TO 1990]"));
156 YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=4 234 @attr 1=30 @attr 2=2 1990", "dc.date:[234 TO 1990]"));
157 YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=5 234 @attr 1=30 @attr 2=1 1990", "dc.date:{234 TO 1990}"));
158 YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=4 234 @attr 1=30 @attr 2=2 1990", "dc.date:[234 TO 1990]"));
159 YAZ_CHECK(compare(ct, "@or @attr 1=30 @attr 2=4 234 @attr 1=30 @attr 2=2 1990", "dc.date:[234 TO *] OR dc.date:[* TO 1990]"));
160 YAZ_CHECK(compare(ct, "@or @attr 1=30 @attr 2=2 234 @attr 1=30 @attr 2=4 1990", "dc.date:[* TO 234] OR dc.date:[1990 TO *]"));
163 YAZ_CHECK(compare(ct, "@attr 2=103 @attr 1=_ALLRECORDS 1", "solr.allRecords=1"));
165 YAZ_CHECK(compare(ct, "@attr 1=500 abc", 0));
166 solr_transform_close(ct);
170 int main (int argc, char **argv)
172 YAZ_CHECK_INIT(argc, argv);
182 * c-file-style: "Stroustrup"
183 * indent-tabs-mode: nil
185 * vim: shiftwidth=4 tabstop=8 expandtab