Hw14 tymofieieva

hr/tests/test_employees.py

@@ -1,4 +1,5 @@
 from django.contrib.messages import get_messages
+from django.core.files.uploadedfile import SimpleUploadedFile
 from django.test import (
     Client,
     TestCase,
@@ -93,3 +94,75 @@ def test_successful_employee_creation_message(self):
         self.assertEqual(len(messages), 1)
         self.assertEqual(str(messages[0]), 'Працівника успішно створено.')
 
+
+class EmployeeProfileView(TestCase):
+    def setUp(self):
+        self.client = Client()
+        self.employees = EmployeeFactory.create_batch(10)
+        self.employee = self.employees[0]
+        self.employee.avatar = SimpleUploadedFile("test.jpg", b"content", content_type="image/jpeg")
+        self.employee.cv = SimpleUploadedFile("test.pdf", b"fake pdf content", content_type="application/pdf")
+        self.employee.save()
+        self.admin_user = EmployeeFactory(is_staff=True, is_superuser=True)
+        self.non_admin_user = EmployeeFactory(is_staff=False, is_superuser=False)
+        self.url = reverse('hr:employee_profile', kwargs={'pk': self.employee.pk})
+
+    def test_access_employee_view(self):
+        self.client.force_login(self.admin_user)
+        response = self.client.get(self.url)
+        self.assertEqual(response.status_code, 200)
+
+    def test_employee_view_content(self):
+        self.client.force_login(self.admin_user)
+        response = self.client.get(self.url )
+      #  self.assertTrue('employee' in response.context)
+        self.assertContains(response, self.employee.first_name)
+        self.assertContains(response, self.employee.last_name)
+        self.assertContains(response, self.employee.email)
+
+    def test_access_employee_view_if_not_superuser(self):
+        self.client.force_login(self.non_admin_user)
+        response = self.client.get(self.url)
+        self.assertEqual(response.status_code, 403)
+
+
+class EmployeeDeleteView(TestCase):
+    def setUp(self):
+        self.client = Client()
+        self.admin_user = EmployeeFactory(is_staff=True, is_superuser=True)
+        self.non_admin_user = EmployeeFactory(is_staff=False, is_superuser=False)
+        self.employee = EmployeeFactory()
+  #      self.position = PositionFactory()
+        self.url = reverse('hr:employee_delete', kwargs={'pk': self.employee.pk})
+
+    def test_successful_employee_delete(self):
+        self.client.force_login(self.admin_user)
+        response = self.client.post(self.url)
+        self.assertEqual(response.status_code, 302)
+        self.assertFalse(Employee.objects.filter(pk=self.employee.pk).exists())
+
+
+class EmployeeUpdateViewTest(TestCase):
+    def setUp(self):
+        self.client = Client()
+        self.admin_user = EmployeeFactory(is_staff=True, is_superuser=True)
+        self.non_admin_user = EmployeeFactory(is_staff=False, is_superuser=False)
+        self.employee = EmployeeFactory()
+     #   self.position = PositionFactory()
+        self.url = reverse('hr:employee_update', kwargs={'pk': self.employee.pk})
+
+    def test_successful_employee_update(self):
+        self.client.force_login(self.admin_user)
+        new_first_name = 'Klod'
+        response = self.client.post(self.url,{'first_name': new_first_name, 'username': self.employee.username})
+        self.employee.refresh_from_db()
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(self.employee.first_name, new_first_name)
+
+    def test_update_employee_by_non_admin(self):
+        self.client.force_login(self.non_admin_user)
+        new_first_name = 'LLLLLL'
+        response = self.client.post(self.url,{'first_name': new_first_name, 'username': self.employee.username})
+        self.employee.refresh_from_db()
+        self.assertEqual(response.status_code, 403)
+        self.assertNotEqual(self.employee.first_name, new_first_name)

hr/tests/test_employees_api.py

@@ -5,10 +5,10 @@
     APITestCase,
 )
 
-from hr.models import Employee
+from hr.models import Employee, Department
 from hr.tests.factories import (
     EmployeeFactory,
-    PositionFactory,
+    PositionFactory, DepartmentFactory,
 )
 
 
@@ -49,3 +49,41 @@ def test_search_employee(self):
         response = self.client.get(reverse('api-hr:employee-list'), {'search': 'Test'})
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
+
+
+
+class PositionViewSetTestCase(APITestCase):
+    def setUp(self):
+        self.client = APIClient()
+        self.user = EmployeeFactory()
+        self.position = PositionFactory()
+        self.department = DepartmentFactory()
+        self.client.force_authenticate(user=self.user)
+
+    def test_get_position_list(self):
+        response = self.client.get(reverse('api-hr:position-list'))
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+    def test_create_position(self):
+        data = {
+            'title': "new_position",
+            'department': self.department.pk,
+            'is_manager': False,
+            'is_active': True,
+            'job_description': "description",
+            'monthly_rate': 20000
+        }
+        response = self.client.post(reverse('api-hr:position-list'), data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+    def test_update_position(self):
+        position = PositionFactory(department=self.department)
+        data = {'title': 'Updated Title'}
+        response = self.client.patch(reverse('api-hr:position-detail', kwargs={'pk': position.pk}), data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+    def test_delete_employee(self):
+        position = PositionFactory(department=self.department)
+        response = self.client.delete(reverse('api-hr:position-detail', kwargs={'pk': position.pk}))
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+