Auto-format the "test key uploading" fixture and helper functions
Marco Ricci commited on 2025-12-27 15:15:35
Zeige 1 geänderte Dateien mit 7 Einfügungen und 3 Löschungen.
- tests/conftest.py f5c28b5..195c0d4
| ... | ... |
@@ -650,13 +650,15 @@ def spawn_named_agent( # noqa: C901 |
| 650 | 650 |
ssh_agent.SSHAgentClient, |
| 651 | 651 |
"SOCKET_PROVIDERS", |
| 652 | 652 |
["stub_agent_with_address_and_deterministic_dsa"] |
| 653 |
- if "stub_agent_with_address_and_deterministic_dsa" in agent_name |
|
| 653 |
+ if "stub_agent_with_address_and_deterministic_dsa" |
|
| 654 |
+ in agent_name |
|
| 654 | 655 |
else ["stub_agent_with_address"], |
| 655 | 656 |
) |
| 656 | 657 |
client = exit_stack.enter_context( |
| 657 | 658 |
ssh_agent.SSHAgentClient.ensure_agent_subcontext( |
| 658 | 659 |
machinery.StubbedSSHAgentSocketWithAddressAndDeterministicDSA() |
| 659 |
- if "stub_agent_with_address_and_deterministic_dsa" in agent_name |
|
| 660 |
+ if "stub_agent_with_address_and_deterministic_dsa" |
|
| 661 |
+ in agent_name |
|
| 660 | 662 |
else machinery.StubbedSSHAgentSocketWithAddress() |
| 661 | 663 |
) |
| 662 | 664 |
) |
| ... | ... |
@@ -854,7 +856,9 @@ def _prepare_payload( |
| 854 | 856 |
return (return_code, bytes(payload) + lifetime_constraint) |
| 855 | 857 |
|
| 856 | 858 |
|
| 857 |
-def _load_keys_optimistically(spawned_agent_info: data.SpawnedSSHAgentInfo) -> set[str]: |
|
| 859 |
+def _load_keys_optimistically( |
|
| 860 |
+ spawned_agent_info: data.SpawnedSSHAgentInfo, |
|
| 861 |
+) -> set[str]: |
|
| 858 | 862 |
successfully_loaded_keys: set[str] = set() |
| 859 | 863 |
for key_type, key_struct in data.ALL_KEYS.items(): |
| 860 | 864 |
agent_type, client, isolated = spawned_agent_info |
| 861 | 865 |