Fixed flakey test by waiting for complete before doing dirty migrate

loopholelabs · Feb 26, 2024 · 30c03ad · 30c03ad
1 parent e1a7d31
commit 30c03ad
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/storage/migrator/migrator_test.go b/pkg/storage/migrator/migrator_test.go
@@ -273,6 +273,9 @@ func TestMigratorWithReaderWriter(t *testing.T) {
 
 	mig.Migrate(num_blocks)
 
+	err = mig.WaitForCompletion()
+	assert.NoError(t, err)
+
 	for {
 		blocks := mig.GetLatestDirty()
 		if blocks == nil {