def test_get_version(self): common.services.register(ServiceName.REQUEST_ID, MagicMock()) handler = AgentControlHandler() response = handler.get_version(VersionRequest()) assert_that(response.result, equal_to(VersionResultCode.OK)) assert_that(response.version, not_none()) assert_that(response.revision, not_none())
def test_get_schedulers(self, leaf_scheduler_cls): leaf_scheduler = MagicMock() leaf_scheduler_cls.return_value = leaf_scheduler common.services.register(Host.Iface, MagicMock()) scheduler_handler = SchedulerHandler() common.services.register(Scheduler.Iface, scheduler_handler) agent_config = MagicMock() agent_config.reboot_required = False common.services.register(ServiceName.AGENT_CONFIG, agent_config) common.services.register(ServiceName.REQUEST_ID, MagicMock()) agent_control_handler = AgentControlHandler() request = GetSchedulersRequest() response = agent_control_handler.get_schedulers(request) child_host = ChildInfo(id="foo", address="address", port=12345) assert_that(response.schedulers, is_(empty())) leaf_scheduler_id = "leaf1" scheduler_handler.configure( [SchedulerRole(leaf_scheduler_id, "parent1", host_children=[child_host])]) leaf_scheduler._get_hosts.return_value = [child_host] response = agent_control_handler.get_schedulers(request) assert_that(response.schedulers, has_length(1)) scheduler = response.schedulers[0] assert_that(scheduler.role.id, is_(leaf_scheduler_id)) assert_that(scheduler.role.host_children[0], is_(child_host)) assert_that(len(scheduler.role.host_children), is_(1)) assert_that(response.result, is_(GetSchedulersResultCode.OK))
def test_get_version(self): handler = AgentControlHandler() response = handler.get_version(VersionRequest()) assert_that(response.result, equal_to(VersionResultCode.OK)) assert_that(response.version, not_none()) assert_that(response.revision, not_none())