(feat): Add test coverage for Action.DefaultExecutor()

cmd/ebs-bootstrap.go

@@ -35,7 +35,7 @@ func main() {
 	fb := backend.NewLinuxFileBackend(ufs)
 	ub := backend.NewLinuxOwnerBackend(uos)
 	dmb := backend.NewLinuxDeviceMetricsBackend(lds, fssf)
-	dae := action.NewDefaultActionExecutor(c)
+	dae := action.NewDefaultActionExecutor()
 
 	// Modify Config
 	modifiers := []config.Modifier{

internal/action/action.go

@@ -5,7 +5,6 @@ import (
 	"log"
 	"strings"
 
-	"github.com/reecetech/ebs-bootstrap/internal/config"
 	"github.com/reecetech/ebs-bootstrap/internal/model"
 )
 
@@ -23,13 +22,11 @@ type ActionExecutor interface {
 }
 
 type DefaultActionExecutor struct {
-	config *config.Config
-	read   func(buffer *string) error
+	read func(buffer *string) error
 }
 
-func NewDefaultActionExecutor(c *config.Config) *DefaultActionExecutor {
+func NewDefaultActionExecutor() *DefaultActionExecutor {
 	return &DefaultActionExecutor{
-		config: c,
 		read: func(buffer *string) error {
 			_, err := fmt.Scanln(buffer)
 			return err

internal/action/action_test.go

@@ -0,0 +1,129 @@
+package action
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/reecetech/ebs-bootstrap/internal/model"
+	"github.com/reecetech/ebs-bootstrap/internal/utils"
+)
+
+type MockAction struct {
+	execute func() error
+	mode    model.Mode
+}
+
+func (ma *MockAction) Execute() error {
+	return ma.execute()
+}
+
+func (ma *MockAction) Success() string {
+	return "Successfully executed action"
+}
+
+func (ma *MockAction) Prompt() string {
+	return "Would you like to execute action"
+}
+
+func (ma *MockAction) Refuse() string {
+	return "Refused to execute action"
+}
+
+func (ma *MockAction) GetMode() model.Mode {
+	return ma.mode
+}
+
+func (ma *MockAction) SetMode(mode model.Mode) Action {
+	ma.mode = mode
+	return ma
+}
+
+func TestDefaultActionExecutor(t *testing.T) {
+	var input = func(i string) func(buffer *string) error {
+		return func(buffer *string) error {
+			*buffer = i
+			return nil
+		}
+	}
+	var fail = func(err string) func(buffer *string) error {
+		return func(buffer *string) error {
+			return fmt.Errorf(err)
+		}
+	}
+
+	subtests := []struct {
+		Name        string
+		Read        func(buffer *string) error
+		Error       error
+		Mode        model.Mode
+		ExpectedErr error
+	}{
+		{
+			Name:        "Mode=Empty + Read=Disabled + Action=Success",
+			Read:        fail("🔴 Standard Input Disabled"),
+			Error:       nil,
+			Mode:        model.Empty,
+			ExpectedErr: fmt.Errorf("🔴 Unsupported mode was encountered. Refused to execute action"),
+		},
+		{
+			Name:        "Mode=Healthcheck + Read=Avoid + Action=Success",
+			Read:        fail("🔴 Standard Input Disabled"),
+			Error:       nil,
+			Mode:        model.Healthcheck,
+			ExpectedErr: fmt.Errorf("🔴 Healthcheck mode enabled. Refused to execute action"),
+		},
+		{
+			Name:        "Mode=Prompt + Read=Input<y> + Action=Success",
+			Read:        input("y"),
+			Error:       nil,
+			Mode:        model.Prompt,
+			ExpectedErr: nil,
+		},
+		{
+			Name:        "Mode=Prompt + Read=Input<yes> + Action=Success",
+			Read:        input("yes"),
+			Error:       nil,
+			Mode:        model.Prompt,
+			ExpectedErr: nil,
+		},
+		{
+			Name:        "Mode=Prompt + Read=Failure + Action=Success",
+			Read:        fail("🔴 Failed to Read From Standard Input"),
+			Error:       nil,
+			Mode:        model.Prompt,
+			ExpectedErr: fmt.Errorf("🔴 Action rejected. Refused to execute action"),
+		},
+		{
+			Name:        "Mode=Prompt + Read=Input<n> + Action=Success",
+			Read:        input("n"),
+			Error:       nil,
+			Mode:        model.Prompt,
+			ExpectedErr: fmt.Errorf("🔴 Action rejected. Refused to execute action"),
+		},
+		{
+			Name:        "Mode=Force + Read=Disabled + Action=Success",
+			Read:        fail("🔴 Standard Input Disabled"),
+			Error:       nil,
+			Mode:        model.Force,
+			ExpectedErr: nil,
+		},
+		{
+			Name:        "Mode=Force + Read=Disabled + Action=Failure",
+			Read:        fail("🔴 Standard Input Disabled"),
+			Error:       fmt.Errorf("🔴 Error encountered while executing action"),
+			Mode:        model.Force,
+			ExpectedErr: fmt.Errorf("🔴 Error encountered while executing action"),
+		},
+	}
+	for _, subtest := range subtests {
+		dae := &DefaultActionExecutor{
+			read: subtest.Read,
+		}
+		a := &MockAction{
+			execute: func() error { return subtest.Error },
+			mode:    subtest.Mode,
+		}
+		err := dae.Execute([]Action{a})
+		utils.CheckError("dae.Execute()", t, subtest.ExpectedErr, err)
+	}
+}