From: Dennis Schafroth Date: Wed, 11 Aug 2010 14:36:56 +0000 (+0200) Subject: Test cases for RPN to SOLR conversion X-Git-Tag: v4.1.0~40^2~1^2~6 X-Git-Url: http://lists.indexdata.dk/?a=commitdiff_plain;h=582c324a5b5dc365195133e92f4ca6d02991e43b;p=yaz-moved-to-github.git Test cases for RPN to SOLR conversion --- diff --git a/test/test_rpn2solr.c b/test/test_rpn2solr.c new file mode 100644 index 0000000..8367d8c --- /dev/null +++ b/test/test_rpn2solr.c @@ -0,0 +1,117 @@ +/* This file is part of the YAZ toolkit. + * Copyright (C) 1995-2010 Index Data + * See the file LICENSE for details. + */ + +#include +#include + +#include +#include +#include +#include +#include + +static int compare(solr_transform_t ct, const char *pqf, const char *solr) +{ + int ret = 0; + ODR odr = odr_createmem(ODR_ENCODE); + WRBUF w = wrbuf_alloc(); + Z_RPNQuery *q = p_query_rpn(odr, pqf); + + if (q) + { + int r = solr_transform_rpn2solr_wrbuf(ct, w, q); + + if (r != 0) + { + /* transform error */ + yaz_log(YLOG_LOG, "%s -> Error %d", pqf, r); + if (!solr) /* also expected error? */ + ret = 1; + } + else if (r == 0) + { + yaz_log(YLOG_LOG, "%s -> %s", pqf, wrbuf_cstr(w)); + if (solr && !strcmp(wrbuf_cstr(w), solr)) + { + ret = 1; + } + } + } + wrbuf_destroy(w); + odr_destroy(odr); + return ret; +} + +static void tst1(void) +{ + solr_transform_t ct = solr_transform_create(); + + YAZ_CHECK(compare(ct, "abc", "abc")); + YAZ_CHECK(compare(ct, "\"a b c\"", "\"a b c\"")); + YAZ_CHECK(compare(ct, "a b", "a b")); + YAZ_CHECK(compare(ct, "@not a b", "a AND NOT b")); + YAZ_CHECK(compare(ct, "@and @or a b c", "(a OR b) AND c")); + YAZ_CHECK(compare(ct, "@and a b", "a AND b")); + YAZ_CHECK(compare(ct, "@or a b", "a OR b")); + YAZ_CHECK(compare(ct, "@attr 1=field abc", "field:abc")); + YAZ_CHECK(compare(ct, "@attr 1=4 abc", 0)); /* should fail */ + + solr_transform_define_pattern(ct, "index.title", "1=4"); + YAZ_CHECK(compare(ct, "@attr 1=4 abc", "title:abc")); + + solr_transform_define_pattern(ct, "index.foo", "1=bar"); + YAZ_CHECK(compare(ct, "@attr 1=bar abc", "foo:abc")); + + solr_transform_close(ct); +} + +static void tst2(void) +{ + WRBUF w = wrbuf_alloc(); + solr_transform_t ct = 0; + const char *srcdir = getenv("srcdir"); + if (srcdir) + { + wrbuf_puts(w, srcdir); + wrbuf_puts(w, "/"); + } + wrbuf_puts(w, "../etc/pqf.properties"); + + ct = solr_transform_open_fname(wrbuf_cstr(w)); + YAZ_CHECK(compare(ct, "@attr 1=4 abc", "dc.title:abc")); + YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=108 abc", "dc.title:abc")); + YAZ_CHECK(compare(ct, "@attr 1=4 @attr 3=1 @attr 6=1 abc", "dc.title:abc")); + YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=1 @attr 6=1 abc", "dc.title:abc")); + YAZ_CHECK(compare(ct, "@attr 1=1016 abc", "abc")); + YAZ_CHECK(compare(ct, "@attr 2=1 @attr 1=30 1980", "dc.date:[* to 1980]")); + YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=3 1980", "dc.date:1980")); + YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=5 1980", "dc.date:[* to 1980]")); + YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=2 1980", "dc.date:[* to 1980]")); + YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=4 1980", "dc.date:[1980 to *]")); + + YAZ_CHECK(compare(ct, "@attr 2=103 @attr 1=_ALLRECORDS 1", "solr.allRecords=1")); + YAZ_CHECK(compare(ct, "@attr 1=500 abc", 0)); + solr_transform_close(ct); + wrbuf_destroy(w); +} + +int main (int argc, char **argv) +{ + YAZ_CHECK_INIT(argc, argv); + YAZ_CHECK_LOG(); + tst1(); + tst2(); + YAZ_CHECK_TERM; +} + +/* + * Local variables: + * c-basic-offset: 4 + * c-file-style: "Stroustrup" + * indent-tabs-mode: nil + * End: + * vim: shiftwidth=4 tabstop=8 expandtab + */ +