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')