@@ -2,8 +2,8 @@ package tests
22
33import (
44 "context"
5- "fmt "
6- "strings "
5+ "flag "
6+ "os "
77 "testing"
88 "time"
99
@@ -12,30 +12,31 @@ import (
1212 CoreV1 "k8s.io/api/core/v1"
1313 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
1414 "k8s.io/client-go/kubernetes"
15- envtest "sigs. k8s.io/controller-runtime/pkg/envtest "
15+ " k8s.io/client-go/tools/clientcmd "
1616)
1717
18- func startCluster () (* kubernetes.Clientset , * envtest.Environment ) {
19- testenv := & envtest.Environment {}
20- cfg , err := testenv .Start ()
18+ var (
19+ kubeConfigPath string
20+ openebsNamespace string
21+ clientSet * kubernetes.Clientset
22+ LocalPVProvisionerLabelSelector = "openebs.io/component-name=openebs-localpv-provisioner"
23+ )
24+
25+ func init () {
26+ flag .StringVar (& kubeConfigPath , "kubeconfig" , os .Getenv ("KUBECONFIG" ), "path to kubeconfig to invoke kubernetes API calls" )
27+ flag .StringVar (& openebsNamespace , "openebs-namespace" , "openebs" , "kubernetes namespace where the OpenEBS components are present" )
28+ config , err := clientcmd .BuildConfigFromFlags ("" , kubeConfigPath )
2129 if err != nil {
2230 panic (err .Error ())
2331 }
24- clientset , err : = kubernetes .NewForConfig (cfg )
32+ clientSet , err = kubernetes .NewForConfig (config )
2533 if err != nil {
2634 panic (err .Error ())
2735 }
28- return clientset , testenv
29- }
30-
31- func stopCluster (testenv * envtest.Environment ) {
32- testenv .Stop ()
3336}
3437
3538func TestMain (t * testing.T ) {
3639 ctx := context .Background ()
37- clientSet , clusterTestEnv := startCluster ()
38- defer stopCluster (clusterTestEnv )
3940
4041 // Ensure openebs-localpv-provisioner pod running
4142 setup .TestOpenEbsPodRunningState (t , clientSet , ctx , 90 )
@@ -75,15 +76,15 @@ func TestMain(t *testing.T) {
7576
7677 time .Sleep (5 * time .Second )
7778
78- pvcs , err := clientSet .CoreV1 ().PersistentVolumeClaims ("default" ).List (ctx , metav1.ListOptions {})
79+ setup .TestDanglingPVCDeleted (t , clientSet , ctx , statefulsetName , 90 )
80+
81+ err = clientSet .BatchV1 ().Jobs ("default" ).Delete (ctx , "test-job" , metav1.DeleteOptions {})
7982 if err != nil {
80- fmt .Printf ("error %s, getting PVCs\n " , err .Error ())
81- }
82- for _ , pvc := range pvcs .Items {
83- if strings .Contains (pvc .Name , statefulsetName ) {
84- t .Fatalf ("Dangling PVCs %v not deleted by Job" , pvc .Name )
85- }
83+ panic (err .Error ())
8684 }
8785
88- // clientSet.BatchV1().Jobs("default").Delete(ctx, "test-job", metav1.DeleteOptions{})
86+ err = clientSet .StorageV1 ().StorageClasses ().Delete (ctx , "test-storage-class" , metav1.DeleteOptions {})
87+ if err != nil {
88+ panic (err .Error ())
89+ }
8990}
0 commit comments