poe lint

Author: yuandrew

tests/test_plugins.py

@@ -150,15 +150,19 @@ async def test_worker_plugin_basic_config(client: Client) -> None:
         plugins=[MyWorkerPlugin()],
     )
     assert worker.config().get("task_queue") == "replaced_queue"
-    assert [p.name() for p in worker.config().get("plugins")] == [MyWorkerPlugin().name()]
+    assert [p.name() for p in worker.config().get("plugins", [])] == [
+        MyWorkerPlugin().name()
+    ]
 
     # Test client plugin propagation to worker plugins
     new_config = client.config()
     new_config["plugins"] = [MyCombinedPlugin()]
     client = Client(**new_config)
     worker = Worker(client, task_queue="queue", activities=[never_run_activity])
     assert worker.config().get("task_queue") == "combined"
-    assert [p.name() for p in worker.config().get("plugins")] == [MyCombinedPlugin().name()]
+    assert [p.name() for p in worker.config().get("plugins", [])] == [
+        MyCombinedPlugin().name()
+    ]
 
     # Test both. Client propagated plugins are called first, so the worker plugin overrides in this case
     worker = Worker(
@@ -168,7 +172,7 @@ async def test_worker_plugin_basic_config(client: Client) -> None:
         plugins=[MyWorkerPlugin()],
     )
     assert worker.config().get("task_queue") == "replaced_queue"
-    assert [p.name() for p in worker.config().get("plugins")] == [
+    assert [p.name() for p in worker.config().get("plugins", [])] == [
         MyCombinedPlugin().name(),
         MyWorkerPlugin().name(),
     ]

tests/test_runtime.py

@@ -21,7 +21,12 @@
     TelemetryFilter,
 )
 from temporalio.worker import Worker
-from tests.helpers import assert_eq_eventually, assert_eventually, find_free_port, worker_versioning_enabled
+from tests.helpers import (
+    assert_eq_eventually,
+    assert_eventually,
+    find_free_port,
+    worker_versioning_enabled,
+)
 
 
 @workflow.defn
@@ -260,8 +265,12 @@ async def check_metrics() -> None:
 
 
 def test_runtime_options_to_bridge_config() -> None:
-    assert RuntimeOptions()._to_bridge_config().worker_heartbeat_interval_millis == 30_000
-    bridge_config = RuntimeOptions(worker_heartbeat_interval=timedelta(seconds=60))._to_bridge_config()
+    assert (
+        RuntimeOptions()._to_bridge_config().worker_heartbeat_interval_millis == 30_000
+    )
+    bridge_config = RuntimeOptions(
+        worker_heartbeat_interval=timedelta(seconds=60)
+    )._to_bridge_config()
     assert bridge_config.worker_heartbeat_interval_millis == 60_000
 
     bridge_config1 = RuntimeOptions(worker_heartbeat_interval=None)._to_bridge_config()

tests/worker/test_worker.py

@@ -1139,7 +1139,9 @@ async def test_can_run_autoscaling_polling_worker(
         activity_pollers = [l for l in matches if "activity_task" in l]
         assert len(activity_pollers) == 1
         assert activity_pollers[0].endswith("2")
-        workflow_pollers = [l for l in matches if "workflow_task" in l and w.task_queue in l]
+        workflow_pollers = [
+            l for l in matches if "workflow_task" in l and w.task_queue in l
+        ]
         assert len(workflow_pollers) == 2
         # There's sticky & non-sticky pollers, and they may have a count of 1 or 2 depending on
         # initialization timing.