fix ut

apache · Jan 3, 2024 · 1d76b54 · 1d76b54
1 parent 7e30701
commit 1d76b54
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/interface/common/resdb_state_accessor_test.cpp b/interface/common/resdb_state_accessor_test.cpp
@@ -32,6 +32,7 @@ using ::google::protobuf::util::MessageDifferencer;
 using ::testing::ElementsAre;
 using ::testing::Invoke;
 using ::testing::Test;
+using ::testing::Return;
 
 void AddReplicaToList(const std::string& ip, int port,
                       std::vector<ReplicaInfo>* replica) {
@@ -89,11 +90,13 @@ TEST_F(StateClientTest, GetAllReplicaState) {
       .Times(1)
       .WillRepeatedly(Invoke([&](const std::string& ip, int port) {
         auto client = std::make_unique<MockNetChannel>(ip, port);
+	LOG(ERROR)<<"get net:";
         EXPECT_CALL(*client, RecvRawMessage)
             .WillRepeatedly(Invoke([&](google::protobuf::Message* message) {
               *reinterpret_cast<ReplicaState*>(message) = state;
               return 0;
             }));
+        ON_CALL(*client, SendRequest).WillByDefault(Return(0));
         return client;
       }));
   auto ret = client.GetReplicaStates();