rloomans/new-teamtemp

from django.urls import reverse

from teamtemp.tests.view.admin_only_view_testcase import AdminOnlyViewTestCase


class AdminViewTestCases(AdminOnlyViewTestCase):
    def test_no_admin_set_view(self):
        set_url = reverse('set', kwargs={'survey_id': self.teamtemp.id})
        response = self.client.get(set_url, follow=True)
        self.assertDoesLoginRedirect(response, redirect_to=set_url)
        self.assertIsPasswordForm(response)

    def test_admin_set_view(self):
        self.setUpAdmin()
        response = self.client.get(
            reverse(
                'set',
                kwargs={
                    'survey_id': self.teamtemp.id}),
            follow=True)
        self.assertIsNotPasswordForm(response)
        self.assertTemplateUsed(response, 'set.html')
        self.assertContains(response, "Settings %s" % self.teamtemp.id)