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_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_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_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_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_init(self):
worker = swarm_engine_worker.Worker("foo", "lo")
self.assertEqual("127.0.0.1", worker.advertise_address)
self.assertEqual("lo", worker.interface)
self.assertEqual(socket.gethostname(), worker.hostname)
self.assertIsNotNone(worker.uuid)
self.assertEqual("foo", worker.swarm_uuid)
self.assertIsNone(worker._remote_logger)
self.assertIsNotNone(worker._container_list)
def test_unregister_worker_in_swarm(self):
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
self.swarm_engine.register_worker_in_swarm("foo", worker)
self.swarm_engine.unregister_worker_in_swarm("foo", "bar")
self.assertEqual({}, self.swarm_engine.swarm._worker_list)
def test_join_docker_swarm(self):
worker = swarm_engine_worker.Worker("foo", "lo")
self.daemon.register_worker_at_local_daemon(worker)
self.assertTrue(
self.daemon.join_docker_swarm("foo", "127.0.0.1", "bar"))
def test_table_builder_worker_daemon_status_to_table(self):
daemon = swarmrob_d.SwarmRobDaemon("lo", None)
worker = swarm_engine_worker.Worker("foo", "lo")
daemon.register_worker_at_local_daemon(worker)
table_builder.worker_daemon_status_to_table(
jsonpickle.decode(daemon.get_worker_status_as_json()))
def test_table_builder_worker_status_to_table(self):
worker = swarm_engine_worker.Worker("foo", "lo")
table_builder.worker_status_to_table(
jsonpickle.decode(worker.get_info_as_json()))
def test_unregister_worker_in_swarm_swarm_uuid_none(self):
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
self.swarm_engine.register_worker_in_swarm("foo", worker)
self.swarm_engine.unregister_worker_in_swarm(None, "bar")
self.assertEqual({'bar': worker}, self.swarm_engine.swarm._worker_list)
def default_setup():
srv = service.Service()
worker = swarm_engine_worker.Worker("foo", "lo")
composition = service_composition.ServiceComposition()
return srv, worker, composition
def test_get_worker_count(self):
self.swarm.add_worker_to_list(self.worker)
self.swarm.add_worker_to_list(swarm_engine_worker.Worker("bar", "lo"))
self.swarm.add_worker_to_list(swarm_engine_worker.Worker("baz", "lo"))
self.assertEqual(3, self.swarm.get_worker_count())
def test_get_unique_worker_hostname_name_exists_twice(self):
self.swarm.add_worker_to_list(self.worker)
worker2 = swarm_engine_worker.Worker("foo", "lo")
worker2.hostname = "foobar_1"
self.swarm.add_worker_to_list(worker2)
self.assertEqual("foobar_2", self.swarm.get_unique_worker_hostname("foobar"))
def test_init_with_worker_uuid(self):
worker = swarm_engine_worker.Worker("foo", "lo", "bar")
self.assertEqual("bar", worker.uuid)
def test_unregister_worker_at_local_daemon_worker_uuid_none(self):
worker = swarm_engine_worker.Worker("foo", "lo")
self.daemon.register_worker_at_local_daemon(worker)
self.daemon.unregister_worker_at_local_daemon(None)
self.assertEqual({"foo": worker}, self.daemon._swarm_list_of_worker)
def test_register_worker_at_local_daemon_register_the_same_twice(self):
worker = swarm_engine_worker.Worker("foo", "lo")
self.daemon.register_worker_at_local_daemon(worker)
self.daemon.register_worker_at_local_daemon(worker)
self.assertEqual({"foo": worker}, self.daemon._swarm_list_of_worker)
def default_setup():
return swarm_engine_worker.Worker("foo", "lo")
def test_join_docker_swarm_catching_runtime_exception(self):
worker = swarm_engine_worker.Worker("foo", "lo")
self.daemon.register_worker_at_local_daemon(worker)
self.assertFalse(self.daemon.join_docker_swarm("foo", None, None))
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 test_worker(self):
worker = swarm_engine_worker.Worker("foo", "lo")
self.daemon.register_worker_at_local_daemon(worker)
self.assertEqual(mode.Mode.WORKER,
jsonpickle.decode(self.daemon.get_mode()))
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_get_worker_status_as_json(self):
worker = swarm_engine_worker.Worker("foo", "lo")
self.daemon.register_worker_at_local_daemon(worker)
self.assertEqual(
1, len(jsonpickle.decode(self.daemon.get_worker_status_as_json())))
