harmacist/healthchecks
1
0
Fork 0
forked from GithubBackups/healthchecks
Code Issues Pull Requests Projects Releases Wiki Activity

Don't throw an exception if user's current project is unset.

Browse Source
This commit is contained in:
Pēteris Caune 2019-09-18 14:56:58 +03:00
parent accdfb637b
commit ca5e19fd2d
No known key found for this signature in database
GPG Key ID: E28D7679E9A9EDE2
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
hc/payments/tests/test_pricing.py
View File

@ -35,6 +35,15 @@ class PricingTestCase(BaseTestCase):
r = self.client.get("/pricing/") r = self.client.get("/pricing/")
self.assertContains(r, "To manage billing for this project") self.assertContains(r, "To manage billing for this project")
def test_it_handles_null_project(self):
self.profile.current_project = None
self.profile.save()
self.client.login(username="alice@example.org", password="password")
r = self.client.get("/pricing/")
self.assertContains(r, "Unlimited Team Members")
def test_it_shows_active_plan(self): def test_it_shows_active_plan(self):
self.sub = Subscription(user=self.alice) self.sub = Subscription(user=self.alice)
self.sub.subscription_id = "test-id" self.sub.subscription_id = "test-id"

7
hc/payments/views.py
View File

@ -16,9 +16,10 @@ def token(request):
def pricing(request): def pricing(request):
if request.user.is_authenticated and request.user != request.project.owner: if request.user.is_authenticated:
ctx = {"page": "pricing"} if request.project and request.project.owner != request.user:
return render(request, "payments/pricing_not_owner.html", ctx) ctx = {"page": "pricing"}
return render(request, "payments/pricing_not_owner.html", ctx)
# Don't use Subscription.objects.for_user method here, so a # Don't use Subscription.objects.for_user method here, so a
# subscription object is not created just by viewing a page. # subscription object is not created just by viewing a page.