fixing unit tests

Signed-off-by: yairgott <[email protected]>

Author: yairgott

testing/schema_manager_test.cc

@@ -360,9 +360,18 @@ TEST_F(SchemaManagerTest, TestEmptyAuxLoadAfterRDBReplication) {
       &fake_ctx_, eid, REDISMODULE_SUBEVENT_LOADING_REPL_START, nullptr);
   ON_CALL(*kMockRedisModule, GetContextFromIO(testing::_))
       .WillByDefault(testing::Return(&fake_ctx_));
+  ON_CALL(*kMockRedisModule, GetDbIdFromIO(testing::_))
+      .WillByDefault(testing::Return(db_num_));
+  ON_CALL(*kMockRedisModule, GetDetachedThreadSafeContext(testing::_))
+      .WillByDefault(testing::Return(&fake_ctx_));
+  ON_CALL(*kMockRedisModule, SelectDb(&fake_ctx_, db_num_))
+      .WillByDefault(testing::Return(REDISMODULE_OK));
+  ON_CALL(*kMockRedisModule, FreeThreadSafeContext(testing::_))
+      .WillByDefault(testing::Return());
   RedisModuleIO *fake_rdb = reinterpret_cast<RedisModuleIO *>(0xDEADBEEF);
   EXPECT_CALL(*kMockRedisModule, LoadUnsigned(fake_rdb))
       .WillOnce(testing::Return(0));
+
   VMSDK_EXPECT_OK(SchemaManager::Instance().AuxLoad(fake_rdb, 0,
                                                     REDISMODULE_AUX_AFTER_RDB));
 

vmsdk/src/testing_infra/module.h

@@ -250,6 +250,7 @@ class MockRedisModule {
   MOCK_METHOD(char **, GetClusterNodesList,
               (RedisModuleCtx * ctx, size_t *numnodes));
   MOCK_METHOD(RedisModuleCtx *, GetContextFromIO, (RedisModuleIO * rdb));
+  MOCK_METHOD(int, GetDbIdFromIO, (RedisModuleIO * rdb));
   MOCK_METHOD(void, FreeClusterNodesList, (char **ids));
   MOCK_METHOD(int, CallReplyType, (RedisModuleCallReply * reply));
   MOCK_METHOD(RedisModuleString *, CreateStringFromCallReply,
@@ -1340,6 +1341,10 @@ inline RedisModuleCtx *TestRedisModule_GetContextFromIO(RedisModuleIO *rdb) {
   return kMockRedisModule->GetContextFromIO(rdb);
 }
 
+inline int TestRedisModule_GetDbIdFromIO(RedisModuleIO *rdb) {
+  return kMockRedisModule->GetDbIdFromIO(rdb);
+}
+
 inline void TestRedisModule_FreeClusterNodesList(char **ids) {
   return kMockRedisModule->FreeClusterNodesList(ids);
 }
@@ -1488,6 +1493,7 @@ inline void TestRedisModule_Init() {
   RedisModule_SendClusterMessage = &TestRedisModule_SendClusterMessage;
   RedisModule_GetClusterNodesList = &TestRedisModule_GetClusterNodesList;
   RedisModule_GetContextFromIO = &TestRedisModule_GetContextFromIO;
+  RedisModule_GetDbIdFromIO = &TestRedisModule_GetDbIdFromIO;
   RedisModule_FreeClusterNodesList = &TestRedisModule_FreeClusterNodesList;
   RedisModule_CallReplyType = &TestRedisModule_CallReplyType;
   RedisModule_CreateStringFromCallReply =