Skip to content

Commit 2e6f369

Browse files
committed
Fix linting: black-format test files
Co-Authored-By: Claude Sonnet 4.6 <[email protected]> Claude-Session: https://claude.ai/code/session_01L7YspLCAcj6CjqwxVezzXe
1 parent dfe70d9 commit 2e6f369

2 files changed

Lines changed: 6 additions & 18 deletions

File tree

tests/test_gateway_handler.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -76,9 +76,7 @@ async def test_calls_copy_to_container(
7676
assert args[0] == "netengine_gateway"
7777
assert args[2] == "/etc/nftables/rules/home1.nft"
7878

79-
async def test_calls_nft_exec(
80-
self, handler: GatewayHandler, mock_docker: MagicMock
81-
) -> None:
79+
async def test_calls_nft_exec(self, handler: GatewayHandler, mock_docker: MagicMock) -> None:
8280
await handler.apply_rules("home1", "table ip netengine_home1 {}")
8381
mock_docker.exec_command.assert_called_once_with(
8482
"netengine_gateway", ["nft", "-f", "/etc/nftables/rules/home1.nft"]
@@ -130,9 +128,7 @@ async def test_calls_nft_with_main_nft(
130128
"netengine_gateway", ["nft", "-f", "/etc/nftables/rules/main.nft"]
131129
)
132130

133-
async def test_raises_on_failure(
134-
self, handler: GatewayHandler, mock_docker: MagicMock
135-
) -> None:
131+
async def test_raises_on_failure(self, handler: GatewayHandler, mock_docker: MagicMock) -> None:
136132
mock_docker.exec_command.return_value = (1, "reload failed")
137133
with pytest.raises(Exception, match="reload"):
138134
await handler.reload()

tests/test_registry_handlers.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -129,16 +129,12 @@ async def test_allocate_address_raises_when_no_pool(
129129
with pytest.raises(RegistryError, match="No address pool"):
130130
await handler.allocate_address("office1", "residential")
131131

132-
async def test_allocate_address_returns_cidr(
133-
self, handler: MagicMock, sb: MagicMock
134-
) -> None:
132+
async def test_allocate_address_returns_cidr(self, handler: MagicMock, sb: MagicMock) -> None:
135133
sb._result.data = [{"cidr": "10.1.0.0/24"}]
136134
cidr = await handler.allocate_address("office1", "residential")
137135
assert cidr == "10.1.0.0/24"
138136

139-
async def test_allocate_address_upserts_lease(
140-
self, handler: MagicMock, sb: MagicMock
141-
) -> None:
137+
async def test_allocate_address_upserts_lease(self, handler: MagicMock, sb: MagicMock) -> None:
142138
sb._result.data = [{"cidr": "10.1.0.0/24"}]
143139
await handler.allocate_address("office1", "residential")
144140
upsert_calls = sb._query.upsert.call_args_list
@@ -148,18 +144,14 @@ async def test_allocate_address_upserts_lease(
148144
assert lease_call is not None
149145
assert lease_call.args[0]["and_name"] == "office1"
150146

151-
async def test_register_domain_upserts_record(
152-
self, handler: MagicMock, sb: MagicMock
153-
) -> None:
147+
async def test_register_domain_upserts_record(self, handler: MagicMock, sb: MagicMock) -> None:
154148
await handler.register_domain("acme.internal", "acme", ["ns1.internal"])
155149
sb.table.assert_called_with("domain_records")
156150
sb._query.upsert.assert_called_with(
157151
{"domain": "acme.internal", "org_name": "acme", "ns_records": ["ns1.internal"]}
158152
)
159153

160-
async def test_register_domain_sends_dns_update_event(
161-
self, handler: MagicMock
162-
) -> None:
154+
async def test_register_domain_sends_dns_update_event(self, handler: MagicMock) -> None:
163155
await handler.register_domain("acme.internal", "acme", [])
164156
handler.pgmq.send.assert_called_once()
165157
assert handler.pgmq.send.call_args.args[0] == "dns_updates"

0 commit comments

Comments
 (0)