def test_find_users_by_days_no_users(self):
'''Testing where there are no users with expired certifications'''
db = CertTrackerDatabase(USER, PASSWORD, HOST, PORT, DATABASE)
db.fetch_data()
users = db.get_users()
date = datetime(2019,5,29)
lab_users, pis = find_users_by_days(users,date.date())
self.assertEqual(len(lab_users), 0)
def test_find_users_by_type_30_days_1_user(self):
'''Testing 1 user with certification expiring in 30 days'''
with open('lfs_lab_cert_tracker/fixtures/test_user_certs.json') as f:
d = json.load(f)
dateExpire = datetime.now() + timedelta(days=30)
dateCompletion = datetime.now() + timedelta(days=30) - timedelta(days=1825)
data = {
"model": "lfs_lab_cert_tracker.usercert",
"pk": 17,
"fields": {
"expiry_date": dateExpire.strftime('%Y-%m-%d'),
"cert_id": 14,
"user_id": 14,
"cert_file": "users/5/certificates/23/certificate_1.jpg",
"uploaded_date": "2018-06-11",
"completion_date": dateCompletion.strftime('%Y-%m-%d')
}
}
d.append(data)
json_data = json.dumps(d)
with open('lfs_lab_cert_tracker/fixtures/test_user_certs2.json', 'w') as g:
json.dump(d,g)
subprocess.run('python manage.py loaddata test_user_certs2')
db = CertTrackerDatabase(USER, PASSWORD, HOST, PORT, DATABASE)
db.fetch_data()
users = db.get_users()
lab_users, pis = find_users_by_type(users,1)
self.assertEqual(len(lab_users),1)
self.assertEqual(len(pis[4]),1)
dateExpire = datetime.now() + timedelta(days=365)
dateCompletion = datetime.now() + timedelta(days=365) - timedelta(days=1825)
data = {
"model": "lfs_lab_cert_tracker.usercert",
"pk": 17,
"fields": {
"expiry_date": dateExpire.strftime('%Y-%m-%d'),
"cert_id": 14,
"user_id": 14,
"cert_file": "users/5/certificates/23/certificate_1.jpg",
"uploaded_date": "2018-06-11",
"completion_date": dateCompletion.strftime('%Y-%m-%d')
}
}
d.append(data)
json_data = json.dumps(d)
with open('lfs_lab_cert_tracker/fixtures/test_user_certs2.json', 'w') as g:
json.dump(d,g)
subprocess.run('python manage.py loaddata test_user_certs2')
