From 0afa3f7aba6efa454aff2a44fd591bc087c1bab3 Mon Sep 17 00:00:00 2001
From: Bart Venter <bartventer@outlook.com>
Date: Thu, 19 Sep 2024 07:52:54 +0000
Subject: [PATCH] refactor(drivertest): Update tenant isolation context
 handling

---
 pkg/drivertest/isolation.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/drivertest/isolation.go b/pkg/drivertest/isolation.go
index 90bdb6d..ad3f5f2 100644
--- a/pkg/drivertest/isolation.go
+++ b/pkg/drivertest/isolation.go
@@ -37,7 +37,7 @@ func testTenantIsolation1(t *testing.T, db *multitenancy.DB, opts Options) {
 			defer wg.Done()
 			tenantCtx, cancel := context.WithCancel(ctx)
 			defer cancel()
-			tx := db.WithContext(ctx)
+			tx := db.WithContext(tenantCtx)
 			if err := tx.MigrateTenantModels(tenantCtx, tenant.ID); err != nil {
 				errCh <- err
 				return