Skip to content

Commit 8cbb8ee

Browse files
[PRMT-70] Improve efficiency of LG retrieval (#675)
* [PRMT-70]- updated format to not use float * [PRMT-70]- updated last_updated to use int * [PRMT-70]- fixed stiching file ovewritting a file * [PRMT-70]- updated test to reflect correct file name * [PRMT-70]- updated test to test bucket name
1 parent 500a68a commit 8cbb8ee

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

lambdas/services/pdf_stitching_service.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,8 @@ def create_stitched_reference(
119119
"file_location": f"s3://{self.target_bucket}/{document_reference.nhs_number}/{reference_id}",
120120
"file_name": f"1of1_{stripped_filename}",
121121
"file_size": stitch_file_size,
122-
"last_updated": date_now.timestamp(),
122+
"last_updated": int(datetime.now(timezone.utc).timestamp()),
123+
"s3_file_key": f"{document_reference.nhs_number}/{reference_id}",
123124
},
124125
deep=True,
125126
)

lambdas/tests/unit/services/test_pdf_stitching_service.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,8 @@ def test_create_stitched_reference(mock_service, mock_uuid, document_reference):
258258
assert actual.uploading is False
259259
assert actual.last_updated == 1735732800
260260
assert actual.file_size == file_size
261+
assert actual.s3_file_key == f"{TEST_NHS_NUMBER}/{TEST_UUID}"
262+
assert actual.s3_bucket_name == MOCK_LG_BUCKET
261263

262264

263265
def test_process_stitching(mock_service, mock_download_fileobj):

0 commit comments

Comments
 (0)