from django.contrib.auth import get_user_model from django.test import TestCase, Client from django.urls import reverse import access_controller.settings as sets class ControlAccessTests(TestCase): fixtures = ['test_data_base.json'] def setUp(self) -> None: self.User = get_user_model() self.client = Client() def test_admin_has_perm(self): self.client.force_login(self.User.objects.get(email='admin@gmail.com')) self.response = self.client.get(reverse('control')) self.assertEqual(self.response.status_code, 200) def test_engineer_dont_have_perm(self): self.client.force_login(self.User.objects.get(email='123@test.ru')) self.response = self.client.get(reverse('control')) self.assertEqual(self.response.status_code, 403)