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_multi_available(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}, {"id": "5678", "name": "othername"}] 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="othername") env_id = rc._get_environment_id(cp, "owner", None) expected = "5678" self.assertEquals(expected, env_id)
def test_get_environment_id_multi_available_bad_name(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}, {"id": "5678", "name": "othername"}] with patch('rhsm.connection.UEPConnection', new_callable=StubUEP) as mock_uep: mock_uep.getEnvironmentList = env_list mock_uep.supports_resource = Mock(return_value=True) self.stub_cp_provider.basic_auth_cp = mock_uep rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="not_an_env") with nested(Capture(silent=True), self.assertRaises(SystemExit)): rc._get_environment_id(mock_uep, 'owner', None)
def test_get_environment_id_multi_available(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}, {"id": "5678", "name": "othername"}] 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="othername") env_id = rc._get_environment_id(cp, 'owner', None) expected = "5678" self.assertEquals(expected, env_id)
def test_get_environment_id_multi_available(self): def env_list(*args, **kwargs): return [{"id": "1234", "name": "somename"}, {"id": "5678", "name": "othername"}] with patch('rhsm.connection.UEPConnection', new_callable=StubUEP) as mock_uep: mock_uep.getEnvironmentList = env_list mock_uep.supports_resource = Mock(return_value=True) self.stub_cp_provider.basic_auth_cp = mock_uep rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="othername") env_id = rc._get_environment_id(mock_uep, 'owner', None) expected = "5678" 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
def test_get_environment_id_multi_available_bad_name(self): def env_list(*args, **kwargs): return [{ "id": "1234", "name": "somename" }, { "id": "5678", "name": "othername" }] with patch('rhsm.connection.UEPConnection', new_callable=StubUEP) as mock_uep: mock_uep.getEnvironmentList = env_list mock_uep.supports_resource = Mock(return_value=True) self.stub_cp_provider.basic_auth_cp = mock_uep rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="not_an_env") with nested(Capture(silent=True), self.assertRaises(SystemExit)): rc._get_environment_id(mock_uep, 'owner', None)
def test_get_environment_id_multi_available(self): def env_list(*args, **kwargs): return [{ "id": "1234", "name": "somename" }, { "id": "5678", "name": "othername" }] with patch('rhsm.connection.UEPConnection', new_callable=StubUEP) as mock_uep: mock_uep.getEnvironmentList = env_list mock_uep.supports_resource = Mock(return_value=True) self.stub_cp_provider.basic_auth_cp = mock_uep rc = RegisterCommand() rc.options = Mock() rc.options.activation_keys = None rc._prompt_for_environment = Mock(return_value="othername") env_id = rc._get_environment_id(mock_uep, 'owner', None) expected = "5678" self.assertEqual(expected, env_id)