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)