Homework 17 ihor zakatsiura

hr/tests/test_employees.py

@@ -3,6 +3,7 @@
     Client,
     TestCase,
 )
+from django.core.files.uploadedfile import SimpleUploadedFile
 from django.urls import reverse
 
 from hr.models import Employee
@@ -93,3 +94,91 @@ def test_successful_employee_creation_message(self):
         self.assertEqual(len(messages), 1)
         self.assertEqual(str(messages[0]), 'Працівника успішно створено.')
 
+
+class EmployeeProfileViewTest(TestCase):
+    def setUp(self):
+        self.client = Client()
+        self.admin_user = EmployeeFactory(is_staff=True, is_superuser=True)
+
+        avatar_file = SimpleUploadedFile(
+            name='test_avatar.jpg',
+            content=b'\x47\x49\x46\x38\x39\x61\x01\x00\x01\x00\x80\x00',
+            content_type='image/jpeg'
+        )
+
+        cv_file = SimpleUploadedFile(
+            name='test_cv.pdf',
+            content=b'%PDF-1.4 test pdf content',
+            content_type='application/pdf'
+        )
+        self.employee = EmployeeFactory(avatar=avatar_file, cv=cv_file)
+
+        self.client.force_login(self.admin_user)
+        self.url = reverse('hr:employee_profile', kwargs={'pk': self.employee.pk})
+
+
+    def test_access_employee_profile(self):
+        response = self.client.get(self.url)
+        self.assertEqual(response.status_code, 200)
+        self.assertTemplateUsed(response, 'employee_profile.html')
+        self.assertEqual(response.context['employee'], self.employee)
+
+
+
+class EmployeeDeleteViewTest(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_to_delete = EmployeeFactory()
+        self.url = reverse('hr:employee_delete', kwargs={'pk': self.employee_to_delete.pk})
+
+    def test_delete_employee_by_admin(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_to_delete.pk).exists())
+
+    def test_delete_employee_by_non_admin(self):
+        self.client.force_login(self.non_admin_user)
+        response = self.client.post(self.url)
+        self.assertNotEqual(response.status_code, 302)
+        self.assertTrue(Employee.objects.filter(pk=self.employee_to_delete.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_to_update = EmployeeFactory(first_name='OldName')
+        self.position = PositionFactory()
+        self.url = reverse('hr:employee_update', kwargs={'pk': self.employee_to_update.pk})
+
+    def test_update_employee_by_admin(self):
+        self.client.force_login(self.admin_user)
+        updated_data = {
+            'username': self.employee_to_update.username,
+            'first_name': 'NewName',
+            'last_name': self.employee_to_update.last_name,
+            'email': self.employee_to_update.email,
+            'position': self.position.id,
+        }
+        response = self.client.post(self.url, updated_data)
+        self.assertEqual(response.status_code, 302)
+        self.employee_to_update.refresh_from_db()
+        self.assertEqual(self.employee_to_update.first_name, 'NewName')
+
+    def test_update_employee_by_non_admin(self):
+        self.client.force_login(self.non_admin_user)
+        updated_data = {
+            'username': self.employee_to_update.username,
+            'first_name': 'NewName',
+            'last_name': self.employee_to_update.last_name,
+            'email': self.employee_to_update.email,
+            'position': self.position.id,
+        }
+        response = self.client.post(self.url, updated_data)
+        self.assertNotEqual(response.status_code, 302)
+        self.employee_to_update.refresh_from_db()
+        self.assertNotEqual(self.employee_to_update.first_name, 'NewName')

hr/tests/test_employees_api.py

@@ -5,10 +5,11 @@
     APITestCase,
 )
 
-from hr.models import Employee
+from hr.models import Employee, Position
 from hr.tests.factories import (
     EmployeeFactory,
     PositionFactory,
+    DepartmentFactory,
 )
 
 
@@ -49,3 +50,55 @@ 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(is_staff=True, is_superuser=True)
+        self.client.force_authenticate(user=self.user)
+
+        self.department = DepartmentFactory()
+        self.position = PositionFactory(department=self.department)
+
+    def test_get_position_list(self):
+        url = reverse('api-hr:position-list')
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertTrue(len(response.data) >= 1)
+
+    def test_get_position_detail(self):
+        url = reverse('api-hr:position-detail', kwargs={'pk': self.position.pk})
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.data['id'], self.position.pk)
+
+    def test_create_position(self):
+        url = reverse('api-hr:position-list')
+        data = {
+            'title': 'New Position',
+            'department': self.department.pk,
+            'is_manager': False,
+            'is_active': True,
+            'job_description': 'Description for new position',
+            'monthly_rate': '1000.00',
+        }
+        response = self.client.post(url, data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+        self.assertTrue(Position.objects.filter(title='New Position').exists())
+
+
+    def test_update_position(self):
+        url = reverse('api-hr:position-detail', kwargs={'pk': self.position.pk})
+        data = {
+            'title': 'Updated Position Title',
+        }
+        response = self.client.patch(url, data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.position.refresh_from_db()
+        self.assertEqual(self.position.title, 'Updated Position Title')
+
+    def test_delete_position(self):
+        url = reverse('api-hr:position-detail', kwargs={'pk': self.position.pk})
+        response = self.client.delete(url)
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+        self.assertFalse(Position.objects.filter(pk=self.position.pk).exists())