def test_get_pending_work_documents(self, db):
logger = Logger(FakeMemoryLog())
d = db.create_download(logger, "test-request-id", "123456789")
success_result_of(d)
d = db.mark_download_manifest_downloaded(logger, "test-request-id")
success_result_of(d)
doc = Document(
id="test-document-id",
download_id="test-request-id",
document_id="{ABCD}",
doc_type="00356",
filename="file.pdf",
received_at=datetime.datetime.utcnow(),
source="CUI",
content_location=None,
errored=False,
)
d = db.create_documents(logger, [doc])
success_result_of(d)
d = db.get_pending_work(logger)
downloads, documents = success_result_of(d)
assert downloads == []
[document] = documents
assert document.id == "test-document-id"
d = db.set_document_content_location(logger, doc, "/path/to/content")
success_result_of(d)
d = db.get_pending_work(logger)
assert success_result_of(d) == ([], [])
def test_set_document_content_location(self, db):
logger = Logger(FakeMemoryLog())
d = db.create_download(logger, "test-request-id", "123456789")
success_result_of(d)
doc = Document(
id="test-document-id",
download_id="test-request-id",
document_id="{ABCD}",
doc_type="00356",
filename="file.pdf",
received_at=datetime.datetime.utcnow(),
source="CUI",
content_location=None,
errored=False,
)
d = db.create_documents(logger, [doc])
success_result_of(d)
d = db.set_document_content_location(logger, doc, "/path/to/content")
success_result_of(d)
download = success_result_of(db.get_download(
logger, "test-request-id"
))
assert download.completed
assert download.percent_completed == 100
assert download.documents[0].content_location == "/path/to/content"
def __init__(self):
self._data = {
"started":
DownloadStatus(
request_id="started",
file_number="123456789",
state="STARTED",
documents=[],
),
"manifest-downloaded":
DownloadStatus(request_id="manifest-downloaded",
file_number="123456789",
state="MANIFEST_DOWNLOADED",
documents=[
Document(id="",
download_id="manifest-downloaded",
document_id="",
doc_type="",
filename="abc.pdf",
received_at=None,
source="",
content_location=None,
errored=False),
]),
"download-in-progress":
DownloadStatus(request_id="download-in-progress",
file_number="123456789",
state="MANIFEST_DOWNLOADED",
documents=[
Document(id="",
download_id="manifest-downloaded",
document_id="",
doc_type="",
filename="abc.pdf",
received_at=None,
source="",
content_location=None,
errored=False),
Document(id="",
download_id="manifest-downloaded",
document_id="",
doc_type="",
filename="def.pdf",
received_at=None,
source="",
content_location=None,
errored=True),
Document(id="",
download_id="manifest-downloaded",
document_id="",
doc_type="",
filename="ghe.pdf",
received_at=None,
source="",
content_location="/not-real/",
errored=False),
]),
"manifest-download-error":
DownloadStatus(
request_id="manifest-download-error",
file_number="123456789",
state="ERRORED",
documents=[],
),
"completed":
DownloadStatus(request_id="completed",
file_number="123456789",
state="MANIFEST_DOWNLOADED",
documents=[
Document(id="",
download_id="manifest-downloaded",
document_id="",
doc_type="",
filename="abc.pdf",
received_at=None,
source="",
content_location="/not-real/",
errored=False),
Document(id="",
download_id="manifest-downloaded",
document_id="",
doc_type="",
filename="def.pdf",
received_at=None,
source="",
content_location="/not-real/",
errored=False),
Document(id="",
download_id="manifest-downloaded",
document_id="",
doc_type="",
filename="ghe.pdf",
received_at=None,
source="",
content_location="/not-real/",
errored=False),
])
}
