1 files changed, 78 insertions, 0 deletions
diff --git a/client/go/cmd/query_test.go b/client/go/cmd/query_test.go
new file mode 100644
index 00000000000..f55956bc1ce
--- /dev/null
+++ b/client/go/cmd/query_test.go
@@ -0,0 +1,78 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// query command tests
+// Author: bratseth
+
+package cmd
+
+import (
+    "github.com/stretchr/testify/assert"
+    "strconv"
+    "testing"
+)
+
+func TestQuery(t *testing.T) {
+    assertQuery(t,
+                "?yql=select+from+sources+%2A+where+title+contains+%27foo%27",
+                "select from sources * where title contains 'foo'")
+}
+
+func TestQueryNonJsonResult(t *testing.T) {
+    assertQuery(t,
+                "?yql=select+from+sources+%2A+where+title+contains+%27foo%27",
+                "select from sources * where title contains 'foo'")
+}
+
+func TestQueryWithMultipleParameters(t *testing.T) {
+    assertQuery(t,
+                "?hits=5&yql=select+from+sources+%2A+where+title+contains+%27foo%27",
+                "select from sources * where title contains 'foo'", "hits=5")
+}
+
+func TestQueryWithExplicitYqlParameter(t *testing.T) {
+    assertQuery(t,
+                "?yql=select+from+sources+%2A+where+title+contains+%27foo%27",
+                "yql=select from sources * where title contains 'foo'")
+}
+
+func TestIllegalQuery(t *testing.T) {
+    assertQueryError(t, 401, "query error message")
+}
+
+func TestServerError(t *testing.T) {
+    assertQueryServiceError(t, 501, "server error message")
+}
+
+func assertQuery(t *testing.T, expectedQuery string, query ...string) {
+    client := &mockHttpClient{ nextBody: "{\"query\":\"result\"}", }
+	assert.Equal(t,
+	             "{\n    \"query\": \"result\"\n}\n",
+	             executeCommand(t, client, []string{"query"}, query),
+	             "query output")
+    assert.Equal(t, getTarget(queryContext).query + "/search/" + expectedQuery, client.lastRequest.URL.String())
+}
+
+func assertQueryNonJsonResult(t *testing.T, expectedQuery string, query ...string) {
+    client := &mockHttpClient{ nextBody: "query result", }
+	assert.Equal(t,
+	             "query result\n",
+	             executeCommand(t, client, []string{"query"}, query),
+	             "query output")
+    assert.Equal(t, getTarget(queryContext).query + "/search/" + expectedQuery, client.lastRequest.URL.String())
+}
+
+func assertQueryError(t *testing.T, status int, errorMessage string) {
+    client := &mockHttpClient{ nextStatus: status, nextBody: errorMessage, }
+	assert.Equal(t,
+	             "\x1b[31mInvalid query (Status " + strconv.Itoa(status) + "):\n" + errorMessage + "\n",
+	             executeCommand(t, client, []string{"query"}, []string{"yql=select from sources * where title contains 'foo'"}),
+	             "error output")
+}
+
+func assertQueryServiceError(t *testing.T, status int, errorMessage string) {
+    client := &mockHttpClient{ nextStatus: status, nextBody: errorMessage, }
+	assert.Equal(t,
+	             "\x1b[31mError from container at 127.0.0.1:8080 (Status " + strconv.Itoa(status) + "):\n" + errorMessage + "\n",
+	             executeCommand(t, client, []string{"query"}, []string{"yql=select from sources * where title contains 'foo'"}),
+	             "error output")
+}
+