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

remove channel doesn't crash

Browse Source
This commit is contained in:
Di Wu 2016-01-19 06:07:18 -08:00
parent 96b41ded9b
commit 01a9505cc7
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hc/front/tests/test_remove_channel.py
View File

@ -45,4 +45,4 @@ class RemoveChannelTestCase(BaseTestCase):
self.client.login(username="alice@example.org", password="password")
r = self.client.post(url)
assert r.status_code == 404
assert r.status_code == 302

10
hc/front/views.py
View File

@ -336,11 +336,11 @@ def verify_email(request, code, token):
def remove_channel(request, code):
assert request.method == "POST"
channel = get_object_or_404(Channel, code=code)
if channel.user != request.user:
return HttpResponseForbidden()
channel.delete()
channel = Channel.objects.filter(code=code).first()
if channel:
if channel.user != request.user:
return HttpResponseForbidden()
channel.delete()
return redirect("hc-channels")