Skip to content

Commit ac49e26

Browse files
committed
Test contract signatures for 1.3.0 and 1.4.1 Safes
- Related to #2535 and #2526
1 parent 3213671 commit ac49e26

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

safe_transaction_service/safe_messages/tests/test_views.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -281,10 +281,10 @@ def test_safe_messages_create_view(self, get_owners_mock: MagicMock):
281281
},
282282
)
283283

284-
def test_safe_messages_create_using_1271_signature_view(self):
284+
def _test_safe_messages_create_using_1271_signature_view(self, safe_deployment_fn):
285285
account = Account.create()
286-
safe_owner = self.deploy_test_safe(owners=[account.address])
287-
safe = self.deploy_test_safe(owners=[safe_owner.address])
286+
safe_owner = safe_deployment_fn(owners=[account.address])
287+
safe = safe_deployment_fn(owners=[safe_owner.address])
288288

289289
safe_address = safe.address
290290
message = get_eip712_payload_mock()
@@ -317,6 +317,16 @@ def test_safe_messages_create_using_1271_signature_view(self):
317317
self.assertEqual(SafeMessage.objects.count(), 1)
318318
self.assertEqual(SafeMessageConfirmation.objects.count(), 1)
319319

320+
def test_safe_messages_create_using_1271_signature_v1_3_0_view(self):
321+
return self._test_safe_messages_create_using_1271_signature_view(
322+
self.deploy_test_safe_v1_3_0
323+
)
324+
325+
def test_safe_messages_create_using_1271_signature_v1_4_1_view(self):
326+
return self._test_safe_messages_create_using_1271_signature_view(
327+
self.deploy_test_safe_v1_4_1
328+
)
329+
320330
@mock.patch(
321331
"safe_transaction_service.safe_messages.serializers.get_safe_owners",
322332
return_value=[],

0 commit comments

Comments
 (0)