|
|
- from django.test import override_settings
-
- from hc.payments.models import Subscription
- from hc.test import BaseTestCase
-
-
- class PricingTestCase(BaseTestCase):
- def test_anonymous(self):
- r = self.client.get("/pricing/")
- self.assertContains(r, "Unlimited Team Members", status_code=200)
-
- # A subscription object should have NOT been created
- assert Subscription.objects.count() == 0
-
- def test_authenticated(self):
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.get("/pricing/")
- self.assertContains(r, "Unlimited Team Members", status_code=200)
-
- # A subscription object still should have NOT been created
- assert Subscription.objects.count() == 0
-
- @override_settings(USE_PAYMENTS=True)
- def test_pricing_is_visible_for_all(self):
- for email in ("[email protected]", "[email protected]"):
- self.client.login(username=email, password="password")
-
- r = self.client.get("/docs/")
- self.assertContains(r, "Pricing")
-
- def test_it_offers_to_switch(self):
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.get("/pricing/")
- self.assertContains(r, "To manage billing for this project")
-
- def test_it_shows_active_plan(self):
- self.sub = Subscription(user=self.alice)
- self.sub.subscription_id = "test-id"
- self.sub.plan_id = "P20"
- self.sub.plan_name = "Business ($20 / month)"
- self.sub.save()
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.get("/pricing/")
- self.assertContains(r, "Business ($20 / month)", status_code=200)
|
