def inner(instance, user, *args):
account = AccountManager().get_user_account(user)
balance_before = account.balance()
fun(instance, user, *args)
balance_after = account.balance()
BalanceThresholds.process_transaction(project_id=user,
balance_before=balance_before,
balance_after=balance_after)
def handler(cls, **kwargs):
if AccountManager().get_user_account(
kwargs['project_id']).balance() >= 0:
return
# need to enable all_tenants because of this (read comments):
# https://bugs.launchpad.net/python-novaclient/+bug/1134382
[server.stop() for server in NovaClient.get_client().servers.list(
search_opts={'all_tenants': True,
'tenant_id': kwargs['project_id'],
'status': 'ACTIVE'})]
def setUp(self):
super(PaymentsTests, self).setUp()
self.stripe = stripe
self.stripe.api_key = settings.MERCHANT_SETTINGS['stripe']['API_KEY']
self.am = AccountManager()
self.card1_data = {
'number': '4242424242424242',
'exp_month': 12,
'exp_year': 2016,
'cvc': '123'
}
self.card1_params = {
'last4': '4242',
'email': '*****@*****.**',
'default': False,
'tenant_id': '1'
}
def balance(request):
return AccountManager().get_user_account(request.user.tenant_id).balance()
