Added expected calls to GetLogger in test

fimanishi · Jan 22, 2025 · c16094f · c16094f
1 parent d03c676
commit c16094f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/common/persistence/nosql/nosql_task_store_test.go b/common/persistence/nosql/nosql_task_store_test.go
@@ -73,6 +73,7 @@ func setupNoSQLStoreMocks(t *testing.T) (*nosqlTaskStore, *nosqlplugin.MockDB) {
 			TestTaskType).
 		Return(&nosqlSt, nil).
 		AnyTimes()
+	shardedNosqlStoreMock.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()
 
 	store := &nosqlTaskStore{
 		shardedNosqlStore: shardedNosqlStoreMock,