@@ -80,9 +80,7 @@ async def test_dispatches_get_system_status_query(self):
8080
8181 container , query_bus = _mock_container_with_query_bus (query_return = {"status" : "ok" })
8282
83- with patch (
84- "orb.interface.system_command_handlers.get_container" , return_value = container
85- ):
83+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
8684 result = await handle_provider_health (_ns ())
8785
8886 query_bus .execute .assert_awaited_once ()
@@ -100,9 +98,7 @@ async def test_dispatches_get_provider_config_query(self):
10098
10199 container , query_bus = _mock_container_with_query_bus (query_return = {"provider" : "aws" })
102100
103- with patch (
104- "orb.interface.system_command_handlers.get_container" , return_value = container
105- ):
101+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
106102 result = await handle_provider_config (_ns ())
107103
108104 query_bus .execute .assert_awaited_once ()
@@ -120,9 +116,7 @@ async def test_dispatches_get_provider_metrics_query_with_provider_name(self):
120116
121117 container , query_bus = _mock_container_with_query_bus (query_return = {"latency_ms" : 42 })
122118
123- with patch (
124- "orb.interface.system_command_handlers.get_container" , return_value = container
125- ):
119+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
126120 result = await handle_provider_metrics (_ns (provider = "aws" ))
127121
128122 query_bus .execute .assert_awaited_once ()
@@ -138,9 +132,7 @@ async def test_dispatches_with_no_provider_name(self):
138132
139133 container , query_bus = _mock_container_with_query_bus (query_return = {})
140134
141- with patch (
142- "orb.interface.system_command_handlers.get_container" , return_value = container
143- ):
135+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
144136 await handle_provider_metrics (_ns ())
145137
146138 q = query_bus .execute .call_args [0 ][0 ]
@@ -157,9 +149,7 @@ async def test_dispatches_get_system_status_query_with_flags(self):
157149
158150 container , query_bus = _mock_container_with_query_bus (query_return = {"healthy" : True })
159151
160- with patch (
161- "orb.interface.system_command_handlers.get_container" , return_value = container
162- ):
152+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
163153 result = await handle_system_status (_ns (detailed = True ))
164154
165155 query_bus .execute .assert_awaited_once ()
@@ -176,9 +166,7 @@ async def test_detailed_defaults_to_false_when_not_set(self):
176166
177167 container , query_bus = _mock_container_with_query_bus (query_return = {})
178168
179- with patch (
180- "orb.interface.system_command_handlers.get_container" , return_value = container
181- ):
169+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
182170 await handle_system_status (_ns ())
183171
184172 q = query_bus .execute .call_args [0 ][0 ]
@@ -217,9 +205,7 @@ async def test_returns_providers_from_config_port(self):
217205 container = MagicMock ()
218206 container .get .return_value = mock_config_port
219207
220- with patch (
221- "orb.interface.system_command_handlers.get_container" , return_value = container
222- ):
208+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
223209 result = await handle_list_providers (_ns ())
224210
225211 assert result ["count" ] == 1
@@ -236,9 +222,7 @@ async def test_no_provider_config_returns_empty_list(self):
236222 container = MagicMock ()
237223 container .get .return_value = mock_config_port
238224
239- with patch (
240- "orb.interface.system_command_handlers.get_container" , return_value = container
241- ):
225+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
242226 result = await handle_list_providers (_ns ())
243227
244228 assert result ["count" ] == 0
@@ -251,9 +235,7 @@ async def test_exception_returns_error_dict(self):
251235 container = MagicMock ()
252236 container .get .side_effect = RuntimeError ("container exploded" )
253237
254- with patch (
255- "orb.interface.system_command_handlers.get_container" , return_value = container
256- ):
238+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
257239 result = await handle_list_providers (_ns ())
258240
259241 assert result ["count" ] == 0
@@ -306,9 +288,7 @@ async def test_returns_metrics_when_collector_available(self):
306288 container = MagicMock ()
307289 container .get_optional .return_value = mock_metrics
308290
309- with patch (
310- "orb.interface.system_command_handlers.get_container" , return_value = container
311- ):
291+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
312292 result = await handle_system_metrics (_ns ())
313293
314294 assert result ["metrics" ] == {"requests_total" : 5 }
@@ -320,9 +300,7 @@ async def test_returns_empty_metrics_when_collector_unavailable(self):
320300 container = MagicMock ()
321301 container .get_optional .return_value = None
322302
323- with patch (
324- "orb.interface.system_command_handlers.get_container" , return_value = container
325- ):
303+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
326304 result = await handle_system_metrics (_ns ())
327305
328306 assert result ["metrics" ] == {}
@@ -337,9 +315,7 @@ async def test_returns_error_dict_when_get_metrics_raises(self):
337315 container = MagicMock ()
338316 container .get_optional .return_value = mock_metrics
339317
340- with patch (
341- "orb.interface.system_command_handlers.get_container" , return_value = container
342- ):
318+ with patch ("orb.interface.system_command_handlers.get_container" , return_value = container ):
343319 result = await handle_system_metrics (_ns ())
344320
345321 assert result ["metrics" ] == {}
@@ -357,9 +333,7 @@ class TestHandleSelectProviderStrategy:
357333 async def test_returns_selected_provider_from_args (self ):
358334 from orb .interface .system_command_handlers import handle_select_provider_strategy
359335
360- with patch (
361- "orb.providers.registry.get_provider_registry"
362- ) as mock_registry_fn :
336+ with patch ("orb.providers.registry.get_provider_registry" ) as mock_registry_fn :
363337 mock_registry = MagicMock ()
364338 mock_registry .get_registered_providers .return_value = ["aws" ]
365339 mock_registry_fn .return_value = mock_registry
@@ -372,9 +346,7 @@ async def test_returns_selected_provider_from_args(self):
372346 async def test_falls_back_to_first_registered_provider_when_no_args (self ):
373347 from orb .interface .system_command_handlers import handle_select_provider_strategy
374348
375- with patch (
376- "orb.providers.registry.get_provider_registry"
377- ) as mock_registry_fn :
349+ with patch ("orb.providers.registry.get_provider_registry" ) as mock_registry_fn :
378350 mock_registry = MagicMock ()
379351 mock_registry .get_registered_providers .return_value = ["aws" ]
380352 mock_registry_fn .return_value = mock_registry
0 commit comments