tests: add unit tests for failures

castai · Sep 28, 2022 · a9c8093 · a9c8093
1 parent b0327ba
commit a9c8093
Showing 1 changed file with 79 additions and 0 deletions.
diff --git a/actions/drain_node_handler_test.go b/actions/drain_node_handler_test.go
@@ -158,6 +158,85 @@ func TestDrainNodeHandler(t *testing.T) {
 		_, err = clientset.CoreV1().Pods("default").Get(context.Background(), podName, metav1.GetOptions{})
 		r.True(apierrors.IsNotFound(err))
 	})
+
+	t.Run("eviction timeout - force remove pods - failure", func(t *testing.T) {
+		r := require.New(t)
+		nodeName := "node1"
+		podName := "pod1"
+		clientset := setupFakeClientWithNodePodEviction(nodeName, podName)
+
+		h := drainNodeHandler{
+			log:       log,
+			clientset: clientset,
+			cfg: drainNodeConfig{
+				podsDeleteTimeout:             7 * time.Second,
+				podDeleteRetries:              5,
+				podDeleteRetryDelay:           5 * time.Second,
+				podEvictRetryDelay:            5 * time.Second,
+				podsTerminationWaitRetryDelay: 10 * time.Second,
+			}}
+
+		req := &castai.ActionDrainNode{
+			NodeName:            "node1",
+			DrainTimeoutSeconds: 1,
+			Force:               true,
+		}
+
+		clientset.PrependReactor("delete", "pods", func(action ktest.Action) (handled bool, ret runtime.Object, err error) {
+			deleteAction := action.(ktest.DeleteActionImpl)
+			if deleteAction.Name == podName {
+				return true, nil, &apierrors.StatusError{ErrStatus: metav1.Status{Reason: metav1.StatusReasonInternalError, Message: "internal"}}
+			}
+			return false, nil, nil
+		})
+
+		err := h.Handle(context.Background(), req)
+		r.EqualError(err, "forcefully deleting pods: sending delete pods requests: deleting pod pod1 in namespace default: internal")
+
+		_, err = clientset.CoreV1().Pods("default").Get(context.Background(), podName, metav1.GetOptions{})
+		r.NoError(err)
+	})
+
+	t.Run("eviction timeout - force remove pods with grace 0 - failure", func(t *testing.T) {
+		r := require.New(t)
+		nodeName := "node1"
+		podName := "pod1"
+		clientset := setupFakeClientWithNodePodEviction(nodeName, podName)
+
+		h := drainNodeHandler{
+			log:       log,
+			clientset: clientset,
+			cfg: drainNodeConfig{
+				podsDeleteTimeout:             7 * time.Second,
+				podDeleteRetries:              5,
+				podDeleteRetryDelay:           5 * time.Second,
+				podEvictRetryDelay:            5 * time.Second,
+				podsTerminationWaitRetryDelay: 10 * time.Second,
+			}}
+
+		req := &castai.ActionDrainNode{
+			NodeName:            "node1",
+			DrainTimeoutSeconds: 1,
+			Force:               true,
+		}
+
+		clientset.PrependReactor("delete", "pods", func(action ktest.Action) (handled bool, ret runtime.Object, err error) {
+			deleteAction := action.(ktest.DeleteActionImpl)
+			if deleteAction.Name == podName {
+				if deleteAction.DeleteOptions.GracePeriodSeconds == nil {
+					return true, nil, context.DeadlineExceeded
+				}
+				return true, nil, &apierrors.StatusError{ErrStatus: metav1.Status{Reason: metav1.StatusReasonInternalError, Message: "internal"}}
+			}
+			return false, nil, nil
+		})
+
+		err := h.Handle(context.Background(), req)
+		r.EqualError(err, "deleting gracePeriod=0 pods: sending delete pods requests: deleting pod pod1 in namespace default: internal")
+
+		_, err = clientset.CoreV1().Pods("default").Get(context.Background(), podName, metav1.GetOptions{})
+		r.NoError(err)
+	})
 }
 
 func prependEvictionReaction(t *testing.T, c *fake.Clientset, success bool) {