def test_notify_subscription(self):
out = StringIO()
u = User.objects.create_user('test_username', '*****@*****.**', 'testpw')
u.vpnuser.add_paid_time(timedelta(days=2))
u.vpnuser.save()
s = Subscription(user=u, backend_id='paypal', status='active')
s.save()
call_command('expire_notify', stdout=out)
self.assertEqual(len(mail.outbox), 0)
u = User.objects.get(username='******')
def test_notify_subscription(self):
out = StringIO()
u = User.objects.create_user('test_username', '*****@*****.**', 'testpw')
u.vpnuser.add_paid_time(timedelta(days=2))
u.vpnuser.save()
s = Subscription(user=u, backend_id='paypal', status='active')
s.save()
call_command('expire_notify', stdout=out)
self.assertEqual(len(mail.outbox), 0)
u = User.objects.get(username='******')
def test_paid_between_subscr_payments(self):
u = User.objects.get(username='******')
vu = u.vpnuser
s = Subscription(user=u, backend_id='paypal', status='new')
s.save()
self.assertFalse(vu.is_paid)
s.status = 'active'
s.save()
self.assertTrue(vu.is_paid)
def test_profile_no_active_subscription(self):
subscription = Subscription()
subscription.user = self.user
subscription.state = Subscription.STATE_CANCELLED
subscription.save()
response = self.client.get(reverse('shortener_app:profile'))
self.assertTemplateUsed(response, 'shortener_app/profile.html')
self.assertContains(response, "Price")
self.assertEquals(response.status_code, 200)
def test_profile_active_subscription(self):
subscription = Subscription()
subscription.user = self.user
subscription.state = Subscription.STATE_ACTIVE
subscription.save()
response = self.client.get(reverse('shortener_app:profile'))
self.assertTemplateUsed(response, 'shortener_app/profile.html')
self.assertContains(response, "You have premium account plan")
self.assertEquals(response.status_code, 200)
def test_upgrade_to_premium_user_has_cancelled_subscription(self):
subscription = Subscription()
subscription.user = self.user
subscription.state = Subscription.STATE_CANCELLED
subscription.save()
response = self.client.get(reverse('shortener_app:upgrade_to_premium'))
self.assertTemplateUsed(response,
'shortener_app/upgrade_to_premium.html')
self.assertContains(response, "Price")
self.assertEquals(response.status_code, 200)
def test_paid_between_subscr_payments(self):
u = User.objects.get(username='******')
vu = u.vpnuser
s = Subscription(user=u, backend_id='paypal', status='new')
s.save()
self.assertFalse(vu.is_paid)
s.status = 'active'
s.save()
self.assertTrue(vu.is_paid)
