def test_stop_all_services_two_added(self):
self.worker.start_remote_logger("lo", 0)
service1 = service.Service("foo1", "bar")
service2 = service.Service("foo2", "bar")
self.worker.start_service(jsonpickle.encode(service1), 'foo_network')
self.worker.start_service(jsonpickle.encode(service2), 'foo_network')
self.assertEqual(2, self.worker.stop_all_services())
def test_check_volumes_all_existing(self):
srv = service.Service()
srv.add_volume("/root", "/root")
srv.add_volume("/tmp", "/tmp")
srv.add_volume("/var", "/var")
volume_vector = self.interface.check_volumes(srv)
self.assertEqual([1, 1, 1], volume_vector)
def test_run_container_in_background_none_network(self):
try:
self.interface.init_docker_swarm("lo")
logger = remote_logger.RemoteLogger(None, None, None, None)
srv = service.Service("foo", "hello-world")
self.assertIsNotNone(self.interface.run_container_in_background(srv, logger, None))
except DockerException:
self.fail()
def test_service_format_service_definition_as_table(self):
srv = service.Service()
srv.add_env("KEY", "VALUE")
srv.add_dependency("MASTER")
srv.add_deployment("KEY", "VALUE")
srv.add_volume("SRC", "DST")
srv.add_device("SRC", "DST")
srv.format_service_definition_as_table()
def test_run_container_in_background_none_logger(self):
try:
self.interface.init_docker_swarm("lo")
srv = service.Service("foo", "hello-world")
self.interface.create_network("bar")
time.sleep(1)
self.assertIsNotNone(self.interface.run_container_in_background(srv, None, "bar"))
except DockerException:
self.fail()
def test_run_container_in_background_empty_service_params(self):
try:
logger = remote_logger.RemoteLogger(None, None, None, None)
srv = service.Service(None, None)
self.interface.create_network("bar")
time.sleep(1)
self.interface.run_container_in_background(srv, logger, "bar")
self.fail()
except DockerException:
pass
def test_get_usage_of_worker(self):
worker = worker_dummy.Worker("foo", "lo", "bar")
srv = service.Service("baz", "hello-world")
cpu_usage, vram_usage, swap_usage, image_size, bandwidth_percent = self.cost_calculation._get_usage_of_worker(
worker, srv)
self.assertEqual(0.005, cpu_usage)
self.assertEqual(0.005, vram_usage)
self.assertEqual(0.005, swap_usage)
self.assertEqual(100, image_size)
self.assertEqual(8e-06, bandwidth_percent)
def test_check_devices_existing_device(self):
srv = service.Service()
srv.add_device("/dev", "/dev")
device_vector = self.interface.check_devices(srv)
self.assertEqual([1], device_vector)
def test_check_volumes_existing_volume(self):
srv = service.Service()
srv.add_volume("/tmp", "/tmp")
volume_vector = self.interface.check_volumes(srv)
self.assertEqual([1], volume_vector)
def default_setup():
srv = service.Service()
worker = swarm_engine_worker.Worker("foo", "lo")
composition = service_composition.ServiceComposition()
return srv, worker, composition
def test_check_hardware_non_existent(self):
srv = service.Service()
srv.add_volume("/non_existent", "/non_existent")
srv.add_device("/non_existent", "/non_existent")
self.assertEqual(0, self.worker.check_hardware(jsonpickle.encode(srv)))
def test_check_hardware(self):
srv = service.Service()
srv.add_volume("/var", "/var")
srv.add_device("/dev", "/dev")
self.assertEqual(1, self.worker.check_hardware(jsonpickle.encode(srv)))
def test_cost_calculation_5(self):
srv = service.Service("foo", "foo:bar")
worker = WorkerDummy("foo:baz", 50)
q = queue.Queue()
self.cost_calculation.calculate_costs_and_check_hardware_in_thread(0, srv, worker, q)
self.assertEqual({0: {'cost': 40625, 'hw': 0}}, q.get())
def test_service_composition_format_service_composition_as_table(self):
srv = service.Service()
composition = service_composition.ServiceComposition()
composition.add_service("KEY", srv)
composition.add_service("KEY2", srv)
composition.format_service_composition_as_table()
def test_check_devices_not_existing_device(self):
srv = service.Service()
srv.add_device("/dev/non_existent", "/dev/non_existent")
volume_vector = self.interface.check_devices(srv)
self.assertEqual([0], volume_vector)
def test_check_hardware_service_definition_empty(self):
self.assertEqual(
1,
self.worker.check_hardware(jsonpickle.encode(service.Service())))
def test_check_devices_empty(self):
srv = service.Service()
volume_vector = self.interface.check_devices(srv)
self.assertEqual([], volume_vector)
def setUp(self):
self.worker = default_setup()
self.service = service.Service("foo", "bar")
def setUp(self):
self.service = service.Service()
def setUp(self):
self.swarm_engine = default_setup()
self.service = service.Service("id", "tag")
