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 compare(cql_transform_t ct, const char *pqf, const char *cql)
  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 = cql_transform_rpn2cql_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 (!cql) /* 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 (cql && !strcmp(wrbuf_cstr(w), cql))
  40             {
  41                 ret = 1;
  42             }
  43             else
  44             {
  45                 yaz_log(YLOG_WARN, " expected: %s", cql ? cql : "null");
  46                 yaz_log(YLOG_WARN, " got:      %s", wrbuf_cstr(w));
  47             }
  48         }
  49     }
  50     wrbuf_destroy(w);
  51     odr_destroy(odr);
  52     return ret;
  53 }
  54
  55 static void tst1(void)
  56 {
  57     cql_transform_t ct = cql_transform_create();
  58
  59     YAZ_CHECK(compare(ct, "abc", "abc"));
  60     YAZ_CHECK(compare(ct, "\"a b c\"", "\"a b c\""));
  61     YAZ_CHECK(compare(ct, "@and a b", "a and b"));
  62     YAZ_CHECK(compare(ct, "@or a @and b c", "a or (b and c)"));
  63     YAZ_CHECK(compare(ct, "@or @and a b @and c d", "(a and b) or (c and d)"));
  64     YAZ_CHECK(compare(ct, "@or @or a b @or c d", "(a or b) or (c or d)"));
  65     YAZ_CHECK(compare(ct, "@and @and a b @and c d", "(a and b) and (c and d)"));
  66
  67     YAZ_CHECK(compare(ct, "@attr 1=field abc", "field=abc"));
  68     YAZ_CHECK(compare(ct, "@attr 1=4 abc", 0)); /* should fail */
  69
  70     cql_transform_define_pattern(ct, "index.title", "1=4");
  71     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "title=abc"));
  72
  73     cql_transform_define_pattern(ct, "index.foo", "1=bar");
  74     YAZ_CHECK(compare(ct, "@attr 1=bar abc", "foo=abc"));
  75
  76     cql_transform_close(ct);
  77 }
  78
  79 static void tst2(void)
  80 {
  81     WRBUF w = wrbuf_alloc();
  82     cql_transform_t ct = 0;
  83     const char *srcdir = getenv("srcdir");
  84     if (srcdir)
  85     {
  86         wrbuf_puts(w, srcdir);
  87         wrbuf_puts(w, "/");
  88     }
  89     wrbuf_puts(w, "../etc/pqf.properties");
  90
  91     ct = cql_transform_open_fname(wrbuf_cstr(w));
  92     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "dc.title=abc"));
  93     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=108 abc", "dc.title=/exact abc"));
  94     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 3=1 @attr 6=1 abc", "dc.title=abc"));
  95     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=1 @attr 6=1 abc", "dc.title=abc"));
  96     YAZ_CHECK(compare(ct, "@attr 1=1016 abc", "abc"));
  97     /* Date tests */
  98     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=1 1980", "dc.date<1980"));
  99     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=2 1980", "dc.date<=1980"));
 100     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=3 1980", "dc.date=1980"));
 101     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=4 1980", "dc.date>=1980"));
 102     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=5 1980", "dc.date>1980"));
 103     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"));
 104
 105     /* Truncation */
 106     YAZ_CHECK(compare(ct, "@attr 5=1 water", "water*"));
 107     YAZ_CHECK(compare(ct, "@attr 5=2 water", "*water"));
 108     YAZ_CHECK(compare(ct, "@attr 5=3 water", "*water*"));
 109     YAZ_CHECK(compare(ct, "@attr 5=100 water", "water"));
 110     YAZ_CHECK(compare(ct, "@attr 5=102 water", "water"));
 111     YAZ_CHECK(compare(ct, "@attr 5=104 water", "water"));
 112
 113     YAZ_CHECK(compare(ct, "@attr 5=102 wat.*er", "wat*er"));
 114     YAZ_CHECK(compare(ct, "@attr 5=104 wat?er", "wat*er"));
 115
 116     YAZ_CHECK(compare(ct, "@attr 5=102 wat.er", "wat?er"));
 117     YAZ_CHECK(compare(ct, "@attr 5=104 wat#er", "wat?er"));
 118     YAZ_CHECK(compare(ct, "@attr 5=102 wat?er", "wat\\?er"));
 119     YAZ_CHECK(compare(ct, "@attr 5=104 wat*er", "wat\\*er"));
 120     YAZ_CHECK(compare(ct, "@attr 5=102 wat#er", "wat#er"));
 121
 122     /* \. is 'eaten' by PQF parser */
 123     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\.er", "wat?er"));
 124
 125     /* Escape sequences */
 126     /* note: escape sequences that survive after PQF parse below */
 127     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\?er", "wat\\?er"));
 128     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\?er", "wat\\?er"));
 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, "wat\\\\#er", "wat#er"));
 134     YAZ_CHECK(compare(ct, "@attr 5=100 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     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\.er", "wat.er"));
 139
 140     /* Quoting */
 141     YAZ_CHECK(compare(ct, "@attr 5=100 \"\"", "\"\""));
 142     YAZ_CHECK(compare(ct, "@attr 5=1 \"\"", "\"*\""));
 143     YAZ_CHECK(compare(ct, "@attr 5=2 \"\"", "\"*\""));
 144     YAZ_CHECK(compare(ct, "@attr 5=3 \"\"", "\"**\""));
 145     YAZ_CHECK(compare(ct, "@attr 5=102 \"\"", "\"\""));
 146     YAZ_CHECK(compare(ct, "@attr 5=104 \"\"", "\"\""));
 147
 148     YAZ_CHECK(compare(ct, "@attr 5=1 \"water basket\"", "\"water basket*\""));
 149     YAZ_CHECK(compare(ct, "@attr 5=2 \"water basket\"", "\"*water basket\""));
 150     YAZ_CHECK(compare(ct, "@attr 5=3 \"water basket\"", "\"*water basket*\""));
 151
 152     /* Other */
 153     YAZ_CHECK(compare(ct, "@attr 2=103 @attr 1=_ALLRECORDS 1", "cql.allRecords=1"));
 154     YAZ_CHECK(compare(ct, "@attr 1=500 abc", 0));
 155     cql_transform_close(ct);
 156     wrbuf_destroy(w);
 157 }
 158
 159 int main (int argc, char **argv)
 160 {
 161     YAZ_CHECK_INIT(argc, argv);
 162     YAZ_CHECK_LOG();
 163     tst1();
 164     tst2();
 165     YAZ_CHECK_TERM;
 166 }
 167
 168 /*
 169  * Local variables:
 170  * c-basic-offset: 4
 171  * c-file-style: "Stroustrup"
 172  * indent-tabs-mode: nil
 173  * End:
 174  * vim: shiftwidth=4 tabstop=8 expandtab
 175  */
 176