def create_many_users_and_documents(self, num_users=15):
for x in range(num_users):
if x == 0:
user = Employee.objects.get(email='*****@*****.**')
else:
user = create_mock_user()
if x % 5 == 0: print(x)
self.users.append(user)
create_mock_passport(
user,
has_image=random.choice([True, True, False]),
expired=random.choice([True, False, False, False, False, False])
)
create_mock_ros_form(
user,
has_image=random.choice([True, True, False]),
expired=random.choice([True, False, False, False, False, False])
)
create_mock_work_permit(
user,
has_image=random.choice([True, True, False]),
expired=random.choice([True, False, False, False, False, False])
)
create_mock_achievement_certificate(user)
create_mock_degree(user)
create_mock_teaching_certificate(user)
create_mock_resume(user)
def test_expired_work_permit_is_expired(self):
expired_work_permit = create_mock_work_permit(create_mock_user(),
expired=True)
self.assertEqual(
expired_work_permit.expired, True,
('exp. date', expired_work_permit.expiration_date.isoformat(),
'curr date', datetime.now().date().isoformat()))
self.assertTrue(
expired_work_permit.expiration_date < datetime.now().date())
def test_valid_work_permit_is_not_expired(self):
valid_work_permit = create_mock_work_permit(create_mock_user(),
expired=False,
has_image=False)
self.assertEqual(valid_work_permit.expired, False,
valid_work_permit.expired)
self.assertTrue(
valid_work_permit.expiration_date > datetime.now().date(),
('exp. date', valid_work_permit.expiration_date.isoformat(),
'curr date', datetime.now().date().isoformat()))
def assign_documents_to_user(user):
create_mock_passport(
user,
has_image=random.choice([True, True, False]),
expired=random.choice([True, False, False, False, False, False])
)
create_mock_ros_form(
user,
has_image=random.choice([True, True, False]),
expired=random.choice([True, False, False, False, False, False])
)
create_mock_work_permit(
user,
has_image=random.choice([True, True, False]),
expired=random.choice([True, False, False, False, False, False])
)
create_mock_achievement_certificate(user)
create_mock_degree(user)
create_mock_teaching_certificate(user)
create_mock_resume(user)
def test_work_permit_data_not_complete(self):
incomplete_work_permit = create_mock_work_permit(create_mock_user(),
has_image=False)
self.assertEqual(incomplete_work_permit.data_complete, False)
def test_work_permit_data_complete(self):
complete_work_permit = create_mock_work_permit(
employee=create_mock_user(), expired=False, has_image=True)
self.assertEqual(complete_work_permit.data_complete, True)
def setUp(self):
self.employee = create_mock_user()
self.work_permit = create_mock_work_permit(self.employee)