def test_get_org_missing(self):
config = CliConfig()
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_default(self):
config = CliConfig()
config.keychain = mock.Mock()
org_config = OrgConfig({}, "test")
config.keychain.get_default_org.return_value = ("test", org_config)
org_name, org_config_result = config.get_org()
self.assertEqual("test", org_name)
self.assertIs(org_config, org_config_result)
def test_check_org_expired(self, confirm):
config = CliConfig()
config.keychain = mock.Mock()
org_config = OrgConfig(
{
"scratch": True,
"date_created": date.today() - timedelta(days=2),
"expired": True,
},
"test",
)
confirm.return_value = True
config.check_org_expired("test", org_config)
config.keychain.create_scratch_org.assert_called_once()
def test_check_org_expired_decline(self, confirm):
config = CliConfig()
config.keychain = mock.Mock()
org_config = OrgConfig(
{
"scratch": True,
"date_created": date.today() - timedelta(days=2),
"expired": True,
},
"test",
)
confirm.return_value = False
with self.assertRaises(click.ClickException):
config.check_org_expired("test", org_config)