uncomment debug

armadaproject · Jun 23, 2023 · 830942a · 830942a
1 parent 61ce1c4
commit 830942a
Showing 1 changed file with 15 additions and 14 deletions.
diff --git a/internal/common/database/db_testutil.go b/internal/common/database/db_testutil.go
@@ -2,6 +2,7 @@ package database
 
 import (
 	"context"
+	"fmt"
 
 	"github.com/jackc/pgx/v4"
 	"github.com/jackc/pgx/v4/pgxpool"
@@ -38,20 +39,20 @@ func WithTestDb(migrations []Migration, action func(db *pgxpool.Pool) error) err
 		return errors.WithStack(err)
 	}
 
-	//defer func() {
-	//	// disconnect all db user before cleanup
-	//	_, err = db.Exec(ctx,
-	//		`SELECT pg_terminate_backend(pg_stat_activity.pid)
-	//		 FROM pg_stat_activity WHERE pg_stat_activity.datname = '`+dbName+`';`)
-	//	if err != nil {
-	//		fmt.Println("Failed to disconnect users")
-	//	}
-	//
-	//	_, err = db.Exec(ctx, "DROP DATABASE "+dbName)
-	//	if err != nil {
-	//		fmt.Println("Failed to drop database")
-	//	}
-	//}()
+	defer func() {
+		// disconnect all db user before cleanup
+		_, err = db.Exec(ctx,
+			`SELECT pg_terminate_backend(pg_stat_activity.pid)
+			 FROM pg_stat_activity WHERE pg_stat_activity.datname = '`+dbName+`';`)
+		if err != nil {
+			fmt.Println("Failed to disconnect users")
+		}
+
+		_, err = db.Exec(ctx, "DROP DATABASE "+dbName)
+		if err != nil {
+			fmt.Println("Failed to drop database")
+		}
+	}()
 
 	err = UpdateDatabase(ctx, testDbPool, migrations)
 	if err != nil {