|
|
- 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(user=self.alice, project=self.project)
- self.check.save()
-
- self.channel = Channel(user=self.alice, kind="email")
- self.channel.project = self.project
- 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(user=self.charlie, project=charlies_project)
- cc.save()
-
- # 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_team = 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)
|
