test(label-cmd): add Label cmd test

Author: zinuzoid

cmd/apps_test.go

@@ -165,6 +165,20 @@ func TestAppsInfo(t *testing.T) {
 }`)
 	})
 
+	server.Mux.HandleFunc("/v2/apps/lorem-ipsum/settings/", func(w http.ResponseWriter, r *http.Request) {
+		testutil.SetHeaders(w)
+		fmt.Fprintf(w, `{
+			"owner": "elrond",
+			"app": "lorem-ipsum",
+			"created": "2014-01-01T00:00:00UTC",
+			"updated": "2014-01-01T00:00:00UTC",
+			"uuid": "de1bf5b5-4a72-4f94-a10c-d2a3741cdf75",
+			"label": {
+				"team": "frontend"
+			}
+		}`)
+	})
+
 	s, err := settings.Load(cmdr.ConfigFile)
 	if err != nil {
 		t.Fatal(err)
@@ -196,6 +210,9 @@ lorem-ipsum-cmd-1911796442-48b58 up (v2)
 === lorem-ipsum Domains
 lorem-ipsum
 
+=== lorem-ipsum Label
+team:            frontend
+
 `, "output")
 }
 

cmd/labels_test.go

@@ -0,0 +1,113 @@
+package cmd
+
+import (
+	"bytes"
+	"fmt"
+	"net/http"
+	"testing"
+
+	"github.com/arschles/assert"
+	"github.com/deis/controller-sdk-go/api"
+	"github.com/deis/workflow-cli/pkg/testutil"
+	"strings"
+)
+
+func TestLabelsList(t *testing.T) {
+	t.Parallel()
+	cf, server, err := testutil.NewTestServerAndClient()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer server.Close()
+	var b bytes.Buffer
+	cmdr := DeisCmd{WOut: &b, ConfigFile: cf}
+
+	server.Mux.HandleFunc("/v2/apps/rivendell/settings/", func(w http.ResponseWriter, r *http.Request) {
+		testutil.SetHeaders(w)
+		fmt.Fprintf(w, `{
+			"owner": "jim",
+			"app": "rivendell",
+		    "label": {"git_repo": "https://github.com/deis/controller-sdk-go", "team" : "deis"},
+			"created": "2014-01-01T00:00:00UTC",
+			"updated": "2014-01-01T00:00:00UTC",
+			"uuid": "de1bf5b5-4a72-4f94-a10c-d2a3741cdf75"
+		}`)
+	})
+
+	err = cmdr.LabelsList("rivendell")
+	assert.NoErr(t, err)
+	assert.Equal(t, strings.TrimSpace(b.String()), `=== rivendell Label
+git_repo:        https://github.com/deis/controller-sdk-go
+team:            deis`, "output")
+
+	server.Mux.HandleFunc("/v2/apps/mordor/settings/", func(w http.ResponseWriter, r *http.Request) {
+		testutil.SetHeaders(w)
+		fmt.Fprintf(w, `{
+			"owner": "priw",
+			"app": "mordor",
+			"created": "2014-01-01T00:00:00UTC",
+			"updated": "2014-01-01T00:00:00UTC",
+			"uuid": "de1bf5b5-4a72-4f94-a10c-d2a3741cdf75"
+		}`)
+	})
+	b.Reset()
+
+	err = cmdr.LabelsList("mordor")
+	assert.NoErr(t, err)
+	assert.Equal(t, b.String(), "=== mordor Label\nNo labels found.\n", "output")
+}
+
+func TestListsSet(t *testing.T) {
+	t.Parallel()
+	cf, server, err := testutil.NewTestServerAndClient()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer server.Close()
+	var b bytes.Buffer
+	cmdr := DeisCmd{WOut: &b, ConfigFile: cf}
+
+	server.Mux.HandleFunc("/v2/apps/lothlorien/settings/", func(w http.ResponseWriter, r *http.Request) {
+		testutil.SetHeaders(w)
+		data := map[string]interface{}{
+			"git_repo": "https://github.com/deis/controller-sdk-go",
+			"team":     "deis",
+		}
+		testutil.AssertBody(t, api.AppSettings{Label: data}, r)
+		fmt.Fprintf(w, "{}")
+	})
+
+	err = cmdr.LabelsSet("lothlorien", []string{
+		"team=deis",
+		"git_repo=https://github.com/deis/controller-sdk-go",
+	})
+	assert.NoErr(t, err)
+	assert.Equal(t, testutil.StripProgress(b.String()), "Applying labels on lothlorien... done\n", "output")
+}
+
+func TestListsUnset(t *testing.T) {
+	t.Parallel()
+	cf, server, err := testutil.NewTestServerAndClient()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer server.Close()
+	var b bytes.Buffer
+	cmdr := DeisCmd{WOut: &b, ConfigFile: cf}
+
+	server.Mux.HandleFunc("/v2/apps/bree/settings/", func(w http.ResponseWriter, r *http.Request) {
+		testutil.SetHeaders(w)
+		testutil.AssertBody(t, api.AppSettings{Label: map[string]interface{}{
+			"team":     nil,
+			"git_repo": nil,
+		}}, r)
+		fmt.Fprintf(w, "{}")
+	})
+
+	err = cmdr.LabelsUnset("bree", []string{
+		"team",
+		"git_repo",
+	})
+	assert.NoErr(t, err)
+	assert.Equal(t, testutil.StripProgress(b.String()), "Removing labels on bree... done\n", "output")
+}

parser/labels_test.go

@@ -0,0 +1,74 @@
+package parser
+
+import (
+	"bytes"
+	"errors"
+	"testing"
+
+	"github.com/arschles/assert"
+	"github.com/deis/workflow-cli/pkg/testutil"
+)
+
+// Create fake implementations of each method that return the argument
+// we expect to have called the function (as an error to satisfy the interface).
+
+func (d FakeDeisCmd) LabelsList(string) error {
+	return errors.New("labels:list")
+}
+
+func (d FakeDeisCmd) LabelsSet(string, []string) error {
+	return errors.New("labels:set")
+}
+
+func (d FakeDeisCmd) LabelsUnset(string, []string) error {
+	return errors.New("labels:unset")
+}
+
+func TestLabels(t *testing.T) {
+	t.Parallel()
+
+	cf, server, err := testutil.NewTestServerAndClient()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer server.Close()
+	var b bytes.Buffer
+	cmdr := FakeDeisCmd{WOut: &b, ConfigFile: cf}
+
+	// cases defines the arguments and expected return of the call.
+	// if expected is "", it defaults to args[0].
+	cases := []struct {
+		args     []string
+		expected string
+	}{
+		{
+			args:     []string{"labels:list"},
+			expected: "",
+		},
+		{
+			args:     []string{"labels:set", "git_repo=https://github.com/deis/workflow", "team=deis"},
+			expected: "",
+		},
+		{
+			args:     []string{"labels:unset", "git_repo", "team"},
+			expected: "",
+		},
+		{
+			args:     []string{"labels"},
+			expected: "labels:list",
+		},
+	}
+
+	// For each case, check that calling the route with the arguments
+	// returns the expected error, which is args[0] if not provided.
+	for _, c := range cases {
+		var expected string
+		if c.expected == "" {
+			expected = c.args[0]
+		} else {
+			expected = c.expected
+		}
+		err = Labels(c.args, cmdr)
+		assert.Err(t, errors.New(expected), err)
+	}
+}