def test_set_active_subscription(self): storage_mock = {'subscriptions': None} profile = Profile(storage_mock) consolidated = Profile._normalize_properties(self.user1, [self.subscription1], False, ENV_DEFAULT) profile._set_subscriptions(consolidated) consolidated = profile._normalize_properties(self.user2, [self.subscription2], False, ENV_DEFAULT) profile._set_subscriptions(consolidated) subscription1 = storage_mock['subscriptions'][0] subscription2 = storage_mock['subscriptions'][1] self.assertTrue(subscription2['isDefault']) profile.set_active_subscription(subscription1['id']) self.assertFalse(subscription2['isDefault']) self.assertTrue(subscription1['isDefault'])
def set_active_subscription(subscription_name_or_id): '''Set the current subscription''' if not id: raise CLIError('Please provide subscription id or unique name.') profile = Profile() profile.set_active_subscription(subscription_name_or_id)