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)