Minor update

agautam478 · Apr 4, 2024 · 714ce49 · 714ce49
1 parent 4688987
commit 714ce49
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/common/persistence/nosql/nosql_execution_store_test.go b/common/persistence/nosql/nosql_execution_store_test.go
@@ -1157,6 +1157,7 @@ func TestRangeDeleteReplicationTaskFromDLQ(t *testing.T) {
 			exclusiveBeginID: 1,
 			inclusiveEndID:   100,
 			setupMock: func(mockDB *nosqlplugin.MockDB) {
+				mockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()
 				mockDB.EXPECT().
 					RangeDeleteReplicationDLQTasks(ctx, shardID, "sourceCluster", int64(1), int64(100)).
 					Return(errors.New("database error"))