fix: 3x shadow database start timeout

supabase · Sep 18, 2023 · dbcc233 · dbcc233
1 parent b7c58ad
commit dbcc233
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/internal/db/diff/diff.go b/internal/db/diff/diff.go
@@ -62,7 +62,7 @@ func run(p utils.Program, ctx context.Context, schema []string, config pgconn.Co
 		return err
 	}
 	defer utils.DockerRemove(shadow)
-	if !start.WaitForHealthyService(ctx, shadow, start.HealthTimeout) {
+	if !start.WaitForHealthyService(ctx, shadow, 3*start.HealthTimeout) {
 		return start.ErrDatabase
 	}
 	if err := MigrateShadowDatabase(ctx, shadow, fsys); err != nil {

diff --git a/internal/db/diff/migra.go b/internal/db/diff/migra.go
@@ -173,7 +173,7 @@ func DiffDatabase(ctx context.Context, schema []string, config pgconn.Config, w
 		return "", err
 	}
 	defer utils.DockerRemove(shadow)
-	if !start.WaitForHealthyService(ctx, shadow, start.HealthTimeout) {
+	if !start.WaitForHealthyService(ctx, shadow, 3*start.HealthTimeout) {
 		return "", start.ErrDatabase
 	}
 	if err := MigrateShadowDatabase(ctx, shadow, fsys, options...); err != nil {

diff --git a/internal/db/diff/migra_test.go b/internal/db/diff/migra_test.go
@@ -247,7 +247,7 @@ func TestDiffDatabase(t *testing.T) {
 	})
 
 	t.Run("throws error on health check failure", func(t *testing.T) {
-		start.HealthTimeout = time.Second
+		start.HealthTimeout = time.Millisecond
 		// Setup in-memory fs
 		fsys := afero.NewMemMapFs()
 		// Setup mock docker