@@ -272,15 +272,11 @@ def test_generate_tag_collisions(importer):
272272
273273def test_import_safe_duplicate (importer ):
274274 """Test importing a Safe that already exists."""
275- from iwa .plugins .olas .importer import DiscoveredService
276-
277- service = DiscoveredService (
278- service_id = 1 , chain_name = "gnosis" , safe_address = "0xSafe" , source_folder = Path ("/tmp" )
279- )
275+ safe_address = "0xSafe"
280276
281277 importer .key_storage .find_stored_account .return_value = MagicMock ()
282278
283- success , msg = importer ._import_safe (service )
279+ success , msg = importer ._import_safe (safe_address )
284280 assert success is False
285281 assert msg == "duplicate"
286282
@@ -310,22 +306,14 @@ def test_import_key_success(importer):
310306
311307def test_import_safe_success (importer ):
312308 """Test successful Safe import with tag generation."""
313- from iwa .plugins .olas .importer import DiscoveredService
314-
315- service = DiscoveredService (
316- service_id = 1 ,
317- chain_name = "gnosis" ,
318- safe_address = "0x78731D3Ca6b7E34aC0F824c42a7cC18A495cabaB" ,
319- source_folder = Path ("/tmp" ),
320- service_name = "my_service" ,
321- )
309+ safe_address = "0x78731D3Ca6b7E34aC0F824c42a7cC18A495cabaB"
322310 importer .key_storage .find_stored_account .return_value = None
323311 importer .key_storage .accounts = {}
324312
325- success , msg = importer ._import_safe (service )
313+ success , msg = importer ._import_safe (safe_address , service_name = "my_service" )
326314 assert success is True
327315 assert msg == "ok"
328- assert "0x78731D3Ca6b7E34aC0F824c42a7cC18A495cabaB" in importer .key_storage .accounts
316+ assert safe_address in importer .key_storage .accounts
329317
330318
331319def test_import_service_config_success (importer ):
0 commit comments