You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.2 KiB
38 lines
1.2 KiB
from mock import patch
|
|
|
|
from hc.accounts.models import Profile
|
|
from hc.payments.models import Subscription
|
|
from hc.test import BaseTestCase
|
|
|
|
|
|
class CancelPlanTestCase(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(CancelPlanTestCase, self).setUp()
|
|
self.sub = Subscription(user=self.alice)
|
|
self.sub.subscription_id = "test-id"
|
|
self.sub.plan_id = "P5"
|
|
self.sub.save()
|
|
|
|
self.profile.ping_log_limit = 1000
|
|
self.profile.check_limit = 500
|
|
self.profile.sms_limit = 50
|
|
self.profile.save()
|
|
|
|
@patch("hc.payments.models.braintree")
|
|
def test_it_works(self, mock_braintree):
|
|
|
|
self.client.login(username="[email protected]", password="password")
|
|
r = self.client.post("/pricing/cancel_plan/")
|
|
self.assertRedirects(r, "/pricing/")
|
|
|
|
self.sub.refresh_from_db()
|
|
self.assertEqual(self.sub.subscription_id, "")
|
|
self.assertEqual(self.sub.plan_id, "")
|
|
|
|
# User's profile should have standard limits
|
|
profile = Profile.objects.get(user=self.alice)
|
|
self.assertEqual(profile.ping_log_limit, 100)
|
|
self.assertEqual(profile.check_limit, 20)
|
|
self.assertEqual(profile.team_limit, 2)
|
|
self.assertEqual(profile.sms_limit, 0)
|