Add tests to cmd package

Author: antham

cmd/create_test.go

@@ -0,0 +1,80 @@
+package cmd
+
+import (
+	"bytes"
+	"errors"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewCreateCmd(t *testing.T) {
+	type scenario struct {
+		name  string
+		setup func(*testing.T) (workspaceManager, []string)
+		test  func(*testing.T, error)
+	}
+	scenarios := []scenario{
+		{
+			"An error occurred when creating a workspace",
+			func(t *testing.T) (workspaceManager, []string) {
+				w := newMockWorkspaceManager(t)
+				args := []string{"api"}
+				w.Mock.On("Create", args[0]).Return(errors.New("an error occurred"))
+				return w, args
+			},
+			func(t *testing.T, err error) {
+				assert.Error(t, err)
+			},
+		},
+		{
+			"An error occurred when creating a workspace env",
+			func(t *testing.T) (workspaceManager, []string) {
+				w := newMockWorkspaceManager(t)
+				args := []string{"api", "prod"}
+				w.Mock.On("CreateEnv", args[0], args[1]).Return(errors.New("an error occurred"))
+				return w, args
+			},
+			func(t *testing.T, err error) {
+				assert.Error(t, err)
+			},
+		},
+		{
+			"Creating a workspace successfully",
+			func(t *testing.T) (workspaceManager, []string) {
+				w := newMockWorkspaceManager(t)
+				args := []string{"api"}
+				w.Mock.On("Create", args[0]).Return(nil)
+				return w, args
+			},
+			func(t *testing.T, err error) {
+				assert.NoError(t, err)
+			},
+		},
+		{
+			"Creating a workspace env successfully",
+			func(t *testing.T) (workspaceManager, []string) {
+				w := newMockWorkspaceManager(t)
+				args := []string{"api", "prod"}
+				w.Mock.On("CreateEnv", args[0], args[1]).Return(nil)
+				return w, args
+			},
+			func(t *testing.T, err error) {
+				assert.NoError(t, err)
+			},
+		},
+	}
+	for _, s := range scenarios {
+		t.Run(s.name, func(t *testing.T) {
+			os.Setenv("EDITOR", "emacs")
+			os.Setenv("SHELL", "/bin/sh")
+			w, args := s.setup(t)
+			cmd := newCreateCmd(w)
+			cmd.SetArgs(args)
+			cmd.SetErr(&bytes.Buffer{})
+			cmd.SetOut(&bytes.Buffer{})
+			s.test(t, cmd.Execute())
+		})
+	}
+}

cmd/remove_test.go

@@ -0,0 +1,56 @@
+package cmd
+
+import (
+	"bytes"
+	"errors"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewRemoveCmd(t *testing.T) {
+	type scenario struct {
+		name  string
+		setup func(*testing.T) (workspaceManager, []string)
+		test  func(*testing.T, error)
+	}
+	scenarios := []scenario{
+		{
+			"An error occurred when removing a workspace",
+			func(t *testing.T) (workspaceManager, []string) {
+				w := newMockWorkspaceManager(t)
+				args := []string{"api"}
+				w.Mock.On("Remove", args[0]).Return(errors.New("an error occurred"))
+				return w, args
+			},
+			func(t *testing.T, err error) {
+				assert.Error(t, err)
+			},
+		},
+		{
+			"Removing a workspace successfully",
+			func(t *testing.T) (workspaceManager, []string) {
+				w := newMockWorkspaceManager(t)
+				args := []string{"api"}
+				w.Mock.On("Remove", args[0]).Return(nil)
+				return w, args
+			},
+			func(t *testing.T, err error) {
+				assert.NoError(t, err)
+			},
+		},
+	}
+	for _, s := range scenarios {
+		t.Run(s.name, func(t *testing.T) {
+			os.Setenv("EDITOR", "emacs")
+			os.Setenv("SHELL", "/bin/sh")
+			w, args := s.setup(t)
+			cmd := newRemoveCmd(w)
+			cmd.SetArgs(args)
+			cmd.SetErr(&bytes.Buffer{})
+			cmd.SetOut(&bytes.Buffer{})
+			s.test(t, cmd.Execute())
+		})
+	}
+}

cmd/run_test.go

@@ -0,0 +1,43 @@
+package cmd
+
+import (
+	"bytes"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewRunCmd(t *testing.T) {
+	type scenario struct {
+		name  string
+		setup func(*testing.T) (workspaceManager, []string)
+		test  func(*testing.T, error)
+	}
+	scenarios := []scenario{
+		{
+			"Running a function successfully",
+			func(t *testing.T) (workspaceManager, []string) {
+				w := newMockWorkspaceManager(t)
+				args := []string{"api", "start"}
+				w.Mock.On("RunFunction", args[0], "", []string{args[1]}).Return(nil)
+				return w, args
+			},
+			func(t *testing.T, err error) {
+				assert.NoError(t, err)
+			},
+		},
+	}
+	for _, s := range scenarios {
+		t.Run(s.name, func(t *testing.T) {
+			os.Setenv("EDITOR", "emacs")
+			os.Setenv("SHELL", "/bin/sh")
+			w, args := s.setup(t)
+			cmd := newRunCmd(w)
+			cmd.SetArgs(args)
+			cmd.SetErr(&bytes.Buffer{})
+			cmd.SetOut(&bytes.Buffer{})
+			s.test(t, cmd.Execute())
+		})
+	}
+}