def test_use_reuse(self):
time = datetime.timedelta(days=30, hours=11)
gc = GiftCode(time=time)
gc.use(self.u)
self.assertEqual(self.u.paid_time_left.days, time.days)
self.assertRaises(models.AlreadyUsedGiftCode, gc.use, self.u)
self.assertEqual(self.u.paid_time_left.days, time.days)
gc.use(self.u, reuse=True)
self.assertTrue(self.u.is_paid)
self.assertEqual(self.u.paid_time_left.days, time.days * 2)
def setUp(self):
super().setUp()
self.testuser = User(username='******', password='******')
self.session.add(self.testuser)
self.testadmin = User(username='******', password='******')
self.testadmin.is_admin = True
self.session.add(self.testadmin)
self.session.flush()
self.testcode = GiftCode(datetime.timedelta(days=7))
self.session.add(self.testcode)
self.session.flush()
def setUp(self):
settings = {'mako.directories': 'ccvpn:templates/'}
self.config = testing.setUp(settings=settings)
setup_routes(self.config)
self.session = setup_database()
self.testuser = User(username='******', password='******')
self.session.add(self.testuser)
self.testadmin = User(username='******', password='******')
self.testadmin.is_admin = True
self.session.add(self.testadmin)
self.session.flush()
self.testcode = GiftCode(datetime.timedelta(days=7))
self.session.add(self.testcode)
self.session.flush()
def test_use_freeonly(self):
gc = GiftCode()
gc.free_only = True
self.assertRaises(models.AlreadyUsedGiftCode, gc.use, self.pu)
gc.use(self.u)
self.assertTrue(self.u.is_paid)
def test_username_if_used(self):
gc = GiftCode()
self.assertIs(gc.username_if_used, False)
gc.used = self.u.id
gc.user = self.u
self.assertEqual(gc.username_if_used, self.u.username)