Fixed possible race in unit test assertion.

m3db · Oct 6, 2021 · 52b554e · 52b554e
1 parent e09cc8b
commit 52b554e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/dbnode/storage/database_test.go b/src/dbnode/storage/database_test.go
@@ -724,7 +724,7 @@ func TestDatabaseAddNamespaceBootstrapEnqueue(t *testing.T) {
 	// Usually this update should complete in a few seconds.
 	require.True(t, xclock.WaitUntil(func() bool {
 		return nsHooks.addCount() == 1
-	}, 10*time.Minute))
+	}, 1*time.Minute))
 	require.True(t, xclock.WaitUntil(func() bool {
 		return len(d.Namespaces()) == 3
 	}, 2*time.Second))
@@ -1752,6 +1752,12 @@ func assertFileOpsEnabled(t *testing.T, d *db) {
 	mediator := d.mediator.(*mediator)
 	coldFlushManager := mediator.databaseColdFlushManager.(*coldFlushManager)
 	fileSystemManager := mediator.databaseFileSystemManager.(*fileSystemManager)
+
+	coldFlushManager.RLock()
 	require.True(t, coldFlushManager.enabled)
+	coldFlushManager.RUnlock()
+
+	fileSystemManager.RLock()
 	require.True(t, fileSystemManager.enabled)
+	fileSystemManager.RUnlock()
 }