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.
56 lines
2.1 KiB
56 lines
2.1 KiB
from hc.accounts.models import Project
|
|
from hc.api.models import Channel, Check
|
|
from hc.test import BaseTestCase
|
|
|
|
|
|
class SwitchChannelTestCase(BaseTestCase):
|
|
def setUp(self):
|
|
super(SwitchChannelTestCase, self).setUp()
|
|
self.check = Check.objects.create(project=self.project)
|
|
|
|
self.channel = Channel(project=self.project, kind="email")
|
|
self.channel.value = "[email protected]"
|
|
self.channel.save()
|
|
|
|
self.url = "/checks/%s/channels/%s/enabled" % (
|
|
self.check.code,
|
|
self.channel.code,
|
|
)
|
|
|
|
def test_it_enables(self):
|
|
self.client.login(username="[email protected]", password="password")
|
|
self.client.post(self.url, {"state": "on"})
|
|
|
|
self.assertTrue(self.channel in self.check.channel_set.all())
|
|
|
|
def test_it_disables(self):
|
|
self.check.channel_set.add(self.channel)
|
|
|
|
self.client.login(username="[email protected]", password="password")
|
|
self.client.post(self.url, {"state": "off"})
|
|
|
|
self.assertFalse(self.channel in self.check.channel_set.all())
|
|
|
|
def test_it_checks_ownership(self):
|
|
self.client.login(username="[email protected]", password="password")
|
|
r = self.client.post(self.url, {"state": "on"})
|
|
self.assertEqual(r.status_code, 404)
|
|
|
|
def test_it_checks_channels_ownership(self):
|
|
charlies_project = Project.objects.create(owner=self.charlie)
|
|
cc = Check.objects.create(project=charlies_project)
|
|
|
|
# Charlie will try to assign Alice's channel to his check:
|
|
self.url = "/checks/%s/channels/%s/enabled" % (cc.code, self.channel.code)
|
|
|
|
self.client.login(username="[email protected]", password="password")
|
|
r = self.client.post(self.url, {"state": "on"})
|
|
self.assertEqual(r.status_code, 400)
|
|
|
|
def test_it_allows_cross_team_access(self):
|
|
self.bobs_profile.current_project = None
|
|
self.bobs_profile.save()
|
|
|
|
self.client.login(username="[email protected]", password="password")
|
|
r = self.client.post(self.url, {"state": "on"})
|
|
self.assertEqual(r.status_code, 200)
|