1 files changed, 78 insertions, 0 deletions
diff --git a/client/go/cmd/deploy_test.go b/client/go/cmd/deploy_test.go
new file mode 100644
index 00000000000..64ede50546c
--- /dev/null
+++ b/client/go/cmd/deploy_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.
+// deploy command tests
+// Author: bratseth
+
+package cmd
+
+import (
+    "github.com/stretchr/testify/assert"
+    "strconv"
+    "testing"
+)
+
+func TestDeployZip(t *testing.T) {
+    client := &mockHttpClient{}
+	assert.Equal(t,
+	             "\x1b[32mSuccess\n",
+	             executeCommand(t, client, []string{"deploy", "activate", "testdata/application.zip"}, []string{}))
+	assertDeployRequestMade("http://127.0.0.1:19071", client, t)
+}
+
+func TestDeployZipWithURLTargetArgument(t *testing.T) {
+    client := &mockHttpClient{}
+	assert.Equal(t,
+	             "\x1b[32mSuccess\n",
+	             executeCommand(t, client, []string{"deploy", "activate", "testdata/application.zip", "-t", "http://target:19071"}, []string{}))
+	assertDeployRequestMade("http://target:19071", client, t)
+}
+
+func TestDeployZipWitLocalTargetArgument(t *testing.T) {
+    client := &mockHttpClient{}
+	assert.Equal(t,
+	             "\x1b[32mSuccess\n",
+	             executeCommand(t, client, []string{"deploy", "activate", "testdata/application.zip", "-t", "local"}, []string{}))
+	assertDeployRequestMade("http://127.0.0.1:19071", client, t)
+}
+
+func TestDeployDirectory(t *testing.T) {
+    client := &mockHttpClient{}
+	assert.Equal(t,
+	             "\x1b[32mSuccess\n",
+	             executeCommand(t, client, []string{"deploy", "activate", "testdata/src/main/application"}, []string{}))
+	assertDeployRequestMade("http://127.0.0.1:19071", client, t)
+}
+
+func TestDeployApplicationPackageError(t *testing.T) {
+    assertApplicationPackageError(t, 401, "Application package error")
+}
+
+func TestDeployError(t *testing.T) {
+    assertDeployServerError(t, 501, "Deploy service error")
+}
+
+// TODO: Test prepare and activate prepared
+
+func assertDeployRequestMade(target string, client *mockHttpClient, t *testing.T) {
+    assert.Equal(t, target + "/application/v2/tenant/default/prepareandactivate", client.lastRequest.URL.String())
+    assert.Equal(t, "application/zip", client.lastRequest.Header.Get("Content-Type"))
+    assert.Equal(t, "POST", client.lastRequest.Method)
+    var body = client.lastRequest.Body
+    assert.NotNil(t, body)
+    buf := make([]byte, 7) // Just check the first few bytes
+    body.Read(buf)
+    assert.Equal(t, "PK\x03\x04\x14\x00\b", string(buf))
+}
+
+func assertApplicationPackageError(t *testing.T, status int, errorMessage string) {
+    client := &mockHttpClient{ nextStatus: status, nextBody: errorMessage, }
+	assert.Equal(t,
+	             "\x1b[31mInvalid application package (Status " + strconv.Itoa(status) + "):\n" + errorMessage + "\n",
+	             executeCommand(t, client, []string{"deploy", "activate", "testdata/src/main/application"}, []string{}))
+}
+
+func assertDeployServerError(t *testing.T, status int, errorMessage string) {
+    client := &mockHttpClient{ nextStatus: status, nextBody: errorMessage, }
+	assert.Equal(t,
+	             "\x1b[31mError from deploy service at 127.0.0.1:19071 (Status " + strconv.Itoa(status) + "):\n" + errorMessage + "\n",
+	             executeCommand(t, client, []string{"deploy", "activate", "testdata/src/main/application"}, []string{}))
+}