chore: fix custom dest mgr tests for kv

rudderlabs · May 3, 2024 · 1b512bf · 1b512bf
1 parent 9611422
commit 1b512bf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/router/customdestinationmanager/customdestinationmanager_test.go b/router/customdestinationmanager/customdestinationmanager_test.go
@@ -172,6 +172,7 @@ func TestKVManagerInvocations(t *testing.T) {
 		`)
 		ctrl := gomock.NewController(t)
 		mockKVStoreManager := mock_kvstoremanager.NewMockKVStoreManager(ctrl)
+		mockKVStoreManager.EXPECT().ShouldSendDataAsJSON(someDestination.Config).Return(false).Times(1)
 		mockKVStoreManager.EXPECT().HSet("someHash", "someKey", "someValue").Times(1)
 		mockKVStoreManager.EXPECT().StatusCode(nil).Times(1)
 		customManager.send(event, mockKVStoreManager, someDestination.Config)
@@ -189,6 +190,7 @@ func TestKVManagerInvocations(t *testing.T) {
 		  }
 		`)
 		mockKVStoreManager := mock_kvstoremanager.NewMockKVStoreManager(ctrl)
+		mockKVStoreManager.EXPECT().ShouldSendDataAsJSON(someDestination.Config).Return(false).Times(1)
 		mockKVStoreManager.EXPECT().HMSet("someKey", map[string]interface{}{
 			"field1": "value1",
 			"field2": "value2",