Unit tests for test command

1 files changed, 129 insertions, 0 deletions

diff --git a/client/go/cmd/test_test.go b/client/go/cmd/test_test.go
new file mode 100644
index 00000000000..73ea13331c7
--- /dev/null
+++ b/client/go/cmd/test_test.go
@@ -0,0 +1,129 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// test command tests
+// Author: jonmv
+
+package cmd
+
+import (
+	"github.com/vespa-engine/vespa/client/go/util"
+	"github.com/vespa-engine/vespa/client/go/vespa"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSuite(t *testing.T) {
+	client := &mockHttpClient{}
+	searchResponse, _ := ioutil.ReadFile("testdata/tests/response.json")
+	client.NextStatus(200)
+	client.NextStatus(200)
+	for i := 0; i < 9; i++ {
+		client.NextResponse(200, string(searchResponse))
+	}
+
+	expectedBytes, _ := ioutil.ReadFile("testdata/tests/expected-suite.out")
+	outBytes, errBytes := execute(command{args: []string{"test", "testdata/tests/system-test"}}, t, client)
+	assert.Equal(t, string(expectedBytes), outBytes)
+	assert.Equal(t, "", errBytes)
+
+	baseUrl := "http://127.0.0.1:8080"
+	urlWithQuery := baseUrl + "/search/?presentation.timing=true&query=artist%3A+foo&timeout=3.4s"
+	requests := []*http.Request{createFeedRequest(baseUrl), createFeedRequest(baseUrl), createSearchRequest(urlWithQuery), createSearchRequest(urlWithQuery)}
+	for i := 0; i < 7; i++ {
+		requests = append(requests, createSearchRequest(baseUrl+"/search/"))
+	}
+	assertRequests(requests, client, t)
+}
+
+func TestTestWithoutAssertions(t *testing.T) {
+	client := &mockHttpClient{}
+	_, errBytes := execute(command{args: []string{"test", "testdata/tests/system-test/foo/query.json"}}, t, client)
+	assert.Equal(t, "a test must have at least one assertion, but none were found in 'testdata/tests/system-test/foo/query.json'\n", errBytes)
+}
+
+func TestSingleTest(t *testing.T) {
+	client := &mockHttpClient{}
+	searchResponse, _ := ioutil.ReadFile("testdata/tests/response.json")
+	client.NextStatus(200)
+	client.NextStatus(200)
+	client.NextResponse(200, string(searchResponse))
+	client.NextResponse(200, string(searchResponse))
+
+	expectedBytes, _ := ioutil.ReadFile("testdata/tests/expected.out")
+	outBytes, errBytes := execute(command{args: []string{"test", "testdata/tests/system-test/test.json"}}, t, client)
+	assert.Equal(t, string(expectedBytes), outBytes)
+	assert.Equal(t, "", errBytes)
+
+	baseUrl := "http://127.0.0.1:8080"
+	rawUrl := baseUrl + "/search/?presentation.timing=true&query=artist%3A+foo&timeout=3.4s"
+	assertRequests([]*http.Request{createFeedRequest(baseUrl), createFeedRequest(baseUrl), createSearchRequest(rawUrl), createSearchRequest(rawUrl)}, client, t)
+}
+
+func TestSingleTestWithCloudAndEndpoints(t *testing.T) {
+	cmd := command{args: []string{"test", "testdata/tests/system-test/test.json", "-t", "cloud", "-a", "t.a.i"}}
+	cmd.homeDir = filepath.Join(t.TempDir(), ".vespa")
+	os.MkdirAll(cmd.homeDir, 0700)
+	keyFile := filepath.Join(cmd.homeDir, "key")
+	certFile := filepath.Join(cmd.homeDir, "cert")
+
+	os.Setenv("VESPA_CLI_DATA_PLANE_KEY_FILE", keyFile)
+	os.Setenv("VESPA_CLI_DATA_PLANE_CERT_FILE", certFile)
+	os.Setenv("VESPA_CLI_ENDPOINTS", "{\"endpoints\":[{\"cluster\":\"container\",\"url\":\"https://url\"}]}")
+
+	kp, _ := vespa.CreateKeyPair()
+	ioutil.WriteFile(keyFile, kp.PrivateKey, 0600)
+	ioutil.WriteFile(certFile, kp.Certificate, 0600)
+
+	client := &mockHttpClient{}
+	searchResponse, _ := ioutil.ReadFile("testdata/tests/response.json")
+	client.NextStatus(200)
+	client.NextStatus(200)
+	client.NextResponse(200, string(searchResponse))
+	client.NextResponse(200, string(searchResponse))
+
+	expectedBytes, _ := ioutil.ReadFile("testdata/tests/expected.out")
+	outBytes, errBytes := execute(cmd, t, client)
+	assert.Equal(t, string(expectedBytes), outBytes)
+	assert.Equal(t, "", errBytes)
+
+	baseUrl := "https://url"
+	rawUrl := baseUrl + "/search/?presentation.timing=true&query=artist%3A+foo&timeout=3.4s"
+	assertRequests([]*http.Request{createFeedRequest(baseUrl), createFeedRequest(baseUrl), createSearchRequest(rawUrl), createSearchRequest(rawUrl)}, client, t)
+}
+
+func createFeedRequest(urlPrefix string) *http.Request {
+	return createRequest("POST",
+		urlPrefix+"/document/v1/test/music/docid/doc?timeout=3.4s",
+		"{\"fields\":{\"artist\":\"Foo Fighters\"}}")
+}
+
+func createSearchRequest(rawUrl string) *http.Request {
+	return createRequest("GET", rawUrl, "")
+}
+
+func createRequest(method string, uri string, body string) *http.Request {
+	requestUrl, _ := url.ParseRequestURI(uri)
+	return &http.Request{
+		URL:    requestUrl,
+		Method: method,
+		Header: nil,
+		Body:   ioutil.NopCloser(strings.NewReader(body)),
+	}
+}
+
+func assertRequests(requests []*http.Request, client *mockHttpClient, t *testing.T) {
+	if assert.Equal(t, len(requests), len(client.requests)) {
+		for i, e := range requests {
+			a := client.requests[i]
+			assert.Equal(t, e.URL.String(), a.URL.String())
+			assert.Equal(t, e.Method, a.Method)
+			assert.Equal(t, util.ReaderToJSON(e.Body), util.ReaderToJSON(a.Body))
+		}
+	}
+}