Add additional tests

Author: ahornace

suggester/src/test/java/org/opengrok/suggest/FieldWFSTCollectionTest.java

@@ -41,17 +41,20 @@
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
+import java.util.AbstractMap.SimpleEntry;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.HashSet;
 import java.util.List;
+import java.util.Map.Entry;
 import java.util.stream.Collectors;
 
 import static junit.framework.Assert.assertTrue;
 import static junit.framework.TestCase.assertEquals;
 import static junit.framework.TestCase.assertFalse;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.collection.IsIterableContainingInAnyOrder.containsInAnyOrder;
+import static org.hamcrest.collection.IsIterableContainingInOrder.contains;
 
 public class FieldWFSTCollectionTest {
 
@@ -288,4 +291,20 @@ public void testUnknownFieldIgnored() throws IOException {
         assertTrue(res.isEmpty());
     }
 
+    @Test
+    public void testGetSearchCountMapSorted() throws IOException {
+        addText(FIELD, "test1 test2");
+        init(true);
+
+        Term t1 = new Term(FIELD, "test1");
+        Term t2 = new Term(FIELD, "test2");
+
+        f.incrementSearchCount(t1, 10);
+        f.incrementSearchCount(t2, 5);
+
+        List<Entry<BytesRef, Integer>> searchCounts = f.getSearchCountsSorted(FIELD, 0, 10);
+
+        assertThat(searchCounts, contains(new SimpleEntry<>(t1.bytes(), 10), new SimpleEntry<>(t2.bytes(), 5)));
+    }
+
 }

suggester/src/test/java/org/opengrok/suggest/SuggesterTest.java

@@ -0,0 +1,59 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * See LICENSE.txt included in this distribution for the specific
+ * language governing permissions and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at LICENSE.txt.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
+ */
+package org.opengrok.suggest;
+
+import org.junit.Test;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.time.Duration;
+
+public class SuggesterTest {
+
+    @Test(expected = IllegalArgumentException.class)
+    public void testNullSuggesterDir() {
+        new Suggester(null, 10, Duration.ofMinutes(5), false, true, null);
+    }
+
+    @Test(expected = IllegalArgumentException.class)
+    public void testNullDuration() throws IOException {
+        Path tempFile = Files.createTempFile("opengrok", "test");
+        try {
+            new Suggester(tempFile.toFile(), 10, null, false, true, null);
+        } finally {
+            tempFile.toFile().delete();
+        }
+    }
+
+    @Test(expected = IllegalArgumentException.class)
+    public void testNegativeDuration() throws IOException {
+        Path tempFile = Files.createTempFile("opengrok", "test");
+        try {
+            new Suggester(tempFile.toFile(), 10, Duration.ofMinutes(-4), false, true, null);
+        } finally {
+            tempFile.toFile().delete();
+        }
+    }
+
+}

suggester/src/test/java/org/opengrok/suggest/query/SuggesterFuzzyQueryTest.java

@@ -0,0 +1,38 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * See LICENSE.txt included in this distribution for the specific
+ * language governing permissions and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at LICENSE.txt.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
+ */
+package org.opengrok.suggest.query;
+
+import org.apache.lucene.index.Term;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+public class SuggesterFuzzyQueryTest {
+
+    @Test
+    public void testLength() {
+        SuggesterFuzzyQuery q = new SuggesterFuzzyQuery(new Term("test", "test"), 2, 0);
+        assertEquals("test".length(), q.length());
+    }
+
+}

suggester/src/test/java/org/opengrok/suggest/query/SuggesterPrefixQueryTest.java

@@ -0,0 +1,38 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * See LICENSE.txt included in this distribution for the specific
+ * language governing permissions and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at LICENSE.txt.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
+ */
+package org.opengrok.suggest.query;
+
+import org.apache.lucene.index.Term;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+public class SuggesterPrefixQueryTest {
+
+    @Test
+    public void testLength() {
+        SuggesterPrefixQuery q = new SuggesterPrefixQuery(new Term("test", "prefix"));
+        assertEquals("prefix".length(), q.length());
+    }
+
+}

suggester/src/test/java/org/opengrok/suggest/query/SuggesterRangeQueryTest.java

@@ -0,0 +1,46 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * See LICENSE.txt included in this distribution for the specific
+ * language governing permissions and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at LICENSE.txt.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
+ */
+package org.opengrok.suggest.query;
+
+import org.apache.lucene.util.BytesRef;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+public class SuggesterRangeQueryTest {
+
+    @Test
+    public void testLengthLower() {
+        SuggesterRangeQuery q = new SuggesterRangeQuery("test", new BytesRef("lowerTerm"), new BytesRef("upper"),
+                true, false, SuggesterRangeQuery.SuggestPosition.LOWER);
+        assertEquals("lowerTerm".length(), q.length());
+    }
+
+    @Test
+    public void testLengthUpper() {
+        SuggesterRangeQuery q = new SuggesterRangeQuery("test", new BytesRef("l"), new BytesRef("upper"),
+                true, false, SuggesterRangeQuery.SuggestPosition.UPPER);
+        assertEquals("upper".length(), q.length());
+    }
+
+}

suggester/src/test/java/org/opengrok/suggest/query/SuggesterRegexpQueryTest.java

@@ -0,0 +1,37 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * See LICENSE.txt included in this distribution for the specific
+ * language governing permissions and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at LICENSE.txt.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
+ */
+package org.opengrok.suggest.query;
+
+import org.apache.lucene.index.Term;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class SuggesterRegexpQueryTest {
+
+    @Test
+    public void testLength() {
+        SuggesterRegexpQuery q = new SuggesterRegexpQuery(new Term("test", ".*test"));
+        Assert.assertEquals(".*test".length(), q.length());
+    }
+
+}

suggester/src/test/java/org/opengrok/suggest/query/SuggesterWildcardQueryTest.java

@@ -0,0 +1,37 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * See LICENSE.txt included in this distribution for the specific
+ * language governing permissions and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at LICENSE.txt.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
+ */
+package org.opengrok.suggest.query;
+
+import org.apache.lucene.index.Term;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class SuggesterWildcardQueryTest {
+
+    @Test
+    public void testLength() {
+        SuggesterWildcardQuery q = new SuggesterWildcardQuery(new Term("test", "m?t"));
+        Assert.assertEquals("m?t".length(), q.length());
+    }
+
+}