Skip to content

Commit 41ad3d8

Browse files
committed
WIP
1 parent 94955b6 commit 41ad3d8

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

delta_backend/src/delta.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ def send_message(record):
2727
sqs_client = boto3.client("sqs")
2828
try:
2929
# Send the record to the queue
30+
print(f"Sending record to DLQ: {message_body}")
3031
sqs_client.send_message(QueueUrl=failure_queue_url, MessageBody=json.dumps(message_body))
3132
logger.info("Record saved successfully to the DLQ")
3233
except ClientError as e:
@@ -58,7 +59,7 @@ def handler(event, context):
5859
response = str()
5960
imms_id = str()
6061
operation = str()
61-
if record["eventName"] != "REMOVE":
62+
if record["eventName"] != "DELETE":
6263
new_image = record["dynamodb"]["NewImage"]
6364
imms_id = new_image["PK"]["S"].split("#")[1]
6465
supplier_system = new_image["SupplierSystem"]["S"]
@@ -73,15 +74,15 @@ def handler(event, context):
7374
logger.info(f"Record from DPS skipped for {imms_id}")
7475
return {"statusCode": 200, "body": f"Record from DPS skipped for {imms_id}"}
7576
else:
76-
operation = "REMOVE"
77+
operation = "DELETE"
7778
new_image = record["dynamodb"]["Keys"]
7879
logger.info(f"Record to delta:{new_image}")
7980
imms_id = new_image["PK"]["S"].split("#")[1]
8081
response = delta_table.put_item(
8182
Item={
8283
"PK": str(uuid.uuid4()),
8384
"ImmsID": imms_id,
84-
"Operation": "REMOVE",
85+
"Operation": "DELETE",
8586
"VaccineType": "default",
8687
"SupplierSystem": "default",
8788
"DateTimeStamp": approximate_creation_time.isoformat(),

delta_backend/src/helpers/delta_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def get_data(self):
2020
return self.data
2121

2222

23-
def get_vaccine_type(patientsk) -> str:
23+
def get_vaccine_type(self, patientsk) -> str:
2424
parsed = [str.strip(str.lower(s)) for s in patientsk.split("#")]
2525
return parsed[0]
2626

delta_backend/tests/test_delta.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,7 @@ def test_send_message_success(self, mock_boto_client):
105105
record = {"key": "value"}
106106

107107
# Act
108+
print(f"test_send_message_success.Sending record to DLQ: {record}")
108109
send_message(record)
109110

110111
# Assert
@@ -184,7 +185,7 @@ def test_handler_success_update(self, mock_boto_resource, mock_write_to_db):
184185
def test_handler_success_remove(self, mock_boto_resource):
185186
# Arrange
186187
self.setup_mock_dynamodb(mock_boto_resource)
187-
event = self.get_event(event_name="REMOVE", operation=EndpointOperationNames.DELETE)
188+
event = self.get_event(event_name="DELETE", operation=EndpointOperationNames.DELETE)
188189

189190
# Act
190191
result = handler(event, self.context)

0 commit comments

Comments
 (0)