diff --git a/hc/payments/tests/test_pricing.py b/hc/payments/tests/test_pricing.py index a776fc4c..2c6970e6 100644 --- a/hc/payments/tests/test_pricing.py +++ b/hc/payments/tests/test_pricing.py @@ -1,3 +1,5 @@ +from django.test import override_settings + from hc.payments.models import Subscription from hc.test import BaseTestCase @@ -20,7 +22,9 @@ class PricingTestCase(BaseTestCase): # A subscription object still should have NOT been created assert Subscription.objects.count() == 0 + @override_settings(USE_PAYMENTS=True) def test_pricing_is_hidden_for_team_members(self): + self.client.login(username="bob@example.org", password="password") r = self.client.get("/about/") @@ -28,6 +32,7 @@ class PricingTestCase(BaseTestCase): # Alice's team, but is not its owner. self.assertNotContains(r, "Pricing") + @override_settings(USE_PAYMENTS=True) def test_pricing_is_visible_for_team_owners(self): self.client.login(username="alice@example.org", password="password") diff --git a/hc/payments/views.py b/hc/payments/views.py index b80aaa2c..e6a3aff1 100644 --- a/hc/payments/views.py +++ b/hc/payments/views.py @@ -5,7 +5,6 @@ from django.http import (HttpResponseBadRequest, HttpResponseForbidden, JsonResponse) from django.shortcuts import redirect, render from django.views.decorators.http import require_POST -from hc.accounts.models import Profile from .models import Subscription