def test_get_org_missing(self):
config = CliRuntime()
config.keychain = mock.Mock()
config.keychain.get_org.return_value = None
with self.assertRaises(click.UsageError):
org_name, org_config_result = config.get_org("test", fail_if_missing=True)
def test_get_org(self):
config = CliRuntime()
config.keychain = mock.Mock()
config.keychain.get_org.return_value = org_config = OrgConfig({}, "test")
org_name, org_config_result = config.get_org("test")
self.assertEqual("test", org_name)
self.assertIs(org_config, org_config_result)
