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())))