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 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_ros_form_data_not_complete(self):
incomplete_ros_form = create_mock_ros_form(employee=create_mock_user(),
has_image=False)
self.assertEqual(incomplete_ros_form.data_complete, False)
def test_ros_form_data_complete(self):
complete_ros_form = create_mock_ros_form(employee=create_mock_user(),
has_image=True)
self.assertEqual(complete_ros_form.data_complete, True)
def test_ros_form_is_expired(self):
expired_ros_form = create_mock_ros_form(employee=create_mock_user(),
expired=True)
self.assertEqual(expired_ros_form.expired, True)
self.assertTrue(
expired_ros_form.expiration_date < datetime.now().date())
def test_ros_form_is_not_expired(self):
valid_ros_form = create_mock_ros_form(employee=create_mock_user(),
has_image=False)
self.assertEqual(valid_ros_form.expired, False)
self.assertTrue(valid_ros_form.expiration_date > datetime.now().date())
def setUp(self):
self.employee = create_mock_user()
self.ros_form = create_mock_ros_form(self.employee)