Unit test for UserCanMakeCompanyView

tests/fixtures.py

@@ -6,23 +6,35 @@
 
 @pytest.fixture
 def user():
-    return UserAccount.objects.create_user(
+    return UserAccount.objects.create(
         first_name="John",
         last_name="Doe",
         email="[email protected]",
         password="XXXXXXXXXXX",
     )
 
+
 @pytest.fixture
 def user_second():
-    return UserAccount.objects.create_user(
+    return UserAccount.objects.create(
         first_name="Jane",
         last_name="Doe",
         email="[email protected]",
         password="XXXXXXXXXXX",
     )
 
 
+@pytest.fixture
+def user_no_available_companies():
+    return UserAccount.objects.create(
+        first_name="John",
+        last_name="Doe",
+        email="[email protected]",
+        password="XXXXXXXXXXX",
+        num_of_available_companies=0
+    )
+
+
 @pytest.fixture
 def job_offer(user):
     return JobOffer.objects.create(
@@ -44,4 +56,4 @@ def job_offer_with_company(user, company):
     return JobOffer.objects.create(
         title="Test Job Offer",
         company=company
-    )
+    )

tests/test_candidate/test_views.py

@@ -1,7 +1,7 @@
 import pytest
-from tests.fixtures import user, job_offer, job_offer_with_company, company, user_second
+from tests.fixtures import user, job_offer, job_offer_with_company, user_second, company
 from rest_framework.test import force_authenticate, APIRequestFactory
-from candidate.views import CandidateViewSet, CandidateListView, ChangeCandidateStatus
+from candidate.views import CandidateViewSet, CandidateListView
 import json
 from candidate.models import Candidate
 
@@ -144,4 +144,4 @@ def test_error_return_list_of_candidates_unauthorized_user(user_second, job_offe
     force_authenticate(request, user_second)
     response = view(request, offer_id=1)
 
-    assert response.status_code == 401
+    assert response.status_code == 401

tests/test_company/test_views.py

@@ -1 +1,30 @@
 import pytest
+from tests.fixtures import user, user_no_available_companies
+from rest_framework.test import force_authenticate, APIRequestFactory
+from company.views import UserCanMakeCompanyView
+
+factory = APIRequestFactory()
+
+
+@pytest.mark.django_db
+def test_success_check_if_user_is_able_to_create_company_return_true(user):
+    view = UserCanMakeCompanyView.as_view()
+
+    request = factory.get('/api/company/user/check/new')
+    force_authenticate(request, user=user)
+    response = view(request)
+
+    assert response.status_code == 200
+    assert response.data["info"] == "true"
+
+
+@pytest.mark.django_db
+def test_success_check_if_user_is_not_able_to_create_company_return_false(user_no_available_companies):
+    view = UserCanMakeCompanyView.as_view()
+
+    request = factory.get('/api/company/user/check/new')
+    force_authenticate(request, user=user_no_available_companies)
+    response = view(request)
+
+    assert response.status_code == 200
+    assert response.data["info"] == "false"