From 8ffca97596708927e6a1db202c3b81914128d96c Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Wed, 16 May 2018 08:41:12 -0700
Subject: [PATCH] command/intention/delete: tests

---
 command/intention/delete/delete_test.go | 99 +++++++++++++++++++++++++
 1 file changed, 99 insertions(+)
 create mode 100644 command/intention/delete/delete_test.go

diff --git a/command/intention/delete/delete_test.go b/command/intention/delete/delete_test.go
new file mode 100644
index 0000000000..c5674f7715
--- /dev/null
+++ b/command/intention/delete/delete_test.go
@@ -0,0 +1,99 @@
+package delete
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/hashicorp/consul/agent"
+	"github.com/hashicorp/consul/api"
+	"github.com/mitchellh/cli"
+	"github.com/stretchr/testify/require"
+)
+
+func TestCommand_noTabs(t *testing.T) {
+	t.Parallel()
+	if strings.ContainsRune(New(nil).Help(), '\t') {
+		t.Fatal("help has tabs")
+	}
+}
+
+func TestCommand_Validation(t *testing.T) {
+	t.Parallel()
+
+	ui := cli.NewMockUi()
+	c := New(ui)
+
+	cases := map[string]struct {
+		args   []string
+		output string
+	}{
+		"0 args": {
+			[]string{},
+			"requires exactly 1 or 2",
+		},
+
+		"3 args": {
+			[]string{"a", "b", "c"},
+			"requires exactly 1 or 2",
+		},
+	}
+
+	for name, tc := range cases {
+		t.Run(name, func(t *testing.T) {
+			require := require.New(t)
+
+			c.init()
+
+			// Ensure our buffer is always clear
+			if ui.ErrorWriter != nil {
+				ui.ErrorWriter.Reset()
+			}
+			if ui.OutputWriter != nil {
+				ui.OutputWriter.Reset()
+			}
+
+			require.Equal(1, c.Run(tc.args))
+			output := ui.ErrorWriter.String()
+			require.Contains(output, tc.output)
+		})
+	}
+}
+
+func TestCommand(t *testing.T) {
+	t.Parallel()
+
+	require := require.New(t)
+	a := agent.NewTestAgent(t.Name(), ``)
+	defer a.Shutdown()
+	client := a.Client()
+
+	// Create the intention
+	{
+		_, _, err := client.Connect().IntentionCreate(&api.Intention{
+			SourceName:      "web",
+			DestinationName: "db",
+			Action:          api.IntentionActionDeny,
+		}, nil)
+		require.NoError(err)
+	}
+
+	// Delete it
+	{
+		ui := cli.NewMockUi()
+		c := New(ui)
+
+		args := []string{
+			"-http-addr=" + a.HTTPAddr(),
+			"web", "db",
+		}
+		require.Equal(0, c.Run(args), ui.ErrorWriter.String())
+		require.Contains(ui.OutputWriter.String(), "deleted")
+	}
+
+	// Find it (should be gone)
+	{
+		ixns, _, err := client.Connect().Intentions(nil)
+		require.NoError(err)
+		require.Len(ixns, 0)
+	}
+}