Unit test for CompanyOfferListView

company/views.py

@@ -30,10 +30,10 @@ def get(self, request):
             return Response({"info": "false"})
 
 
-class CompanyOfferListView(View):
+class CompanyOfferListView(APIView):
     permission_classes = [IsAuthenticated]
 
-    def list(self, request, *args, **kwargs):
+    def get(self, request, *args, **kwargs):
         company = Company.objects.get(user=request.user)
 
         if company.user != self.request.user:

tests/test_company/test_views.py

@@ -1,7 +1,7 @@
 import pytest
-from tests.fixtures import user, user_no_available_companies
+from tests.fixtures import user, user_no_available_companies, company, user_second
 from rest_framework.test import force_authenticate, APIRequestFactory
-from company.views import UserCanMakeCompanyView
+from company.views import UserCanMakeCompanyView, CompanyOfferListView
 
 factory = APIRequestFactory()
 
@@ -28,3 +28,22 @@ def test_success_check_if_user_is_not_able_to_create_company_return_false(user_n
 
     assert response.status_code == 200
     assert response.data["info"] == "false"
+
+
+@pytest.mark.django_db
+def test_success_return_list_of_offers_for_specified_company(user, company):
+    view = CompanyOfferListView.as_view()
+    request = factory.get("/api/company/offer/")
+    force_authenticate(request, user=user)
+    response = view(request)
+
+    assert response.status_code == 200
+
+
+@pytest.mark.django_db
+def test_error_return_list_of_offers_for_specified_company_unauthenticated(company):
+    view = CompanyOfferListView.as_view()
+    request = factory.get("/api/company/offer/")
+    response = view(request)
+
+    assert response.status_code == 401