def install_three_tasks_test(self, proxy_getter): """Install with three tasks.""" observers = [] observer = Mock() observer.is_service_available = True observer.service_name = "A" observer.proxy.InstallWithTasks.return_value = ["/A/1"] observers.append(observer) observer = Mock() observer.is_service_available = True observer.service_name = "B" observer.proxy.InstallWithTasks.return_value = ["/B/1", "/B/2"] observers.append(observer) task_proxy = Mock() task_proxy.Steps = 1 proxy_getter.return_value = task_proxy install_manager = InstallManager() install_manager.module_observers = observers main_task = install_manager.install_system_with_task() proxy_getter.assert_has_calls( [call("A", "/A/1"), call("B", "/B/1"), call("B", "/B/2")]) self.assertIsInstance(main_task, SystemInstallationTask) self.assertEqual(main_task._subtasks, [task_proxy, task_proxy, task_proxy])
def install_with_no_modules_test(self): """Install with no modules.""" install_manager = InstallManager() install_manager.module_observers = [] main_task = install_manager.install_system_with_task() self.assertIsInstance(main_task, SystemInstallationTask) self.assertEqual(main_task._subtasks, [])
def install_three_tasks_test(self, proxy_getter): """Install with three tasks.""" observers = [] observer = Mock() observer.is_service_available = True observer.service_name = "A" observer.proxy.InstallWithTasks.return_value = ["/A/1"] observers.append(observer) observer = Mock() observer.is_service_available = True observer.service_name = "B" observer.proxy.InstallWithTasks.return_value = ["/B/1", "/B/2"] observers.append(observer) task_proxy = Mock() task_proxy.Steps = 1 proxy_getter.return_value = task_proxy install_manager = InstallManager() install_manager.module_observers = observers main_task = install_manager.install_system_with_task() proxy_getter.assert_has_calls([ call("A", "/A/1"), call("B", "/B/1"), call("B", "/B/2") ]) self.assertIsInstance(main_task, SystemInstallationTask) self.assertEqual(main_task._subtasks, [task_proxy, task_proxy, task_proxy])
def install_with_no_tasks_test(self): """Install with no tasks.""" observer = Mock() observer.is_service_available = True observer.service_name = "A" observer.proxy.InstallWithTasks.return_value = [] install_manager = InstallManager() install_manager.module_observers = [observer] main_task = install_manager.install_system_with_task() self.assertIsInstance(main_task, SystemInstallationTask) self.assertEqual(main_task._subtasks, [])
def install_one_task_test(self, proxy_getter): """Install with one task.""" observer = Mock() observer.is_service_available = True observer.service_name = "A" observer.proxy.InstallWithTasks.return_value = ["/A/1"] task_proxy = Mock() task_proxy.Steps = 1 proxy_getter.return_value = task_proxy install_manager = InstallManager() install_manager.module_observers = [observer] main_task = install_manager.install_system_with_task() proxy_getter.assert_called_once_with("A", "/A/1") self.assertIsInstance(main_task, SystemInstallationTask) self.assertEqual(main_task._subtasks, [task_proxy])