From 7975d0303cc9e9eab27bd85ee3d2f1eecd634fb6 Mon Sep 17 00:00:00 2001 From: Ryan Lim Date: Fri, 25 Aug 2023 14:06:33 -0700 Subject: [PATCH] typing --- test/test_wdl.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_wdl.py b/test/test_wdl.py index 189347b2..7648a8b1 100644 --- a/test/test_wdl.py +++ b/test/test_wdl.py @@ -297,7 +297,7 @@ def tearDown(self) -> None: ) self.test_bucket.delete() - def retrieve_message(self, url) -> str: + def retrieve_message(self, url: str) -> Dict: """Retrieve a single SQS message and delete it from queue""" resp = self.sqs.receive_message( QueueUrl=url, @@ -305,7 +305,7 @@ def retrieve_message(self, url) -> str: ) # If no messages, just return if not resp.get("Messages", None): - return "" + return {} message = resp["Messages"][0] receipt_handle = message["ReceiptHandle"] @@ -321,7 +321,7 @@ def _wait_sfn( sfn_arn: str, n_stages: int = 1, expect_success: bool = True, - ) -> Tuple[str, Dict, List[Dict]]: + ) -> Tuple[str, Dict, List[Dict], List[Dict]]: execution_name = "swipe-test-{}".format(int(time.time())) res = self.sfn.start_execution( stateMachineArn=sfn_arn, name=execution_name, input=json.dumps(sfn_input)