Fix unit test.

m3db · Jul 17, 2020 · 3c2191c · 3c2191c
1 parent d10a130
commit 3c2191c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/dbnode/storage/fs_test.go b/src/dbnode/storage/fs_test.go
@@ -81,10 +81,11 @@ func TestFileSystemManagerRun(t *testing.T) {
 	fsm := newFileSystemManager(database, nil, DefaultTestOptions())
 	mgr := fsm.(*fileSystemManager)
 	mgr.databaseFlushManager = fm
+	mgr.databaseCleanupManager = cm
 
 	ts := time.Now()
 	gomock.InOrder(
-		cm.EXPECT().Cleanup(ts, true).Return(errors.New("foo")),
+		cm.EXPECT().WarmFlushCleanup(ts, true).Return(errors.New("foo")),
 		fm.EXPECT().Flush(ts).Return(errors.New("bar")),
 	)