def test_valid_referral(self):
with transaction.manager:
referrer = User(username='******', password='******')
self.session.add(referrer)
self.assertFalse(referrer.is_paid)
req = DummyRequest(post={
'username': '******',
'password': '******',
'password2': 'newpw',
'email': 'email@host'
}, params={
'ref': str(referrer.id),
})
resp = views.account.signup(req)
newuser = self.session.query(User).filter_by(username='******').first()
self.assertIsInstance(resp, httpexceptions.HTTPSeeOther)
self.assertTrue(resp.location.endswith('/account/'))
self.assertEqual(newuser.referrer_id, referrer.id)
self.assertFalse(referrer.is_paid)
with transaction.manager:
testorder = Order(user=newuser, amount=1,
method=Order.METHOD.BITCOIN,
time=datetime.timedelta(days=30))
self.session.add(testorder)
testorder.close(force=True)
self.assertTrue(referrer.is_paid)
def test_valid_referral(self):
with transaction.manager:
_referrer = User(username='******', password='******')
self.session.add(_referrer)
referrer = self.session.query(User).filter_by(username='******').first()
self.assertFalse(referrer.is_paid)
req = DummyRequest(post={
'username': '******',
'password': '******',
'password2': 'newpw',
'email': 'email@host'
},
params={
'ref': str(referrer.id),
})
resp = views.account.signup(req)
newuser = self.session.query(User).filter_by(username='******') \
.first()
self.assertIsInstance(resp, httpexceptions.HTTPSeeOther)
self.assertTrue(resp.location.endswith('/account/'))
self.assertEqual(newuser.referrer_id, referrer.id)
self.assertFalse(referrer.is_paid)
testorder = Order(user=newuser,
amount=1,
method=Order.METHOD.BITCOIN,
time=datetime.timedelta(days=30))
self.session.add(testorder)
self.session.add(referrer)
self.session.flush()
testorder.close(force=True)
self.session.flush()
self.session.refresh(referrer)
self.assertTrue(referrer.is_paid)
