def test_get_cost_and_hardware_matrix_no_workers_registered(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.assertEqual(([], []),
self.swarm_engine._get_cost_and_hardware_matrix(
self.composition,
self.composition.get_open_allocations()))
def test_get_cost_and_hardware_row_for_service_no_workers_registered(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.assertEqual(
([], []),
self.swarm_engine._get_cost_and_hardware_row_for_service(
self.service))
def test_assign_services_to_workers(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
self.swarm_engine.create_new_swarm(master, "foo")
self.swarm_engine.register_worker_in_swarm("foo", worker)
self.assertTrue(
self.swarm_engine._assign_services_to_workers(self.composition))
def test_start_remote_logging_server_non_existent_interface(self):
try:
master = swarm_engine_master.Master("non_existent", "127.0.0.1")
master.start_remote_logging_server()
self.fail()
except NetworkException:
pass
def test_get_cost_and_hardware_matrix_swarm_open_allocations_none(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
self.swarm_engine.create_new_swarm(master, "foo")
self.swarm_engine.register_worker_in_swarm("foo", worker)
self.assertEqual((None, None),
self.swarm_engine._get_cost_and_hardware_matrix(
self.composition, None))
def test_start_swarm_by_composition(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.swarm_engine.register_worker_in_swarm(
"foo", worker_dummy.Worker("foo", "lo", "bar"))
self.assertTrue(
self.swarm_engine.start_swarm_by_composition(
self.composition, "foo"))
def test_table_builder_swarm_status_to_table(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = swarm_engine_worker.Worker("foo", "lo")
swarm_engine.SwarmEngine().swarm = swarm.Swarm("foo", master)
master.register_worker_at_master(jsonpickle.encode(master.uuid),
jsonpickle.encode(worker))
table_builder.swarm_status_to_table(
jsonpickle.decode(master.get_swarm_status_as_json()))
def test_allocate_services_to_workers_swarm_open_allocations_none(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
self.swarm_engine.create_new_swarm(master, "foo")
self.swarm_engine.register_worker_in_swarm("foo", worker)
self.assertIsNone(
self.swarm_engine._allocate_services_to_workers(
self.composition, None))
def test_get_cost_and_hardware_row_for_service_composition_none(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
self.swarm_engine.create_new_swarm(master, "foo")
self.swarm_engine.register_worker_in_swarm("foo", worker)
self.assertEqual(
(None, None),
self.swarm_engine._get_cost_and_hardware_row_for_service(None))
def test_start_services_on_workers_no_worker_registered(self):
try:
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.swarm_engine._start_services_on_workers(
self.composition, "foo")
self.fail(
"Starting services without a registered worker should throw a SwarmException"
)
except SwarmException:
pass
def test_start_services_on_workers_composition_none(self):
try:
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = worker_dummy.Worker("foo", "lo", "bar")
self.swarm_engine.create_new_swarm(master, "foo")
self.swarm_engine.register_worker_in_swarm("foo", worker)
self.swarm_engine._start_services_on_workers(None, "foo")
self.fail(
"Starting a swarm with a None composition should throw a SwarmException"
)
except SwarmException:
pass
def default_setup():
swarm_engine.SwarmEngine().swarm = None
swarm_engine.SwarmEngine()._remote_logging_server = None
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
return master, worker
def test_start_remote_logging_server_default_interface_using_none(self):
master = swarm_engine_master.Master(None, "127.0.0.1")
master.start_remote_logging_server()
def default_setup():
master = swarm_engine_master.Master("lo", "127.0.0.1")
worker = swarm_engine_worker.Worker("foo", "lo")
worker.hostname = "foobar"
sw = swarm.Swarm("foo", master)
return master, worker, sw
def setUp(self):
self.swarm_engine = default_setup()
self.master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(self.master, "foo")
def test_create_new_swarm_swarm_uuid_none(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.assertIsNotNone(self.swarm_engine.create_new_swarm(master, None))
self.assertIsNotNone(self.swarm_engine.swarm)
def test_start_swarm_by_composition_no_workers(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.assertFalse(
self.swarm_engine.start_swarm_by_composition(
self.composition, "foo"))
def test_create_docker_network(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.assertEqual("foo", self.swarm_engine._create_docker_network())
def test_allocate_services_to_workers_no_workers_registered(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.assertIsNone(
self.swarm_engine._allocate_services_to_workers(
self.composition, self.composition.get_open_allocations()))
def test_assign_services_to_workers_no_workers_registered(self):
master = swarm_engine_master.Master("lo", "127.0.0.1")
self.swarm_engine.create_new_swarm(master, "foo")
self.assertFalse(
self.swarm_engine._assign_services_to_workers(self.composition))
