blob: 93cca72d3a487e342eb4ff471494d4940429af84 [file] [log] [blame]
Michael Hanlababa392013-12-05 21:57:37 +00001import com.fasterxml.jackson.core.JsonGenerationException;
Michael Hanl9ca5edd2013-12-06 05:13:24 +00002import com.fasterxml.jackson.core.JsonProcessingException;
Michael Hanlababa392013-12-05 21:57:37 +00003import com.fasterxml.jackson.databind.JsonMappingException;
Michael Hanldf206ab2014-05-13 10:22:27 +00004import de.ids_mannheim.korap.query.serialize.CollectionQueryBuilder;
Michael Hanl4fe41cc2013-12-10 17:59:51 +00005import de.ids_mannheim.korap.query.serialize.QuerySerializer;
Joachim Bingel16da4e12013-12-17 09:48:12 +00006import de.ids_mannheim.korap.util.QueryException;
Michael Hanlb64fdce2013-12-08 23:17:19 +00007import org.junit.Assert;
Michael Hanl55191d92013-12-05 11:53:10 +00008import org.junit.Test;
9import org.junit.runner.RunWith;
10import org.junit.runners.JUnit4;
11
12import java.io.IOException;
Michael Hanl55191d92013-12-05 11:53:10 +000013
14/**
15 * @author hanl
16 * @date 04/12/2013
17 */
18
19@RunWith(JUnit4.class)
20public class MetaQuerySerializationTest {
21
Michael Hanl55191d92013-12-05 11:53:10 +000022
23 @Test
24 public void test() throws IOException {
Michael Hanlb64fdce2013-12-08 23:17:19 +000025 StringBuffer b = new StringBuffer();
26 b.append("author:Goethe");
27 b.append(" AND ");
28 b.append("pubPlace:Erfurt");
29 b.append(" AND ");
30 b.append("textClass:wissenschaft");
Michael Hanl0b70bb32014-05-06 16:09:41 +000031// CollectionQuery qu = new CollectionQuery().addMetaFilterQuery(b.toString());
Michael Hanldf206ab2014-05-13 10:22:27 +000032 CollectionQueryBuilder query = new CollectionQueryBuilder().addMetaFilterQuery(b.toString());
Michael Hanlbaf1a5e2014-05-15 19:51:40 +000033 System.out.println(query.toJSON());
Michael Hanl0b70bb32014-05-06 16:09:41 +000034 System.out.println(query.toCollections());
Michael Hanlb64fdce2013-12-08 23:17:19 +000035// System.out.println("value reference " + qu.stringify());
36// System.out.println();
Michael Hanl55191d92013-12-05 11:53:10 +000037 }
38
39 @Test
40 public void testSingle() throws IOException {
Michael Hanldf206ab2014-05-13 10:22:27 +000041 CollectionQueryBuilder query = new CollectionQueryBuilder().addMetaFilter("textClass", "wissenschaft");
Michael Hanlb64fdce2013-12-08 23:17:19 +000042// System.out.println("------ TEXT SINGLE " + query.stringify());
Michael Hanlbaf1a5e2014-05-15 19:51:40 +000043 System.out.println(query.toJSON());
Michael Hanl296a0312013-12-05 17:27:51 +000044 }
Michael Hanl55191d92013-12-05 11:53:10 +000045
Michael Hanl2bd00c62014-11-04 16:26:42 +000046// @Test
Michael Hanl296a0312013-12-05 17:27:51 +000047 public void testDates() throws IOException {
Michael Hanlb64fdce2013-12-08 23:17:19 +000048 StringBuffer b = new StringBuffer();
Michael Hanl4fe13462013-12-11 18:03:42 +000049 b.append("pubDate:>2013-04-01");
Michael Hanlb64fdce2013-12-08 23:17:19 +000050 b.append(" AND ");
Michael Hanl4fe13462013-12-11 18:03:42 +000051 b.append("pubDate:<2012-04-01");
Michael Hanlb64fdce2013-12-08 23:17:19 +000052 b.append(" AND ");
53 b.append("author:Goethe");
Michael Hanldf206ab2014-05-13 10:22:27 +000054 CollectionQueryBuilder query = new CollectionQueryBuilder().addMetaFilterQuery(b.toString());
Michael Hanlbaf1a5e2014-05-15 19:51:40 +000055 System.out.println("value until/since : " + query.toJSON());
Michael Hanl0b70bb32014-05-06 16:09:41 +000056 System.out.println("meta value until/since " + query.toCollections());
Michael Hanlbaf1a5e2014-05-15 19:51:40 +000057 Assert.assertEquals("[{\"@type\":\"korap:meta-filter\",\"@value\":{\"@type\":\"korap:group\",\"relation\":\"and\",\"operands\":[{\"@type\":\"korap:term\",\"@field\":\"korap:field#author\",\"@value\":\"Goethe\"},{\"@type\":\"korap:group\",\"@field\":\"korap:field#pubDate\",\"relation\":\"between\",\"operands\":[{\"@type\":\"korap:date\",\"@value\":\"2012-04-01\"},{\"@type\":\"korap:date\",\"@value\":\"2013-04-01\"}]}]}}]", query.toJSON());
Michael Hanlb64fdce2013-12-08 23:17:19 +000058// System.out.println();
Michael Hanl55191d92013-12-05 11:53:10 +000059 }
Michael Hanl08b93ed2013-12-05 18:04:45 +000060
Michael Hanl2bd00c62014-11-04 16:26:42 +000061// @Test
Michael Hanl08b93ed2013-12-05 18:04:45 +000062 public void testUntil() throws IOException {
Michael Hanlb64fdce2013-12-08 23:17:19 +000063 StringBuffer b = new StringBuffer();
Michael Hanl4fe13462013-12-11 18:03:42 +000064 b.append("pubDate:>2013-12-10");
Michael Hanlb64fdce2013-12-08 23:17:19 +000065 b.append(" AND ");
66 b.append("author:Hesse");
Michael Hanldf206ab2014-05-13 10:22:27 +000067 CollectionQueryBuilder query = new CollectionQueryBuilder().addMetaFilterQuery(b.toString());
Michael Hanlb64fdce2013-12-08 23:17:19 +000068 System.out.println("Running date check (until) with additional attribute author");
Michael Hanlbaf1a5e2014-05-15 19:51:40 +000069 Assert.assertEquals("[{\"@type\":\"korap:meta-filter\",\"@value\":{\"@type\":\"korap:group\",\"relation\":\"and\",\"operands\":[{\"@type\":\"korap:term\",\"@field\":\"korap:field#author\",\"@value\":\"Hesse\"},{\"@type\":\"korap:group\",\"@field\":\"korap:field#pubDate\",\"relation\":\"until\",\"operands\":[{\"@type\":\"korap:date\",\"@value\":\"2013-12-10\"}]}]}}]", query.toJSON());
Michael Hanlb64fdce2013-12-08 23:17:19 +000070// System.out.println("value until : " + query.stringify());
71// System.out.println();
Michael Hanl08b93ed2013-12-05 18:04:45 +000072 }
73
Michael Hanl2bd00c62014-11-04 16:26:42 +000074// @Test
Michael Hanl08b93ed2013-12-05 18:04:45 +000075 public void testSince() throws IOException {
Michael Hanlb64fdce2013-12-08 23:17:19 +000076 StringBuffer b = new StringBuffer();
Michael Hanl4fe13462013-12-11 18:03:42 +000077 b.append("pubDate:<2013-12-10");
Michael Hanlb64fdce2013-12-08 23:17:19 +000078 b.append(" AND ");
79 b.append("author:Kafka");
Michael Hanldf206ab2014-05-13 10:22:27 +000080 CollectionQueryBuilder query = new CollectionQueryBuilder().addMetaFilterQuery(b.toString());
Michael Hanlbaf1a5e2014-05-15 19:51:40 +000081 System.out.println("value since : " + query.toJSON());
Michael Hanl0b70bb32014-05-06 16:09:41 +000082 System.out.println("meta value since " + query.toCollections());
Michael Hanlb64fdce2013-12-08 23:17:19 +000083// System.out.println();
84 System.out.println("Running date check (since) with additional attribute author");
Michael Hanlbaf1a5e2014-05-15 19:51:40 +000085 Assert.assertEquals("[{\"@type\":\"korap:meta-filter\",\"@value\":{\"@type\":\"korap:group\",\"relation\":\"and\",\"operands\":[{\"@type\":\"korap:term\",\"@field\":\"korap:field#author\",\"@value\":\"Kafka\"},{\"@type\":\"korap:group\",\"@field\":\"korap:field#pubDate\",\"relation\":\"since\",\"operands\":[{\"@type\":\"korap:date\",\"@value\":\"2013-12-10\"}]}]}}]", query.toJSON());
Michael Hanl08b93ed2013-12-05 18:04:45 +000086 }
87
Michael Hanl2bd00c62014-11-04 16:26:42 +000088
Michael Hanl53b0fd02013-12-06 21:07:52 +000089 @Test
Michael Hanl2bd00c62014-11-04 16:26:42 +000090 public void testMeta() {
91 QuerySerializer s = new QuerySerializer();
92 try {
93 s.setQuery("shrink({[base=foo]})", "poliqarp");
94 s.addMeta("2-token, 2-token", 10, 0);
95 } catch (QueryException e) {
96 e.printStackTrace();
97 }
98 System.out.println("THE RESULTING QUERY: " + s.toJSON());
99 }
100
101 // @Test
Joachim Bingel16da4e12013-12-17 09:48:12 +0000102 public void testGenerator() throws QueryException {
Michael Hanld0618712014-08-01 15:23:25 +0000103 /*
Michael Hanlababa392013-12-05 21:57:37 +0000104 * just for testing...
105 */
Michael Hanl4fe41cc2013-12-10 17:59:51 +0000106 QuerySerializer jg = new QuerySerializer();
Michael Hanlababa392013-12-05 21:57:37 +0000107 int i = 0;
108 String[] queries;
109 queries = new String[]{
110 "shrink({[base=foo]})",
111 "shrink({[base=foo]}[orth=bar])",
112 "shrink(1:[base=Der]{1:[base=Mann]})",
113 };
114
115 for (String q : queries) {
116 i++;
117 try {
118 System.out.println(q);
119 jg.run(q, "poliqarp", System.getProperty("user.home") + "/bsp" + i + ".json");
120 System.out.println();
121 } catch (NullPointerException npe) {
122 npe.printStackTrace();
123 System.out.println("null\n");
Michael Hanl560b4cd2013-12-06 14:26:44 +0000124 System.out.println();
Michael Hanlababa392013-12-05 21:57:37 +0000125 } catch (JsonGenerationException e) {
126 e.printStackTrace();
127 } catch (JsonMappingException e) {
128 e.printStackTrace();
129 } catch (IOException e) {
130 e.printStackTrace();
131 }
132 }
133 }
134
Michael Hanl9ca5edd2013-12-06 05:13:24 +0000135 @Test
136 public void testLists() {
Michael Hanlb64fdce2013-12-08 23:17:19 +0000137 StringBuffer b = new StringBuffer();
Michael Hanl4fe13462013-12-11 18:03:42 +0000138 b.append("pubDate:<2013-12-10");
Michael Hanlb64fdce2013-12-08 23:17:19 +0000139 b.append(" AND ");
140 b.append("author:Kafka");
Michael Hanldf206ab2014-05-13 10:22:27 +0000141 CollectionQueryBuilder q = new CollectionQueryBuilder().addMetaFilterQuery(b.toString());
Michael Hanl1ee40e62014-03-26 17:22:26 +0000142 q.addMetaExtend("author", "Hesse");
Michael Hanl9ca5edd2013-12-06 05:13:24 +0000143
Michael Hanlbaf1a5e2014-05-15 19:51:40 +0000144 System.out.println("--- ALL " + q.toJSON());
Michael Hanl1ee40e62014-03-26 17:22:26 +0000145 System.out.println();
Michael Hanl560b4cd2013-12-06 14:26:44 +0000146
Michael Hanl9ca5edd2013-12-06 05:13:24 +0000147 }
148
149 @Test
150 public void testJSONArray() throws JsonProcessingException {
Michael Hanlb64fdce2013-12-08 23:17:19 +0000151 StringBuffer b = new StringBuffer();
Michael Hanl4fe13462013-12-11 18:03:42 +0000152 b.append("pubDate:<2013-12-10");
Michael Hanlb64fdce2013-12-08 23:17:19 +0000153 b.append(" AND ");
154 b.append("author:Kafka");
Michael Hanldf206ab2014-05-13 10:22:27 +0000155 CollectionQueryBuilder q = new CollectionQueryBuilder().addMetaExtendQuery(b.toString());
Michael Hanlbaf1a5e2014-05-15 19:51:40 +0000156 System.out.println("array repres " + q.toJSON());
Michael Hanl1ee40e62014-03-26 17:22:26 +0000157 System.out.println();
Michael Hanl560b4cd2013-12-06 14:26:44 +0000158 }
159
160 @Test
161 public void testCollections() throws IOException {
Michael Hanldf206ab2014-05-13 10:22:27 +0000162 CollectionQueryBuilder q = new CollectionQueryBuilder().addMetaFilter("corpusID", "A00");
Michael Hanl698da8c2013-12-08 21:12:36 +0000163 q.addMetaExtend("corpusID", "A01");
164
Michael Hanlbaf1a5e2014-05-15 19:51:40 +0000165 System.out.println("results stringified " + q.toJSON());
Michael Hanl0b70bb32014-05-06 16:09:41 +0000166 System.out.println("results to meta" + q.toCollections());
Michael Hanl1ee40e62014-03-26 17:22:26 +0000167 System.out.println();
Michael Hanl53b0fd02013-12-06 21:07:52 +0000168 }
169
Michael Hanl698da8c2013-12-08 21:12:36 +0000170 /**
171 * asserts equality. input should be equal to output,
172 * since there is no other metadata added to the meta query
173 *
174 * @throws IOException
175 */
Michael Hanl53b0fd02013-12-06 21:07:52 +0000176 @Test
177 public void testResources() throws IOException {
178 String meta = "[{\"@type\":\"korap:meta-filter\",\"@value\":{\"@type\":\"korap:term\",\"@field\":\"korap:field#corpusID\",\"@value\":\"WPD\"}}]";
Michael Hanldf206ab2014-05-13 10:22:27 +0000179 CollectionQueryBuilder q = new CollectionQueryBuilder().addResource(meta);
Michael Hanlb64fdce2013-12-08 23:17:19 +0000180 System.out.println("Testing Resource Meta data");
Michael Hanl0b70bb32014-05-06 16:09:41 +0000181 org.junit.Assert.assertEquals("{\"collections\":" + meta + "}", q.toCollections());
Michael Hanl53b0fd02013-12-06 21:07:52 +0000182 }
183
184 @Test
185 public void testA00() throws IOException {
Michael Hanldf206ab2014-05-13 10:22:27 +0000186 CollectionQueryBuilder q = new CollectionQueryBuilder().addMetaExtend("corpusID", "A00").addMetaExtend("corpusID", "A01");
Michael Hanlbaf1a5e2014-05-15 19:51:40 +0000187 System.out.println("A meta: " + q.toJSON());
Michael Hanl1ee40e62014-03-26 17:22:26 +0000188 System.out.println();
Michael Hanl53b0fd02013-12-06 21:07:52 +0000189 }
190
Michael Hanl53b0fd02013-12-06 21:07:52 +0000191 @Test
Michael Hanlb64fdce2013-12-08 23:17:19 +0000192 public void testResources2() throws IOException {
Michael Hanl698da8c2013-12-08 21:12:36 +0000193 String meta = "[{\"@type\":\"korap:meta-filter\",\"@value\":{\"@type\":\"korap:term\",\"@field\":\"korap:field#corpusID\",\"@value\":\"WPD\"}}]";
Michael Hanldf206ab2014-05-13 10:22:27 +0000194 CollectionQueryBuilder q = new CollectionQueryBuilder().addResource(meta);
Michael Hanl698da8c2013-12-08 21:12:36 +0000195 q.addMetaFilter("textClass", "wissenschaft");
Michael Hanlbaf1a5e2014-05-15 19:51:40 +0000196 System.out.println("stringified meta " + q.toJSON());
Michael Hanl0b70bb32014-05-06 16:09:41 +0000197 System.out.println("meta string " + q.toCollections());
Michael Hanl9ca5edd2013-12-06 05:13:24 +0000198 }
199
Michael Hanl55191d92013-12-05 11:53:10 +0000200}