def test_get_environment_id_multi_available_bad_name(self, mock_sys_exit): def env_list(*args, **kwargs): return [{ "id": "1234", "name": "somename" }, { "id": "5678", "name": "othername" }] # Avoid writing to stderr mock_sys_exit.side_effect = SystemExit(-1) StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="not_an_env") try: rc._get_environment_id(cp, 'owner', None) self.fail("No Exception Raised") except SystemExit: pass
def test_get_environment_id_one_available(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}] StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None env_id = rc._get_environment_id(cp, "owner", None) expected = "1234" self.assertEquals(expected, env_id)
def test_get_environment_id_one_available(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}] StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None env_id = rc._get_environment_id(cp, 'owner', None) expected = "1234" self.assertEquals(expected, env_id)
def test_get_environment_id_multi_available_bad_name(self, mock_sys_exit): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}, {"id": "5678", "name": "othername"}] # Avoid writing to stderr mock_sys_exit.side_effect = SystemExit(-1) StubUEP.getEnvironmentList = env_list cp = StubUEP() cp.supports_resource = Mock(return_value=True) rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="not_an_env") try: rc._get_environment_id(cp, "owner", None) self.fail("No Exception Raised") except SystemExit: pass