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
