tests: set singlehosthint flag

faasm · Jan 11, 2024 · 436c3f2 · 436c3f2
1 parent 0f8f4eb
commit 436c3f2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/planner/PlannerClient.cpp b/src/planner/PlannerClient.cpp
@@ -324,7 +324,7 @@ faabric::batch_scheduler::SchedulingDecision PlannerClient::callFunctions(
     } else {
         // In a single-host setting we can skip sending the snapshots to the
         // planner
-        if (!req->singlehost()) {
+        if (!req->singlehosthint()) {
             snapshotKey = req->messages(0).snapshotkey();
         }
     }

diff --git a/tests/test/scheduler/test_executor.cpp b/tests/test/scheduler/test_executor.cpp
@@ -1329,7 +1329,7 @@ TEST_CASE_METHOD(TestExecutorFixture,
       faabric::util::batchExecFactory("dummy", "blah", nThreads);
     req->set_type(faabric::BatchExecuteRequest::THREADS);
     // Set single-host to avoid any snapshot sending
-    req->set_singlehost(true);
+    req->set_singlehosthint(true);
 
     // Prepare executor
     auto exec = std::make_shared<TestExecutor>(*req->mutable_messages(0));

diff --git a/tests/test/scheduler/test_scheduler.cpp b/tests/test/scheduler/test_scheduler.cpp
@@ -241,7 +241,7 @@ TEST_CASE_METHOD(SlowExecutorTestFixture,
     reqOne->set_contextdata(expectedContextData);
 
     // Set the singlehost flag to avoid sending snapshots to the planner
-    reqOne->set_singlehost(true);
+    reqOne->set_singlehosthint(true);
 
     auto actualDecisionOne = plannerCli.callFunctions(reqOne);
 
@@ -297,7 +297,7 @@ TEST_CASE_METHOD(SlowExecutorTestFixture,
     reqTwo->set_type(execMode);
 
     // Set the singlehost flag to avoid sending snapshots to the planner
-    reqTwo->set_singlehost(true);
+    reqTwo->set_singlehosthint(true);
 
     // Schedule the functions
     auto actualDecisionTwo = plannerCli.callFunctions(reqTwo);