Skip to content

Commit b8127fa

Browse files
Fix tests
1 parent ac9f2d6 commit b8127fa

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

safe_transaction_service/history/tests/test_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def test_calculate_hash(self):
2424
"0x40ef28feb993bf127265260d48ab1a427d5b852aa8b38f511fcd368bfc9cbfdf"
2525
)
2626

27-
result_hash = DelegateSignatureHelperV2.calculate_hash(
27+
result_hash, _ = DelegateSignatureHelperV2.calculate_hash_and_preimage(
2828
delegate_address, chain_id, False
2929
)
3030

@@ -36,7 +36,7 @@ def test_calculate_hash(self):
3636
"0x37fe1c77d467e92109ef91637e30a4053bab963de2ea74f9d6c4ba1918ff32e6"
3737
)
3838

39-
result_hash = DelegateSignatureHelperV2.calculate_hash(
39+
result_hash, _ = DelegateSignatureHelperV2.calculate_hash_and_preimage(
4040
delegate_address, chain_id, True
4141
)
4242

safe_transaction_service/history/tests/test_views_v2.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ def test_delegates_post(self):
232232
# Create delegate
233233
self.assertEqual(SafeContractDelegate.objects.count(), 0)
234234
chain_id = self.ethereum_client.get_chain_id()
235-
hash_to_sign = DelegateSignatureHelperV2.calculate_hash(
235+
hash_to_sign, _ = DelegateSignatureHelperV2.calculate_hash_and_preimage(
236236
delegate.address, chain_id, False
237237
)
238238
data["signature"] = to_0x_hex_str(
@@ -271,7 +271,7 @@ def test_delegates_post(self):
271271
self.assertIsNone(safe_contract_delegate.expiry_date)
272272

273273
# Create delegate without a Safe
274-
hash_to_sign = DelegateSignatureHelperV2.calculate_hash(
274+
hash_to_sign, _ = DelegateSignatureHelperV2.calculate_hash_and_preimage(
275275
delegate.address, chain_id, False
276276
)
277277
data = {
@@ -302,7 +302,7 @@ def test_delegate_creation_without_chain_id(self):
302302
chain_id = None
303303
delegate = Account.create()
304304
delegator = Account.create()
305-
hash_to_sign = DelegateSignatureHelperV2.calculate_hash(
305+
hash_to_sign, _ = DelegateSignatureHelperV2.calculate_hash_and_preimage(
306306
delegate.address, chain_id, False
307307
)
308308
data = {
@@ -326,7 +326,7 @@ def test_delegate_creation_without_chain_id_with_safe(self):
326326
chain_id = None
327327
delegate = Account.create()
328328
delegator = Account.create()
329-
hash_to_sign = DelegateSignatureHelperV2.calculate_hash(
329+
hash_to_sign, _ = DelegateSignatureHelperV2.calculate_hash_and_preimage(
330330
delegate.address, chain_id, False
331331
)
332332
data = {
@@ -432,7 +432,7 @@ def test_delegate_delete(self):
432432
delegate = Account.create()
433433
delegator = Account.create()
434434
chain_id = self.ethereum_client.get_chain_id()
435-
hash_to_sign = DelegateSignatureHelperV2.calculate_hash(
435+
hash_to_sign, _ = DelegateSignatureHelperV2.calculate_hash_and_preimage(
436436
delegate.address, chain_id, False
437437
)
438438
# Test delete using delegate signature and then delegator signature

0 commit comments

Comments
 (0)