move flags to test/e2e/common

Signed-off-by: Alex Castilio dos Santos <[email protected]>

test/e2e/common/common.go

@@ -5,6 +5,7 @@
 package common
 
 import (
+	"flag"
 	"os"
 	"os/user"
 	"strconv"
@@ -22,7 +23,11 @@ const (
 	TestPodNamespace    = "kube-system-test"
 )
 
-var AzureLocations = []string{"eastus2", "northeurope", "uksouth", "centralindia", "westus2"}
+var (
+	AzureLocations = []string{"eastus2", "northeurope", "uksouth", "centralindia", "westus2"}
+	CreateInfra    = flag.Bool("create-infra", true, "create a Resource group, vNET and AKS cluster for testing")
+	DeleteInfra    = flag.Bool("delete-infra", true, "delete a Resource group, vNET and AKS cluster for testing")
+)
 
 func ClusterNameForE2ETest(t *testing.T) string {
 	clusterName := os.Getenv("CLUSTER_NAME")

test/e2e/retina_e2e_test.go

@@ -4,7 +4,6 @@ package retina
 
 import (
 	"crypto/rand"
-	"flag"
 	"math/big"
 	"os"
 	"path/filepath"
@@ -22,8 +21,6 @@ func TestE2ERetina(t *testing.T) {
 	ctx, cancel := helpers.Context(t)
 	defer cancel()
 
-	flag.Parse()
-
 	// Truncate the username to 8 characters
 	clusterName := common.ClusterNameForE2ETest(t)
 
@@ -57,6 +54,8 @@ func TestE2ERetina(t *testing.T) {
 	kubeConfigFilePath := filepath.Join(rootDir, "test", "e2e", "test.pem")
 
 	// CreateTestInfra
+	createInfra := common.CreateInfra
+	deleteInfra := common.DeleteInfra
 	createTestInfra := types.NewRunner(t, jobs.CreateTestInfra(subID, rg, clusterName, location, kubeConfigFilePath, *createInfra))
 	createTestInfra.Run(ctx)
 

test/e2e/scale_test.go

@@ -89,6 +89,9 @@ func TestE2ERetina_Scale(t *testing.T) {
 
 	opt.LabelsToGetMetrics = map[string]string{"k8s-app": "retina"}
 
+	createInfra := common.CreateInfra
+	deleteInfra := common.DeleteInfra
+
 	// CreateTestInfra
 	createTestInfra := types.NewRunner(t, jobs.CreateTestInfra(subID, rg, clusterName, location, kubeConfigFilePath, *createInfra))
 	createTestInfra.Run(ctx)