def test_view_paypal(self): testorder = Order(user=self.testuser.id, amount=1, method=Order.METHOD.PAYPAL, time=datetime.timedelta(days=30)) self.session.add(testorder) self.session.flush() req = DummyRequest() req.session['uid'] = self.testuser.id req.matchdict['hexid'] = '%x' % testorder.id resp = views.order.order_view(req) self.assertIsInstance(resp, dict) self.assertEqual(resp['o'], testorder)
def test_view_btc(self): testorder = Order(user=self.testuser.id, amount=1, method=Order.METHOD.BITCOIN, time=datetime.timedelta(days=30)) testorder.payment = {'btc_address': 'TESTADDRESS'} self.session.add(testorder) self.session.flush() req = DummyRequest() req.session['uid'] = self.testuser.id req.matchdict['hexid'] = '%x' % testorder.id resp = views.order.order_view(req) self.assertIsInstance(resp, dict) self.assertEqual(resp['o'], testorder)
def test_view_not_owned(self): otheruser = User(username='******', password='******') self.session.add(otheruser) self.session.flush() testorder = Order(user=otheruser.id, amount=1, method=Order.METHOD.PAYPAL, time=datetime.timedelta(days=30)) self.session.add(testorder) self.session.flush() req = DummyRequest() req.session['uid'] = self.testuser.id req.matchdict['hexid'] = '%x' % testorder.id resp = views.order.order_view(req) self.assertIsInstance(resp, httpexceptions.HTTPUnauthorized)
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)