def test_get_document_by_id(session):
"""Verify that the get_document_by_id returns a document provided an id."""
docs = [
Documents(id=1110, label='shp doc', type=1, required=True),
Documents(id=2110, label='ath doc', type=1, required=True)
]
session.bulk_save_objects(docs)
session.commit()
assert Documents.get_document_by_id(1110)
assert Documents.get_document_by_id(2110)
assert Documents.get_document_by_id(79897777879) is None
def get_document_label(self, data):
"""Returns a document label."""
dereg_details = DeRegDetails.get_by_id(data.dereg_id)
upload_dir_path = GLOBAL_CONF['upload_directory']
document = Documents.get_document_by_id(data.document_id)
data.label = document.label
data.required = Documents.required
data.link = '{server_dir}/{local_dir}/{file_name}'.format(
server_dir=upload_dir_path,
local_dir=dereg_details.tracking_id,
file_name=data.filename)