Skip to content

Commit 80ecae6

Browse files
committed
contract constructor test refactor
1 parent 22fbd77 commit 80ecae6

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

tests/core/contracts/test_contract_constructor.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -91,20 +91,22 @@ def test_contract_constructor_transact_with_constructor_without_arguments(
9191

9292

9393
@pytest.mark.parametrize(
94-
'constructor_args,constructor_kwargs',
94+
'constructor_args,constructor_kwargs, expected_a, expected_b',
9595
(
96-
([1234, b'abcd'], {}),
97-
([1234], {'b': b'abcd'}),
98-
([], {'a': 1234, 'b': b'abcd'}),
99-
([], {'b': b'abcd', 'a': 1234}),
96+
([1234, b'abcd'], {}, EXPECTED_DATA_A, EXPECTED_DATA_B),
97+
([1234], {'b': b'abcd'}, EXPECTED_DATA_A, EXPECTED_DATA_B),
98+
([], {'a': 1234, 'b': b'abcd'}, EXPECTED_DATA_A, EXPECTED_DATA_B),
99+
([], {'b': b'abcd', 'a': 1234}, EXPECTED_DATA_A, EXPECTED_DATA_B),
100100
),
101101
)
102102
def test_contract_constructor_transact_with_constructor_with_arguments(
103103
web3,
104104
WithConstructorArgumentsContract,
105105
WITH_CONSTRUCTOR_ARGUMENTS_RUNTIME,
106106
constructor_args,
107-
constructor_kwargs):
107+
constructor_kwargs,
108+
expected_a,
109+
expected_b):
108110
deploy_txn = WithConstructorArgumentsContract.constructor(
109111
*constructor_args, **constructor_kwargs).transact()
110112

@@ -116,9 +118,9 @@ def test_contract_constructor_transact_with_constructor_with_arguments(
116118

117119
blockchain_code = web3.eth.getCode(contract_address)
118120
assert blockchain_code == decode_hex(WITH_CONSTRUCTOR_ARGUMENTS_RUNTIME)
119-
assert EXPECTED_DATA_A == WithConstructorArgumentsContract(
121+
assert expected_a == WithConstructorArgumentsContract(
120122
address=contract_address).functions.data_a().call()
121-
assert EXPECTED_DATA_B == WithConstructorArgumentsContract(
123+
assert expected_b == WithConstructorArgumentsContract(
122124
address=contract_address).functions.data_b().call()
123125

124126

0 commit comments

Comments
 (0)