You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mesos.apache.org by bm...@apache.org on 2019/07/02 04:38:31 UTC
[mesos] 02/03: Added a test for scheduler driver registering with a
suppressed role.
This is an automated email from the ASF dual-hosted git repository.
bmahler pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/mesos.git
commit 7b2811217257d49818eebf2e519f92d0354c6e20
Author: Andrei Sekretenko <as...@mesosphere.io>
AuthorDate: Tue Jul 2 00:27:55 2019 -0400
Added a test for scheduler driver registering with a suppressed role.
Review: https://reviews.apache.org/r/70944/
---
src/tests/scheduler_driver_tests.cpp | 55 ++++++++++++++++++++++++++++++++++++
1 file changed, 55 insertions(+)
diff --git a/src/tests/scheduler_driver_tests.cpp b/src/tests/scheduler_driver_tests.cpp
index d88e1a1..80f3335 100644
--- a/src/tests/scheduler_driver_tests.cpp
+++ b/src/tests/scheduler_driver_tests.cpp
@@ -499,6 +499,61 @@ TEST_F(MesosSchedulerDriverTest, ExplicitAcknowledgementsUnsetSlaveID)
driver.join();
}
+
+// This test ensures that the driver can register with a suppressed role.
+TEST_F(MesosSchedulerDriverTest, RegisterWithSuppressedRole)
+{
+ mesos::internal::master::Flags masterFlags = CreateMasterFlags();
+ Try<Owned<cluster::Master>> master = StartMaster(masterFlags);
+ ASSERT_SOME(master);
+
+ Owned<MasterDetector> detector = master.get()->createDetector();
+ Try<Owned<cluster::Slave>> slave = StartSlave(detector.get());
+ ASSERT_SOME(slave);
+
+ FrameworkInfo frameworkInfo = DEFAULT_FRAMEWORK_INFO;
+
+ frameworkInfo.clear_roles();
+ frameworkInfo.add_roles("role1");
+ frameworkInfo.add_roles("role2");
+
+ MockScheduler sched;
+ MesosSchedulerDriver driver(
+ &sched,
+ frameworkInfo,
+ {"role2"},
+ master.get()->pid,
+ false,
+ DEFAULT_CREDENTIAL);
+
+ Future<vector<Offer>> offers;
+
+ // We should get offers EXACTLY once.
+ EXPECT_CALL(sched, resourceOffers(&driver, _))
+ .WillOnce(FutureArg<1>(&offers));
+
+ driver.start();
+
+ AWAIT_READY(offers);
+ ASSERT_EQ(1u, offers->size());
+ ASSERT_EQ("role1", offers.get()[0].allocation_info().role());
+
+ Filters filter1day;
+ filter1day.set_refuse_seconds(Days(1).secs());
+
+ driver.declineOffer(offers.get()[0].id(), filter1day);
+ Clock::pause();
+ Clock::settle();
+
+ // Trigger allocation to ensure that role2 is suppressed. We should get no
+ // more offers.
+ Clock::advance(masterFlags.allocation_interval);
+ Clock::settle();
+
+ driver.stop();
+ driver.join();
+}
+
} // namespace tests {
} // namespace internal {
} // namespace mesos {