fix e2e test

cvvz · cvvz · commit 94198d39afc4 · 2025-03-17T11:18:39.000+08:00
diff --git a/test/e2e/e2e_suite_test.go b/test/e2e/e2e_suite_test.go
@@ -109,8 +109,6 @@ var _ = ginkgo.BeforeSuite(func() {
 		kubeconfig := filepath.Join(os.Getenv("HOME"), ".kube", "config")
 		os.Setenv(kubeconfigEnvVar, kubeconfig)
 	}
-	handleFlags()
-	framework.AfterReadingAllFlags(&framework.TestContext)
 
 	options := nfs.DriverOptions{
 		NodeID:     nodeID,
@@ -215,6 +213,12 @@ func execTestCmd(cmds []testCmd) {
 	}
 }
 
+func TestMain(m *testing.M) {
+	handleFlags()
+	framework.AfterReadingAllFlags(&framework.TestContext)
+	os.Exit(m.Run())
+}
+
 func TestE2E(t *testing.T) {
 	gomega.RegisterFailHandler(ginkgo.Fail)
 	ginkgo.RunSpecs(t, "E2E Suite")

Original file line number	Diff line number	Diff line change
`@@ -109,8 +109,6 @@ var _ = ginkgo.BeforeSuite(func() {`
`109`	`109`	`kubeconfig := filepath.Join(os.Getenv("HOME"), ".kube", "config")`
`110`	`110`	`os.Setenv(kubeconfigEnvVar, kubeconfig)`
`111`	`111`	`}`
`112`		`- handleFlags()`
`113`		`- framework.AfterReadingAllFlags(&framework.TestContext)`
`114`	`112`
`115`	`113`	`options := nfs.DriverOptions{`
`116`	`114`	`NodeID: nodeID,`
`@@ -215,6 +213,12 @@ func execTestCmd(cmds []testCmd) {`
`215`	`213`	`}`
`216`	`214`	`}`
`217`	`215`
	`216`	`+func TestMain(m *testing.M) {`
	`217`	`+ handleFlags()`
	`218`	`+ framework.AfterReadingAllFlags(&framework.TestContext)`
	`219`	`+ os.Exit(m.Run())`
	`220`	`+}`
	`221`	`+`
`218`	`222`	`func TestE2E(t *testing.T) {`
`219`	`223`	`gomega.RegisterFailHandler(ginkgo.Fail)`
`220`	`224`	`ginkgo.RunSpecs(t, "E2E Suite")`