def test_acquire_machine(self, setup_machinery, setup_machine_lock): class mock_machinery(object): def availables(self): return True def acquire(self, machine_id, platform, tags): class mock_acquire: name = "mock_mach" label = "mock_label" platform = "mock_platform" return mock_acquire() class mock_machine: machine = "mock" class mock_plat: platform = "plat" class mock_tags: tags = "tags" setup_machinery(mock_machinery()) mock_task_machine = mock_task() mock_task_machine.machine = mock_machine() mock_task_machine.platform = mock_plat() mock_task_machine.tags = mock_tags() analysis_man = AnalysisManager(task=mock_task_machine, error_queue=queue.Queue()) analysis_man.acquire_machine() assert analysis_man.machine.name == "mock_mach"
def test_acquire_machine(self, setup_machinery, setup_machine_lock): class mock_machinery: def availables(self, machine_id, platform, tags, arch): return True def acquire(self, machine_id, platform, tags, arch): class mock_acquire: name = "mock_mach" label = "mock_label" platform = "mock_platform" arch = "x64" return mock_acquire() class mock_machine: machine = "mock" class mock_plat: platform = "plat" class mock_tags: class mock_tag: def __init__(self, name): self.name = name tags = [mock_tag("tag1"), mock_tag("tag2")] def __iter__(self): for tag in self.tags: yield tag class mock_arch: arch = "x64" setup_machinery(mock_machinery()) mock_task_machine = mock_task() mock_task_machine.machine = mock_machine() mock_task_machine.platform = mock_plat() mock_task_machine.tags = mock_tags() mock_task_machine.arch = mock_arch() analysis_man = AnalysisManager(task=mock_task_machine, error_queue=queue.Queue()) analysis_man.acquire_machine() assert analysis_man.machine.name == "mock_mach"