Add env to distinguish type of cluster

kubernetes-sigs · Jun 27, 2022 · 31960c4 · 31960c4
1 parent daf1681
commit 31960c4
Showing 1 changed file with 25 additions and 9 deletions.
diff --git a/examples/real_cluster/main_test.go b/examples/real_cluster/main_test.go
@@ -33,15 +33,31 @@ var testenv env.Environment
 
 func TestMain(m *testing.M) {
 	testenv = env.New()
-	path := conf.ResolveKubeConfigFile()
-	cfg := envconf.NewWithKubeConfig(path)
-	testenv = env.NewWithConfig(cfg)
 	namespace := envconf.RandomName("sample-ns", 16)
-	testenv.Setup(
-		envfuncs.CreateNamespace(namespace),
-	)
-	testenv.Finish(
-		envfuncs.DeleteNamespace(namespace),
-	)
+	if os.Getenv("REAL_CLUSTER") == "true" {
+		path := conf.ResolveKubeConfigFile()
+		cfg := envconf.NewWithKubeConfig(path)
+		testenv = env.NewWithConfig(cfg)
+
+		testenv.Setup(
+			envfuncs.CreateNamespace(namespace),
+		)
+		testenv.Finish(
+			envfuncs.DeleteNamespace(namespace),
+		)
+	} else {
+		kindClusterName := envconf.RandomName("kind-with-config", 16)
+
+		testenv.Setup(
+			envfuncs.CreateKindCluster(kindClusterName),
+			envfuncs.CreateNamespace(namespace),
+		)
+
+		testenv.Finish(
+			envfuncs.DeleteNamespace(namespace),
+			envfuncs.DestroyKindCluster(kindClusterName),
+		)
+	}
+
 	os.Exit(testenv.Run(m))
 }