1 files changed, 195 insertions, 0 deletions
diff --git a/vespaclient-java/src/test/java/com/yahoo/vespaget/CommandLineOptionsTest.java b/vespaclient-java/src/test/java/com/yahoo/vespaget/CommandLineOptionsTest.java
new file mode 100644
index 00000000000..3e707b04256
--- /dev/null
+++ b/vespaclient-java/src/test/java/com/yahoo/vespaget/CommandLineOptionsTest.java
@@ -0,0 +1,195 @@
+// Copyright 2017 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespaget;
+
+import com.yahoo.documentapi.messagebus.protocol.DocumentProtocol;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+
+import java.io.*;
+import java.util.Iterator;
+
+import static org.junit.Assert.*;
+
+/**
+ * Test class for {@link CommandLineOptions}
+ *
+ * @author bjorncs
+ * @since 5.26
+ */
+public class CommandLineOptionsTest {
+
+    private final InputStream emptyStream = new InputStream() {
+
+        @Override
+        public int read() throws IOException {
+            return -1;
+        }
+    };
+
+    @Rule
+    public final ExpectedException exception = ExpectedException.none();
+
+    private ClientParameters getParsedOptions(InputStream in, String... args) {
+        CommandLineOptions options = new CommandLineOptions(in);
+        return options.parseCommandLineArguments(args);
+    }
+
+    private ClientParameters getParsedOptions(String... args) {
+        return getParsedOptions(emptyStream, args);
+    }
+
+    @Test
+    public void testDefaultOptions() {
+        ClientParameters params = getParsedOptions();
+        assertFalse(params.help);
+        assertFalse(params.documentIds.hasNext());
+        assertFalse(params.printIdsOnly);
+        assertEquals("[all]", params.fieldSet);
+        assertEquals("default", params.route);
+        assertTrue(params.cluster.isEmpty());
+        assertEquals("client", params.configId);
+        assertFalse(params.showDocSize);
+        assertEquals(0, params.timeout, 0);
+        assertFalse(params.noRetry);
+        assertEquals(0, params.traceLevel);
+        assertEquals(DocumentProtocol.Priority.NORMAL_2, params.priority);
+        assertTrue(params.loadTypeName.isEmpty());
+    }
+
+    @Test
+    public void testValidOptions() {
+        ClientParameters params = getParsedOptions(
+                "--fieldset", "[fieldset]",
+                "--route", "dummyroute",
+                "--configid", "dummyconfig",
+                "--showdocsize",
+                "--timeout", "0.25",
+                "--noretry",
+                "--trace", "1",
+                "--priority", Integer.toString(DocumentProtocol.Priority.HIGH_3.getValue()),
+                "--loadtype", "dummyloadtype",
+                "id:1", "id:2"
+        );
+
+        assertEquals("[fieldset]", params.fieldSet);
+        assertEquals("dummyroute", params.route);
+        assertEquals("dummyconfig", params.configId);
+        assertTrue(params.showDocSize);
+        assertEquals(0.25, params.timeout, 0.0001);
+        assertTrue(params.noRetry);
+        assertEquals(1, params.traceLevel);
+        assertEquals(DocumentProtocol.Priority.HIGH_3, params.priority);
+        assertEquals("dummyloadtype", params.loadTypeName);
+
+        Iterator<String> documentsIds = params.documentIds;
+        assertEquals("id:1", documentsIds.next());
+        assertEquals("id:2", documentsIds.next());
+        assertFalse(documentsIds.hasNext());
+    }
+
+    @Test
+    public void testInvalidCombination1() {
+        exception.expect(IllegalArgumentException.class);
+        exception.expectMessage("Print ids and headers only options are mutually exclusive.");
+        getParsedOptions("--headersonly", "--printids");
+    }
+
+    @Test
+    public void testInvalidCombination2() {
+        exception.expect(IllegalArgumentException.class);
+        exception.expectMessage("Field set option can not be used in combination with print ids or headers only options.");
+        getParsedOptions("--headersonly", "--fieldset", "[header]");
+    }
+
+    @Test
+    public void testInvalidCombination3() {
+        exception.expect(IllegalArgumentException.class);
+        exception.expectMessage("Field set option can not be used in combination with print ids or headers only options.");
+        getParsedOptions("--printids", "--fieldset", "[header]");
+    }
+
+    @Test
+    public void testInvalidCombination4() {
+        exception.expect(IllegalArgumentException.class);
+        exception.expectMessage("Cluster and route options are mutually exclusive.");
+        getParsedOptions("--route", "dummyroute", "--cluster", "dummycluster");
+    }
+
+    @Test
+    public void testInvalidPriority() {
+        exception.expect(IllegalArgumentException.class);
+        exception.expectMessage("Invalid priority: 16");
+        getParsedOptions("--priority", "16");
+    }
+
+    @Test
+    public void testInvalidTraceLevel1() {
+        exception.expect(IllegalArgumentException.class);
+        exception.expectMessage("Invalid tracelevel: -1");
+        getParsedOptions("--trace", "-1");
+    }
+
+    @Test
+    public void testInvalidTraceLevel2() {
+        exception.expect(IllegalArgumentException.class);
+        exception.expectMessage("Invalid tracelevel: 10");
+        getParsedOptions("--trace", "10");
+    }
+
+    @Test
+    public void testPrintids() {
+        ClientParameters params = getParsedOptions("--printids");
+        assertEquals("[id]", params.fieldSet);
+    }
+
+    @Test
+    public void testHeadersOnly() {
+        ClientParameters params = getParsedOptions("--headersonly");
+        assertEquals("[header]", params.fieldSet);
+    }
+
+    @Test
+    public void testCluster() {
+        ClientParameters params = getParsedOptions("--cluster", "dummycluster");
+        assertEquals("dummycluster", params.cluster);
+        assertTrue(params.route.isEmpty());
+    }
+
+    @Test
+    public void testHelp() {
+        ClientParameters params = getParsedOptions("--help");
+        assertTrue(params.help);
+    }
+
+    @Test
+    public void testDocumentIdsFromInputStream() throws UnsupportedEncodingException {
+        InputStream in = new ByteArrayInputStream("id:1 id:2 id:3".getBytes("UTF-8"));
+        ClientParameters params = getParsedOptions(in, "");
+
+        Iterator<String> documentsIds = params.documentIds;
+        assertEquals("id:1", documentsIds.next());
+        assertEquals("id:2", documentsIds.next());
+        assertEquals("id:3", documentsIds.next());
+        assertFalse(documentsIds.hasNext());
+    }
+
+    @Test
+    public void testPrintHelp() {
+        ByteArrayOutputStream outContent = new ByteArrayOutputStream();
+        PrintStream oldOut = System.out;
+        System.setOut(new PrintStream(outContent));
+        try {
+            CommandLineOptions options = new CommandLineOptions(emptyStream);
+            options.printHelp();
+
+            String output = outContent.toString();
+            assertTrue(output.contains("vespaget <options> [documentid...]"));
+            assertTrue(output.contains("Fetch a document from a Vespa Content cluster."));
+        } finally {
+            System.setOut(oldOut);
+            outContent.reset();
+
+        }
+    }
+}