Merge branch 'feature/add-support-for-xdist'

saritasa-nest · May 22, 2024 · 74fae06 · 74fae06
2 parents 90a13ba + 8b2671a
commit 74fae06
Showing 1 changed file with 32 additions and 8 deletions.
diff --git a/sns_sqs_communicator/testing/plugin.py b/sns_sqs_communicator/testing/plugin.py
@@ -174,10 +174,22 @@ def sqs_queue_name(
     request: pytest.FixtureRequest,
 ) -> str:
     """Get queue name."""
-    return str(
-        request.config.inicfg.get(
-            "sns_sqs_queue_name",
-            "sns-sqs-communicator-queue",
+    worker_input = getattr(
+        request.config,
+        "workerinput",
+        {
+            "workerid": "",
+        },
+    )
+    return "-".join(
+        (
+            str(
+                request.config.inicfg.get(
+                    "sns_sqs_queue_name",
+                    "sns-sqs-communicator-queue",
+                ),
+            ),
+            worker_input["workerid"],
         ),
     )
 
@@ -255,10 +267,22 @@ def sns_topic_name(
     request: pytest.FixtureRequest,
 ) -> str:
     """Get queue name."""
-    return str(
-        request.config.inicfg.get(
-            "sns_sqs_topic_name",
-            "sns-sqs-communicator-topic",
+    worker_input = getattr(
+        request.config,
+        "workerinput",
+        {
+            "workerid": "",
+        },
+    )
+    return "-".join(
+        (
+            str(
+                request.config.inicfg.get(
+                    "sns_sqs_topic_name",
+                    "sns-sqs-communicator-topic",
+                ),
+            ),
+            worker_input["workerid"],
         ),
     )