def test_set_duplicate_multi_environment(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.cp = mock_uep rc.options = Mock() rc.options.activation_keys = None rc.options.environments = None rc._prompt_for_environment = Mock( return_value="somename,othername,somename") with Capture(silent=True): with self.assertRaises(SystemExit): rc._process_environments(mock_uep, "owner")
def test_set_multi_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) mock_uep.has_capability = Mock(return_value=True) self.stub_cp_provider.basic_auth_cp = mock_uep rc = RegisterCommand() rc.cp = mock_uep rc.options = Mock() rc.options.activation_keys = None rc.options.environments = None rc._prompt_for_environment = Mock( return_value="somename,othername") env_id = rc._process_environments(mock_uep, "owner") expected = "1234,5678" self.assertEqual(expected, env_id)
def test_validate_multi_environment_with_activation_key(self): with patch("rhsm.connection.UEPConnection", new_callable=StubUEP) as mock_uep: mock_uep.supports_resource = Mock(return_value=True) self.stub_cp_provider.basic_auth_cp = mock_uep rc = RegisterCommand() rc.cp = mock_uep rc.options = Mock() rc.options.activation_keys = "someAK" rc.options.environments = None ret = rc._process_environments(mock_uep, "owner") self.assertIsNone(ret)
def test_get_environment_id_none_available(self): def env_list(*args, **kwargs): return [] 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.options.environments = None env_id = rc._process_environments(mock_uep, "owner") expected = None self.assertEqual(expected, env_id)