from rest_framework.test import APIRequestFactory, APITestCase
from model_mommy import mommy
from api.v1 import views
from company.models.models import CompanyUserProfile, ManagerRelation, DynamicRelation
User = get_user_model()
class APIViewsTestCase(APITestCase):
@classmethod
def setUpTestData(cls):
supervisory_id = "1337"
emp_group_id = "119"
prd_scope_id = "1334"
user = mommy.make(User)
setattr(cls, 'user', user)
company_user_profile_1 = mommy.make(CompanyUserProfile)
requestor = mommy.make(
CompanyUserProfile, is_manager=True,
supervisory_id=supervisory_id, emp_group_id=emp_group_id,
prd_scope_id=prd_scope_id, user=user
)
cls.user = user
manager_relation = mommy.make(
ManagerRelation, manager_id=requestor.id, mgr_type="Direct Mgr", _quantity=5 )
dynamic_relations = mommy.make(
)
def setUp(self) -> None:
super().setUp()
self.client.force_authenticate(user=self.user)
def test_employee_view(self):
CompanyUserProfile.objects.create(
guid=uuid.uuid4(),
comp_user_id='idcbijnccdocwocd', comp_username='djchbdcjnwdskjcn',
id=3333
)
print('cup count:', CompanyUserProfile.objects.count())
print('user count:', User.objects.count())
print('m_rels count:', ManagerRelation.objects.count())
print('d_rels count:', DynamicRelation.objects.count())
rsp = self.client.get('/api/v1/employees/')
self.assertEqual(rsp.status_code, HTTPStatus.OK)
-- test run output: python manage.py test
Creating test database for alias 'default'...
System check identified no issues (0 silenced).
cup count: 0
user count: 1
m_rels count: 5
d_rels count: 5
Destroying test database for alias 'default'...