@@ -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