Update fivetran_test.go

fivetran · Oct 2, 2024 · ec79c25 · ec79c25
1 parent a5d4066
commit ec79c25
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/fivetran/tests/e2e/fivetran_test.go b/fivetran/tests/e2e/fivetran_test.go
@@ -89,6 +89,7 @@ func cleanupAccount() {
 	cleanupWebhooks()
 	cleanupTeams()
 	cleanupProxyAgents()
+	cleanupPrivateLinks()
 	cleanupHybridDeploymentAgents()
 }
 
@@ -277,6 +278,23 @@ func cleanupProxyAgents() {
 	}
 }
 
+func cleanupPrivateLinks() {
+	plList, err := client.NewPrivateLinkList().Do(context.Background())
+	if err != nil {
+		log.Fatal(err)
+	}
+	for _, pl := range plList.Data.Items {
+		_, err := client.NewPrivateLinkDelete().PrivateLinkId(pl.Id).Do(context.Background())
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+
+	if plList.Data.NextCursor != "" {
+		cleanupPrivateLinks()
+	}
+}
+
 func cleanupHybridDeploymentAgents() {
 	lpaList, err := client.NewHybridDeploymentAgentList().Do(context.Background())
 	if err != nil {