def test_list_certifications(database):
# set up submission
sub = SubmissionFactory()
database.session.add(sub)
database.session.commit()
# set up certify history, make sure the empty one comes last in the list
cert_hist_empty = CertifyHistoryFactory(submission=sub,
created_at=datetime.utcnow() -
timedelta(days=1))
cert_hist = CertifyHistoryFactory(submission=sub)
database.session.add_all([cert_hist_empty, cert_hist])
database.session.commit()
# add some data to certified_files_history for the cert_history ID
sess = database.session
history_id = cert_hist.certify_history_id
sub_id = sub.submission_id
file_hist_1 = CertifiedFilesHistoryFactory(
certify_history_id=history_id,
submission_id=sub_id,
filename="/path/to/file_a.csv",
warning_filename="/path/to/warning_file_a.csv",
narrative="A has a narrative",
file_type=sess.query(FileType).filter_by(name='appropriations').one())
file_hist_2 = CertifiedFilesHistoryFactory(
certify_history_id=history_id,
submission_id=sub_id,
filename="/path/to/file_d2.csv",
warning_filename=None,
file_type=sess.query(FileType).filter_by(name='award').one())
file_hist_3 = CertifiedFilesHistoryFactory(
certify_history_id=history_id,
submission_id=sub_id,
filename=None,
warning_filename="/path/to/warning_file_cross_test.csv",
file_type=None)
database.session.add_all([file_hist_1, file_hist_2, file_hist_3])
database.session.commit()
json_response = fileHandler.list_certifications(sub)
response_dict = json.loads(json_response.get_data().decode('utf-8'))
assert len(response_dict["certifications"]) == 2
has_file_list = response_dict["certifications"][0]
empty_file_list = response_dict["certifications"][1]
# asserts for certification with files associated
assert len(has_file_list["certified_files"]) == 4
assert has_file_list["certified_files"][0]["is_warning"] is False
assert has_file_list["certified_files"][0]["filename"] == "file_a.csv"
assert has_file_list["certified_files"][0][
"narrative"] == "A has a narrative"
assert has_file_list["certified_files"][1]["is_warning"]
assert has_file_list["certified_files"][1]["narrative"] is None
# asserts for certification without files associated
assert len(empty_file_list["certified_files"]) == 0
def test_file_history_url(database, monkeypatch):
sub = SubmissionFactory()
database.session.add(sub)
database.session.commit()
# set up certify history so it works
cert_hist = CertifyHistoryFactory(submission=sub)
database.session.add(cert_hist)
database.session.commit()
file_hist = CertifiedFilesHistoryFactory(certify_history_id=cert_hist.certify_history_id,
submission_id=sub.submission_id, filename="/path/to/file_d2.csv",
warning_filename="/path/to/warning_file_cross.csv",
narrative=None, file_type_id=None)
database.session.add(file_hist)
database.session.commit()
s3_url_handler = Mock()
s3_url_handler.return_value.get_signed_url.return_value = 'some/url/here.csv'
monkeypatch.setattr(fileHandler, 'S3Handler', s3_url_handler)
# checking for local response to non-warning file
json_response = fileHandler.file_history_url(sub, file_hist.certified_files_history_id, False, True)
url = json.loads(json_response.get_data().decode('utf-8'))["url"]
assert url == "/path/to/file_d2.csv"
# local response to warning file
json_response = fileHandler.file_history_url(sub, file_hist.certified_files_history_id, True, True)
url = json.loads(json_response.get_data().decode('utf-8'))["url"]
assert url == "/path/to/warning_file_cross.csv"
# generic test to make sure it's reaching the s3 handler properly
json_response = fileHandler.file_history_url(sub, file_hist.certified_files_history_id, False, False)
url = json.loads(json_response.get_data().decode('utf-8'))["url"]
assert url == 'some/url/here.csv'