til: InvokeWithoutArgs exists

googleapis · Jan 18, 2024 · 956212b · dbolduc · Jan 18, 2024 · 956212b
1 parent 586a3c5
commit 956212b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/google/cloud/pubsub/internal/pull_ack_handler_factory_test.cc b/google/cloud/pubsub/internal/pull_ack_handler_factory_test.cc
@@ -38,6 +38,7 @@ using ::testing::AllOf;
 using ::testing::ByMove;
 using ::testing::Contains;
 using ::testing::ElementsAre;
+using ::testing::InvokeWithoutArgs;
 using ::testing::IsEmpty;
 using ::testing::Property;
 using ::testing::Return;
@@ -73,7 +74,8 @@ TEST(PullAckHandlerTest, AckSimple) {
   // Since the lease manager is started in the constructor of the ack handler,
   // we need to match the lease manager calls.
   EXPECT_CALL(*mock, AsyncModifyAckDeadline(_, _, _))
-      .WillRepeatedly(Return(make_ready_future(Status{})));
+      .WillRepeatedly(
+          InvokeWithoutArgs([]() { return make_ready_future(Status{}); }));
   AsyncSequencer<bool> aseq;
   auto cq = MakeMockCompletionQueue(aseq);
   auto handler =
@@ -107,7 +109,8 @@ TEST(PullAckHandlerTest, TracingEnabled) {
   // Since the lease manager is started in the constructor of the ack handler,
   // we need to match the lease manager calls.
   EXPECT_CALL(*mock, AsyncModifyAckDeadline(_, _, _))
-      .WillRepeatedly(Return(make_ready_future(Status{})));
+      .WillRepeatedly(
+          InvokeWithoutArgs([]() { return make_ready_future(Status{}); }));
   AsyncSequencer<bool> aseq;
   auto cq = MakeMockCompletionQueue(aseq);
   auto handler =
@@ -135,7 +138,9 @@ TEST(PullAckHandlerTest, TracingDisabled) {
   // Since the lease manager is started in the constructor of the ack handler,
   // we need to match the lease manager calls.
   EXPECT_CALL(*mock, AsyncModifyAckDeadline(_, _, _))
-      .WillRepeatedly(Return(make_ready_future(Status{})));
+      .WillRepeatedly(
+          InvokeWithoutArgs([]() { return make_ready_future(Status{}); }));
+
   AsyncSequencer<bool> aseq;
   auto cq = MakeMockCompletionQueue(aseq);
   auto handler =