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 Capture(silent=True):
with self.assertRaises(SystemExit):
rc._get_environment_id(mock_uep, 'owner', None)
def test_get_environment_id_one_available(self):
def env_list(*args, **kwargs):
return [{"id": "1234", "name": "somename"}]
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
env_id = rc._get_environment_id(mock_uep, 'owner', None)
expected = "1234"
self.assertEqual(expected, env_id)