fix a confusing test

PrefectHQ · Nov 8, 2024 · 463186a · 463186a
1 parent ac9adbf
commit 463186a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/workers/test_base_worker.py b/tests/workers/test_base_worker.py
@@ -1938,7 +1938,7 @@ async def run(self):
             async def _worker_metadata(self) -> WorkerMetadata:
                 return WorkerMetadata(
                     **{
-                        "integrations": [{"name": "prefect_aws", "version": "1.0.0"}],
+                        "integrations": [{"name": "prefect-aws", "version": "1.0.0"}],
                         "custom_field": "heya",
                     }
                 )
@@ -1953,7 +1953,7 @@ async def _worker_metadata(self) -> WorkerMetadata:
                 "prefect.workers.base.distributions"
             ) as mock_distributions:
                 mock_load_prefect_collections.return_value = {
-                    "prefect-aws": "1.0.0",
+                    "prefect_aws": "1.0.0",
                 }
                 mock_distributions.return_value = [
                     mock.MagicMock(
@@ -1974,7 +1974,7 @@ async def _worker_metadata(self) -> WorkerMetadata:
                         "heartbeat_interval_seconds": worker.heartbeat_interval_seconds,
                         "worker_metadata": {
                             "integrations": [
-                                {"name": "prefect_aws", "version": "1.0.0"}
+                                {"name": "prefect-aws", "version": "1.0.0"}
                             ],
                             "custom_field": "heya",
                         },