Restructure hooks tests to be reusable

Signed-off-by: Laszlo Uveges <laszlo@giantswarm.io>

pkg/action/hooks_test.go

@@ -66,34 +66,26 @@ func (h *HookFailingKubeClient) Delete(resources kube.ResourceList) (*kube.Resul
 }
 
 func TestHooksCleanUp(t *testing.T) {
-	hookKubeClient := &HookFailingKubeClient{kubefake.PrintingKubeClient{Out: ioutil.Discard}, resource.Info{
-		Name:      "build-config-2",
-		Namespace: "test",
-	}, []resource.Info{}}
-
-	configuration := &Configuration{
-		Releases:     storage.Init(driver.NewMemory()),
-		KubeClient:   hookKubeClient,
-		Capabilities: chartutil.DefaultCapabilities,
-		Log: func(format string, v ...interface{}) {
-			t.Helper()
-			if *verbose {
-				t.Logf(format, v...)
-			}
-		},
-	}
-
 	hookEvent := release.HookPreInstall
 
-	r := &release.Release{
+	testCases := []struct {
-		Name:      "test-release",
+		name                 string
-		Namespace: "test",
+		inputRelease         release.Release
-		Hooks: []*release.Hook{
+		failOn               resource.Info
-			{
+		expectedDeleteRecord []resource.Info
-				Name: "hook-1",
+		expectError          bool
-				Kind: "ConfigMap",
+	}{
-				Path: "templates/service_account.yaml",
+		{
-				Manifest: `apiVersion: v1
+			"Deletion hook runs for previously successful hook on failure of a heavier weight hook",
+			release.Release{
+				Name:      "test-release",
+				Namespace: "test",
+				Hooks: []*release.Hook{
+					{
+						Name: "hook-1",
+						Kind: "ConfigMap",
+						Path: "templates/service_account.yaml",
+						Manifest: `apiVersion: v1
 kind: ConfigMap
 metadata:
   name: build-config-1
@@ -101,24 +93,24 @@ metadata:
 data:
   foo: bar
 `,
-				Weight: -5,
+						Weight: -5,
-				Events: []release.HookEvent{
+						Events: []release.HookEvent{
-					hookEvent,
+							hookEvent,
-				},
+						},
-				DeletePolicies: []release.HookDeletePolicy{
+						DeletePolicies: []release.HookDeletePolicy{
-					release.HookBeforeHookCreation,
+							release.HookBeforeHookCreation,
-					release.HookSucceeded,
+							release.HookSucceeded,
-					release.HookFailed,
+							release.HookFailed,
-				},
+						},
-				LastRun: release.HookExecution{
+						LastRun: release.HookExecution{
-					Phase: release.HookPhaseSucceeded,
+							Phase: release.HookPhaseSucceeded,
-				},
+						},
-			},
+					},
-			{
+					{
-				Name: "hook-2",
+						Name: "hook-2",
-				Kind: "ConfigMap",
+						Kind: "ConfigMap",
-				Path: "templates/job.yaml",
+						Path: "templates/job.yaml",
-				Manifest: `apiVersion: v1
+						Manifest: `apiVersion: v1
 kind: ConfigMap
 metadata:
   name: build-config-2
@@ -126,42 +118,71 @@ metadata:
 data:
   foo: bar
 `,
-				Weight: 0,
+						Weight: 0,
-				Events: []release.HookEvent{
+						Events: []release.HookEvent{
-					hookEvent,
+							hookEvent,
+						},
+						DeletePolicies: []release.HookDeletePolicy{
+							release.HookBeforeHookCreation,
+							release.HookSucceeded,
+							release.HookFailed,
+						},
+						LastRun: release.HookExecution{
+							Phase: release.HookPhaseFailed,
+						},
+					},
+				},
+			}, resource.Info{
+				Name:      "build-config-2",
+				Namespace: "test",
+			}, []resource.Info{
+				{
+					Name:      "build-config-1",
+					Namespace: "test",
 				},
-				DeletePolicies: []release.HookDeletePolicy{
+				{
-					release.HookBeforeHookCreation,
+					Name:      "build-config-2",
-					release.HookSucceeded,
+					Namespace: "test",
-					release.HookFailed,
 				},
-				LastRun: release.HookExecution{
+				{
-					Phase: release.HookPhaseFailed,
+					Name:      "build-config-2",
+					Namespace: "test",
 				},
-			},
+			}, true,
 		},
 	}
 
-	_ = configuration.execHook(r, hookEvent, 600)
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			kubeClient := &HookFailingKubeClient{
+				kubefake.PrintingKubeClient{Out: ioutil.Discard}, tc.failOn, []resource.Info{},
+			}
 
-	if !reflect.DeepEqual(hookKubeClient.deleteRecord, []resource.Info{
+			configuration := &Configuration{
-		{
+				Releases:     storage.Init(driver.NewMemory()),
-			Name:      "build-config-1",
+				KubeClient:   kubeClient,
-			Namespace: "test",
+				Capabilities: chartutil.DefaultCapabilities,
-		},
+				Log: func(format string, v ...interface{}) {
-		{
+					t.Helper()
-			Name:      "build-config-2",
+					if *verbose {
-			Namespace: "test",
+						t.Logf(format, v...)
-		},
+					}
-		{
+				},
-			Name:      "build-config-2",
+			}
-			Namespace: "test",
+
-		},
+			err := configuration.execHook(&tc.inputRelease, hookEvent, 600)
-	}) {
-		t.Fatalf("Got unexpected delete record")
-	}
 
-	//if err != nil {
+			if !reflect.DeepEqual(kubeClient.deleteRecord, tc.expectedDeleteRecord) {
-	//	t.Fatalf("An expected error occured: %#v", err)
+				t.Fatalf("Got unexpected delete record, expected: %#v, but got: %#v", kubeClient.deleteRecord, tc.expectedDeleteRecord)
-	//}
+			}
+
+			if err != nil && !tc.expectError {
+				t.Fatalf("Got an unexpected error.")
+			}
+
+			if err == nil && tc.expectError {
+				t.Fatalf("Expected and error but did not get it.")
+			}
+		})
+	}
 }