Set --gce-project flag too

kubernetes · Jan 21, 2022 · 5363766 · 5363766
1 parent e8919b2
commit 5363766
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/tests/e2e/pkg/tester/tester.go b/tests/e2e/pkg/tester/tester.go
@@ -248,6 +248,26 @@ func (t *Tester) addClusterTagFlag() error {
 	return nil
 }
 
+func (t *Tester) addProjectFlag() error {
+	if hasFlag(t.TestArgs, "gce-project") {
+		return nil
+	}
+
+	cluster, err := t.getKopsCluster()
+	if err != nil {
+		return err
+	}
+
+	projectID := cluster.Spec.Project
+	if projectID == "" {
+		return nil
+	}
+	klog.Infof("Setting --gce-project=%s", projectID)
+	t.TestArgs += " --gce-project=" + projectID
+
+	return nil
+}
+
 func (t *Tester) getZones() ([]string, error) {
 	cluster, err := t.getKopsCluster()
 	if err != nil {
@@ -325,6 +345,10 @@ func (t *Tester) execute() error {
 		return err
 	}
 
+	if err := t.addProjectFlag(); err != nil {
+		return err
+	}
+
 	return t.Test()
 }