who let the dogs out?

Author: resoluteCoder

pkg/workceptor/command_test.go

@@ -2,6 +2,7 @@ package workceptor_test
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"os/exec"
 	"sync"
@@ -79,6 +80,100 @@ func TestSetFromParams2(t *testing.T) {
 
 }
 
+func TestUnredactedStatus(t *testing.T) {
+	wu, mockBaseWorkUnit, _, _ := createTestSetup(t)
+	statusLock := &sync.RWMutex{}
+	mockBaseWorkUnit.EXPECT().GetStatusLock().Return(statusLock).Times(2)
+	mockBaseWorkUnit.EXPECT().GetStatusWithoutExtraData().Return(&workceptor.StatusFileData{})
+	mockBaseWorkUnit.EXPECT().GetStatusCopy().Return(workceptor.StatusFileData{
+		ExtraData: &workceptor.CommandExtraData{},
+	})
+
+	wu.UnredactedStatus()
+}
+
+// func TestStart(t *testing.T) {
+// 	wu, mockBaseWorkUnit, _, _ := createTestSetup(t)
+// 	// w, err := workceptor.New(context.Background(), mockNetceptor, "")
+// 	// if err != nil {
+// 	// 	fmt.Println(err)
+// 	// }
+
+// 	stuff := mockBaseWorkUnit.EXPECT().GetWorkceptor().Return()
+// 	// mockNetceptor.EXPECT().GetLogger()
+
+// 	wu.Start()
+
+// }
+
+func TestRestart(t *testing.T) {
+	wu, mockBaseWorkUnit, _, _ := createTestSetup(t)
+
+	restartTestCases := []struct {
+		name          string
+		expectedCalls func()
+		errorCatch    func(error, *testing.T)
+	}{
+		{
+			name: "return error 1",
+			expectedCalls: func() {
+				mockBaseWorkUnit.EXPECT().Load().Return(errors.New("terminated"))
+			},
+			errorCatch: func(err error, t *testing.T) {
+				if err.Error() != "terminated" {
+					t.Error(err)
+				}
+			},
+		},
+		{
+			name: "return nil 1",
+			expectedCalls: func() {
+				statusFile := &workceptor.StatusFileData{State: 2}
+				mockBaseWorkUnit.EXPECT().Load().Return(nil)
+				statusLock := &sync.RWMutex{}
+				mockBaseWorkUnit.EXPECT().GetStatusLock().Return(statusLock).Times(2)
+				mockBaseWorkUnit.EXPECT().GetStatusWithoutExtraData().Return(statusFile)
+				mockBaseWorkUnit.EXPECT().GetStatusCopy().Return(workceptor.StatusFileData{
+					ExtraData: &workceptor.CommandExtraData{},
+				})
+			},
+			errorCatch: func(err error, t *testing.T) {
+				if err != nil {
+					t.Error(err)
+				}
+			},
+		},
+		{
+			name: "return nil 2",
+			expectedCalls: func() {
+				statusFile := &workceptor.StatusFileData{State: 0}
+				mockBaseWorkUnit.EXPECT().Load().Return(nil)
+				statusLock := &sync.RWMutex{}
+				mockBaseWorkUnit.EXPECT().GetStatusLock().Return(statusLock).Times(2)
+				mockBaseWorkUnit.EXPECT().GetStatusWithoutExtraData().Return(statusFile)
+				mockBaseWorkUnit.EXPECT().GetStatusCopy().Return(workceptor.StatusFileData{
+					ExtraData: &workceptor.CommandExtraData{},
+				})
+				mockBaseWorkUnit.EXPECT().UpdateBasicStatus(gomock.Any(), gomock.Any(), gomock.Any())
+				mockBaseWorkUnit.EXPECT().UnitDir()
+			},
+			errorCatch: func(err error, t *testing.T) {
+				if err != nil {
+					t.Error(err)
+				}
+			},
+		},
+	}
+
+	for _, testCase := range restartTestCases {
+		t.Run(testCase.name, func(t *testing.T) {
+			testCase.expectedCalls()
+			err := wu.Restart()
+			testCase.errorCatch(err, t)
+		})
+	}
+}
+
 func TestCancel(t *testing.T) {
 	wu, mockBaseWorkUnit, _, _ := createTestSetup(t)
 
@@ -194,3 +289,53 @@ func TestCancel(t *testing.T) {
 	}
 
 }
+
+func TestRelease(t *testing.T) {
+	wu, mockBaseWorkUnit, _, _ := createTestSetup(t)
+
+	releaseTestCases := []struct {
+		name          string
+		expectedCalls func()
+		errorCatch    func(error, *testing.T)
+		force         bool
+	}{
+		{
+			name:          "error",
+			expectedCalls: func() {},
+			errorCatch: func(err error, t *testing.T) {
+				if err == nil {
+					t.Error(err)
+				}
+			},
+			force: false,
+		},
+		{
+			name: "happy day",
+			expectedCalls: func() {
+				mockBaseWorkUnit.EXPECT().Release(gomock.Any())
+			},
+			errorCatch: func(err error, t *testing.T) {
+				if err != nil {
+					t.Error(err)
+				}
+			},
+			force: true,
+		},
+	}
+	for _, testCase := range releaseTestCases {
+		t.Run(testCase.name, func(t *testing.T) {
+			mockBaseWorkUnit.EXPECT().CancelContext()
+			mockBaseWorkUnit.EXPECT().GetStatusLock().Return(&sync.RWMutex{}).Times(2)
+			mockBaseWorkUnit.EXPECT().GetStatusWithoutExtraData().Return(&workceptor.StatusFileData{})
+			mockBaseWorkUnit.EXPECT().GetStatusCopy().Return(workceptor.StatusFileData{
+				ExtraData: &workceptor.CommandExtraData{
+					Pid: 1,
+				},
+			})
+			testCase.expectedCalls()
+			err := wu.Release(testCase.force)
+			testCase.errorCatch(err, t)
+		})
+	}
+
+}