1 files changed, 59 insertions, 0 deletions
diff --git a/vespajlib/src/test/java/com/yahoo/text/SnippetGeneratorTest.java b/vespajlib/src/test/java/com/yahoo/text/SnippetGeneratorTest.java
new file mode 100644
index 00000000000..e8ffdb16978
--- /dev/null
+++ b/vespajlib/src/test/java/com/yahoo/text/SnippetGeneratorTest.java
@@ -0,0 +1,59 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.text;
+
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * @author hakon
+ */
+public class SnippetGeneratorTest {
+    private final SnippetGenerator generator = new SnippetGenerator();
+
+    private void assertSnippet(String text, int sizeHint, String expectedSnippet) {
+        assertEquals(expectedSnippet, generator.makeSnippet(text, sizeHint));
+    }
+
+    @Test
+    public void prefixSnippetForReallySmallSizeHint() {
+        assertSnippet(
+                "This is a long text that should be snippeted", 0,
+                "[...44 chars omitted]");
+
+        assertSnippet(
+                "This is a long text that should be snippeted", 1,
+                "[...44 chars omitted]");
+    }
+
+    @Test
+    public void snippet() {
+        assertSnippet(
+                "This is a long text that should be snippeted", 23,
+                "[...44 chars omitted]");
+
+        assertSnippet(
+                "This is a long text that should be snippeted", 24,
+                "T[...43 chars omitted]");
+
+        assertSnippet(
+                "This is a long text that should be snippeted", 30,
+                "This[...37 chars omitted]ted");
+
+    }
+
+    @Test
+    public void noShorteningNeeded() {
+        assertSnippet(
+                "This is a long text that should be snippeted", 39,
+                "This is [...28 chars omitted]nippeted");
+
+        assertSnippet(
+                "This is a long text that should be snippeted", 40,
+                "This is a long text that should be snippeted");
+
+        assertSnippet(
+                "This is a long text that should be snippeted", 50,
+                "This is a long text that should be snippeted");
+    }
+}