tests: Disconnect from databases before starting cleanup

EasyMile · Apr 10, 2024 · 4677d44 · 4677d44
1 parent 763e382
commit 4677d44
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/internal/controller/postgresql/suite_test.go b/internal/controller/postgresql/suite_test.go
@@ -219,6 +219,10 @@ var _ = AfterSuite(func() {
 })
 
 func cleanupFunction() {
+	for k, _ := range dbConns {
+		disconnectConnFromKey(k)
+	}
+
 	// Force delete pgec
 	err := deletePGEC(ctx, k8sClient, pgecName, pgecNamespace)
 	Expect(err).ToNot(HaveOccurred())
@@ -243,10 +247,6 @@ func cleanupFunction() {
 	Expect(err).ToNot(HaveOccurred())
 	err = deleteSecret(ctx, k8sClient, editedSecretName, pgurNamespace)
 	Expect(err).ToNot(HaveOccurred())
-
-	for k, _ := range dbConns {
-		disconnectConnFromKey(k)
-	}
 }
 
 func getSecret(ctx context.Context, cli client.Client, name, namespace string) (*corev1.Secret, error) {