test/test_rpn2solr.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/rpn2solr.h>
  15 #include <yaz/wrbuf.h>
  16 #include <yaz/pquery.h>
  17
  18 static int compare(solr_transform_t ct, const char *pqf, const char *solr)
  19 {
  20     int ret = 0;
  21     ODR odr = odr_createmem(ODR_ENCODE);
  22     WRBUF w = wrbuf_alloc();
  23     Z_RPNQuery *q = p_query_rpn(odr, pqf);
  24
  25     if (q)
  26     {
  27         int r = solr_transform_rpn2solr_wrbuf(ct, w, q);
  28
  29         if (r != 0)
  30         {
  31             /* transform error */
  32             yaz_log(YLOG_LOG, "%s -> Error %d", pqf, r);
  33             if (!solr) /* also expected error? */
  34                 ret = 1;
  35         }
  36         else if (r == 0)
  37         {
  38             yaz_log(YLOG_LOG, "%s -> %s", pqf, wrbuf_cstr(w));
  39             if (solr && !strcmp(wrbuf_cstr(w), solr))
  40                 ret = 1;
  41             else
  42             {
  43                 yaz_log(YLOG_WARN, " expected: %s", solr ? solr : "null");
  44                 yaz_log(YLOG_WARN, " got:      %s", wrbuf_cstr(w));
  45             }
  46         }
  47     }
  48     wrbuf_destroy(w);
  49     odr_destroy(odr);
  50     return ret;
  51 }
  52
  53 static void tst1(void)
  54 {
  55     solr_transform_t ct = solr_transform_create();
  56
  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, "\"\"", "\"\""));
  61     YAZ_CHECK(compare(ct, "@not a b", "a AND NOT b"));
  62     YAZ_CHECK(compare(ct, "@and @or a b c", "(a OR b) AND c"));
  63     YAZ_CHECK(compare(ct, "@and a b", "a AND b"));
  64     YAZ_CHECK(compare(ct, "@or a b", "a OR b"));
  65     YAZ_CHECK(compare(ct, "@or a @and b c", "a OR (b AND c)"));
  66     YAZ_CHECK(compare(ct, "@or @and a b @and c d", "(a AND b) OR (c AND d)"));
  67     YAZ_CHECK(compare(ct, "@or @or a b @or c d", "(a OR b) OR (c OR d)"));
  68     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)"));
  69     YAZ_CHECK(compare(ct, "@and @and a b @and c d", "(a AND b) AND (c AND d)"));
  70     YAZ_CHECK(compare(ct, "@attr 1=field abc", "field:abc"));
  71     YAZ_CHECK(compare(ct, "@attr 1=field \"a b c\"", "field:\"a b c\""));
  72     YAZ_CHECK(compare(ct, "@attr 1=4 abc", 0)); /* should fail */
  73
  74     solr_transform_define_pattern(ct, "index.title", "1=4");
  75     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "title:abc"));
  76
  77     solr_transform_define_pattern(ct, "index.foo", "1=bar");
  78     YAZ_CHECK(compare(ct, "@attr 1=bar abc", "foo:abc"));
  79
  80     /* Truncation */
  81     YAZ_CHECK(compare(ct, "@attr 5=1 water", "water*"));
  82     YAZ_CHECK(compare(ct, "@attr 5=2 water", "*water"));
  83     YAZ_CHECK(compare(ct, "@attr 5=3 water", "*water*"));
  84     YAZ_CHECK(compare(ct, "@attr 5=100 water", "water"));
  85     YAZ_CHECK(compare(ct, "@attr 5=101 water", 0));
  86     YAZ_CHECK(compare(ct, "@attr 5=104 w#ter", "w?ter"));
  87     YAZ_CHECK(compare(ct, "@attr 5=104 w#t?r", "w?t*r"));
  88     YAZ_CHECK(compare(ct, "@attr 5=104 w#te?", "w?te*"));
  89     YAZ_CHECK(compare(ct, "@attr 5=104 w\\#te?", "w?te*")); /* PQF eats # */
  90     YAZ_CHECK(compare(ct, "@attr 5=104 w\\\\#te?", "w#te*"));
  91
  92     /* reserved characters */
  93     YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\\\\\",
  94                       "\\\"" "\\\\"));
  95     YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\\\\\\\\"",
  96                       "\\\"" "\\\\" "\\\""));
  97     YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\", "\\\"\\\\"));
  98
  99     YAZ_CHECK(compare(ct, "@attr 5=104 \\{:\\}", "\\{\\:\\}"));
 100
 101     YAZ_CHECK(compare(ct, "@attr 5=104 \\\"\\\\\\\\\\\"",
 102                       "\\\"" "\\\\" "\\\""));
 103
 104     YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=1 1980", "date:[* TO 1980}"));
 105     YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=2 1980", "date:[* TO 1980]"));
 106     YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=3 1980", "date:1980"));
 107     YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=4 1980", "date:[1980 TO *]"));
 108     YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=5 1980", "date:{1980 TO *]"));
 109     YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=2 234 @attr 1=date @attr 2=4 1990", "date:[1990 TO 234]"));
 110     YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=2 345 @attr 1=date @attr 2=4 234",  "date:[234 TO 345]"));
 111     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]"));
 112     YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=5 234 @attr 1=date @attr 2=2 1990", "date:{234 TO 1990]"));
 113     YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=4 234 @attr 1=date @attr 2=2 1990", "date:[234 TO 1990]"));
 114     YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=5 234 @attr 1=date @attr 2=1 1990", "date:{234 TO 1990}"));
 115     YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=4 234 @attr 1=date @attr 2=2 1990", "date:[234 TO 1990]"));
 116     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]"));
 117     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 *]"));
 118     YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=1 1980", "date:[* TO 1980}"));
 119     YAZ_CHECK(compare(ct, "@attr 1=date @attr 2=2 1980", "date:[* TO 1980]"));
 120     YAZ_CHECK(compare(ct, "@and @attr 1=date @attr 2=2 234 @attr 1=date @attr 2=4 1990", "date:[1990 TO 234]"));
 121
 122     solr_transform_close(ct);
 123 }
 124
 125 static void tst2(void)
 126 {
 127     WRBUF w = wrbuf_alloc();
 128     solr_transform_t ct = 0;
 129     const char *srcdir = getenv("srcdir");
 130     if (srcdir)
 131     {
 132         wrbuf_puts(w, srcdir);
 133         wrbuf_puts(w, "/");
 134     }
 135     wrbuf_puts(w, "../etc/pqf.properties");
 136
 137     ct = solr_transform_open_fname(wrbuf_cstr(w));
 138     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "dc.title:abc"));
 139 #if 0
 140     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=108 abc", "dc.title:abc"));
 141 #endif
 142
 143     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 3=1 @attr 6=1 abc", "dc.title:abc"));
 144     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=1 @attr 6=1 abc", "dc.title:abc"));
 145
 146     YAZ_CHECK(compare(ct, "@attr 1=1016 abc", "abc"));
 147     /* Date check */
 148     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=1 1980", "dc.date:[* TO 1980}"));
 149     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=2 1980", "dc.date:[* TO 1980]"));
 150     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=3 1980", "dc.date:1980"));
 151     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=4 1980", "dc.date:[1980 TO *]"));
 152     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=5 1980", "dc.date:{1980 TO *]"));
 153     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=2 234 @attr 1=30 @attr 2=4 1990", "dc.date:[1990 TO 234]"));
 154     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=2 345 @attr 1=30 @attr 2=4 234", "dc.date:[234 TO 345]"));
 155     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]"));
 156     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=5 234 @attr 1=30 @attr 2=2 1990", "dc.date:{234 TO 1990]"));
 157     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=4 234 @attr 1=30 @attr 2=2 1990", "dc.date:[234 TO 1990]"));
 158     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=5 234 @attr 1=30 @attr 2=1 1990", "dc.date:{234 TO 1990}"));
 159     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=4 234 @attr 1=30 @attr 2=2 1990", "dc.date:[234 TO 1990]"));
 160     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]"));
 161     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 *]"));
 162
 163 #if 0
 164     YAZ_CHECK(compare(ct, "@attr 2=103 @attr 1=_ALLRECORDS 1", "solr.allRecords=1"));
 165 #endif
 166     YAZ_CHECK(compare(ct, "@attr 1=500 abc", 0));
 167     solr_transform_close(ct);
 168     wrbuf_destroy(w);
 169 }
 170
 171 int main (int argc, char **argv)
 172 {
 173     YAZ_CHECK_INIT(argc, argv);
 174     YAZ_CHECK_LOG();
 175     tst1();
 176     tst2();
 177     YAZ_CHECK_TERM;
 178 }
 179
 180 /*
 181  * Local variables:
 182  * c-basic-offset: 4
 183  * c-file-style: "Stroustrup"
 184  * indent-tabs-mode: nil
 185  * End:
 186  * vim: shiftwidth=4 tabstop=8 expandtab
 187  */
 188