Update message parsing

* Update parsing of SQS messages in listen function * Fixture updates to better reflect actual environment
MITLibraries · Feb 9, 2022 · cbe258f · cbe258f
1 parent 851fbdc
commit cbe258f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/awd/cli.py b/awd/cli.py
@@ -267,8 +267,9 @@ def listen(
                 .get("PackageID", {})
                 .get("StringValue")
             )
-            if "'ResultType': 'error'" in message["Body"]:
-                logger.error(f'DOI: {doi}, Result: {message.get("Body")}')
+            body = json.loads(message.get("Body"))
+            if body["ResultType"] == "error":
+                logger.error(f"DOI: {doi}, Result: {body}")
                 sqs.delete(
                     ctx.obj["sqs_base_url"],
                     ctx.obj["sqs_output_queue"],
@@ -283,7 +284,7 @@ def listen(
                 else:
                     continue
             else:
-                logger.info(f'DOI: {doi}, Result: {message.get("Body")}')
+                logger.info(f"DOI: {doi}, Result: {body}")
                 sqs.delete(
                     ctx.obj["sqs_base_url"],
                     ctx.obj["sqs_output_queue"],

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -232,7 +232,7 @@ def result_failure_message_body():
         "ExceptionMessage": "500 Server Error: Internal Server Error",
         "ExceptionTraceback": "Full unformatted stack trace of the Exception",
     }
-    return result_failure_message_body
+    return json.dumps(result_failure_message_body)
 
 
 @pytest.fixture()
@@ -252,7 +252,7 @@ def result_success_message_body():
             }
         ],
     }
-    return result_success_message_body
+    return json.dumps(result_success_message_body)
 
 
 @pytest.fixture()