fix(drain): Pass context to drainer

Fixes #27

Author: TwiN

k8s/client.go

@@ -37,11 +37,11 @@ type KubernetesClientApi interface {
 }
 
 type KubernetesClient struct {
-	client *kubernetes.Clientset
+	client kubernetes.Interface
 }
 
 // NewKubernetesClient creates a new KubernetesClient
-func NewKubernetesClient(client *kubernetes.Clientset) *KubernetesClient {
+func NewKubernetesClient(client kubernetes.Interface) *KubernetesClient {
 	return &KubernetesClient{
 		client: client,
 	}
@@ -134,6 +134,7 @@ func (k *KubernetesClient) Drain(nodeName string, ignoreDaemonSets, deleteEmptyD
 		DeleteEmptyDirData:  deleteEmptyDirData,
 		GracePeriodSeconds:  -1,
 		Timeout:             5 * time.Minute,
+		Ctx:                 context.TODO(),
 		Out:                 drainLogger{NodeName: nodeName},
 		ErrOut:              drainLogger{NodeName: nodeName},
 		OnPodDeletedOrEvicted: func(pod *v1.Pod, usingEviction bool) {

k8s/client_test.go

@@ -0,0 +1,18 @@
+package k8s
+
+import (
+	"testing"
+
+	v1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	fakekubernetes "k8s.io/client-go/kubernetes/fake"
+)
+
+func TestKubernetesClient_Drain(t *testing.T) {
+	fakeKubernetesClient := fakekubernetes.NewSimpleClientset(&v1.Node{ObjectMeta: metav1.ObjectMeta{Name: "default"}})
+	kc := NewKubernetesClient(fakeKubernetesClient)
+	err := kc.Drain("default", true, true)
+	if err != nil {
+		t.Errorf("Unexpected error: %v", err)
+	}
+}