1 files changed, 82 insertions, 0 deletions
diff --git a/predicate-search/src/test/java/com/yahoo/search/predicate/index/PredicateIntervalStoreTest.java b/predicate-search/src/test/java/com/yahoo/search/predicate/index/PredicateIntervalStoreTest.java
new file mode 100644
index 00000000000..1f2fd390cb5
--- /dev/null
+++ b/predicate-search/src/test/java/com/yahoo/search/predicate/index/PredicateIntervalStoreTest.java
@@ -0,0 +1,82 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.search.predicate.index;
+
+import com.google.common.primitives.Ints;
+import org.junit.Test;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import static com.yahoo.search.predicate.serialization.SerializationTestHelper.assertSerializationDeserializationMatches;
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author bjorncs
+ */
+public class PredicateIntervalStoreTest {
+
+    @Test(expected = IllegalArgumentException.class)
+    public void requireThatEmptyIntervalListThrows() {
+        PredicateIntervalStore.Builder builder = new PredicateIntervalStore.Builder();
+        builder.insert(new ArrayList<>());
+    }
+
+    @Test
+    public void requireThatSingleIntervalCanBeInserted() {
+        testInsertAndRetrieve(0x0001ffff);
+    }
+
+    @Test
+    public void requireThatMultiIntervalEntriesCanBeInserted() {
+        testInsertAndRetrieve(0x00010001, 0x00020002, 0x0003ffff);
+        testInsertAndRetrieve(0x00010001, 0x00020002, 0x00030003, 0x00040004, 0x00050005, 0x00060006,
+                0x00070007, 0x00080008, 0x00090009, 0x000a000a);
+    }
+
+    @Test
+    public void requireThatDifferentSizeIntervalArraysCanBeInserted() {
+        PredicateIntervalStore.Builder builder = new PredicateIntervalStore.Builder();
+        int intervals1[] = new int[] {0x00010001, 0x00020002};
+        int intervals2[] = new int[] {0x00010001, 0x00020002, 0x00030003};
+        assertEquals(0, builder.insert(Ints.asList(intervals1)));
+        assertEquals(1, builder.insert(Ints.asList(intervals2)));
+    }
+
+    @Test
+    public void requireThatSerializationAndDeserializationRetainIntervals() throws IOException {
+        PredicateIntervalStore.Builder builder = new PredicateIntervalStore.Builder();
+        builder.insert(Arrays.asList(0x00010001, 0x00020002));
+        builder.insert(Arrays.asList(0x00010001, 0x00020002, 0x00030003));
+        builder.insert(Arrays.asList(0x0fffffff, 0x00020002, 0x00030003));
+        PredicateIntervalStore store = builder.build();
+        assertSerializationDeserializationMatches(
+                store, PredicateIntervalStore::writeToOutputStream, PredicateIntervalStore::fromInputStream);
+    }
+
+    @Test
+    public void requireThatEqualIntervalListsReturnsSameReference() {
+        PredicateIntervalStore.Builder builder = new PredicateIntervalStore.Builder();
+        List<Integer> intervals1 = Arrays.asList(0x00010001, 0x00020002);
+        List<Integer> intervals2 = Arrays.asList(0x00010001, 0x00020002);
+        int ref1 = builder.insert(intervals1);
+        int ref2 = builder.insert(intervals2);
+        PredicateIntervalStore store = builder.build();
+        int[] a1 = store.get(ref1);
+        int[] a2 = store.get(ref2);
+        assertTrue(a1 == a2);
+    }
+
+    private static void testInsertAndRetrieve(int... intervals) {
+        PredicateIntervalStore.Builder builder = new PredicateIntervalStore.Builder();
+        int ref = builder.insert(Ints.asList(intervals));
+        PredicateIntervalStore store = builder.build();
+
+        int retrieved[] = store.get(ref);
+        assertArrayEquals(intervals, retrieved);
+    }
+
+}