SERVER-28214 clean up TaskExecutorMock

1 files changed, 4 insertions, 6 deletions

diff --git a/src/mongo/db/repl/abstract_async_component_test.cpp b/src/mongo/db/repl/abstract_async_component_test.cpp
index 6c2e8cd95f6..b4b71b972dd 100644
--- a/src/mongo/db/repl/abstract_async_component_test.cpp
+++ b/src/mongo/db/repl/abstract_async_component_test.cpp
@@ -337,11 +337,10 @@ TEST_F(AbstractAsyncComponentTest,
 
 TEST_F(AbstractAsyncComponentTest,
        ScheduleWorkAndSaveHandlePassesThroughErrorFromTaskExecutorScheduleWork) {
-    TaskExecutorMock taskExecutorMock(&getExecutor(),
-                                      [](const executor::RemoteCommandRequest&) { return false; });
+    TaskExecutorMock taskExecutorMock(&getExecutor());
     MockAsyncComponent component(&taskExecutorMock);
 
-    taskExecutorMock.shouldFailScheduleWork = true;
+    taskExecutorMock.shouldFailScheduleWorkRequest = []() { return true; };
 
     auto callback = [](const executor::TaskExecutor::CallbackArgs&) {};
     executor::TaskExecutor::CallbackHandle handle;
@@ -381,11 +380,10 @@ TEST_F(AbstractAsyncComponentTest,
 
 TEST_F(AbstractAsyncComponentTest,
        ScheduleWorkAtAndSaveHandlePassesThroughErrorFromTaskExecutorScheduleWork) {
-    TaskExecutorMock taskExecutorMock(&getExecutor(),
-                                      [](const executor::RemoteCommandRequest&) { return false; });
+    TaskExecutorMock taskExecutorMock(&getExecutor());
     MockAsyncComponent component(&taskExecutorMock);
 
-    taskExecutorMock.shouldFailScheduleWorkAt = true;
+    taskExecutorMock.shouldFailScheduleWorkAtRequest = []() { return true; };
 
     auto when = getExecutor().now() + Seconds(1);
     auto callback = [](const executor::TaskExecutor::CallbackArgs&) {};