configurable number of iterations for prune tests

ulope · Apr 18, 2019 · 5b4375d · 5b4375d
1 parent c982fe6
commit 5b4375d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/raiden/tests/integration/rpc/RpcWithStorageTest.sol b/raiden/tests/integration/rpc/RpcWithStorageTest.sol
@@ -15,9 +15,9 @@ contract RpcWithStorageTest {
         return 1;
     }
 
-    function waste_storage() public {
+    function waste_storage(uint256 iterations) public {
         uint256 i;
-        for (i=0; i<1000; i++) {
+        for (i=0; i<iterations; i++) {
             data[data.length++] = i;
         }
         emit RpcEvent(i);

diff --git a/raiden/tests/integration/rpc/test_assumptions_parity.py b/raiden/tests/integration/rpc/test_assumptions_parity.py
@@ -18,22 +18,24 @@
 
 
 @pytest.mark.parametrize('blockchain_extra_config', [STATE_PRUNNING])
-def test_request_prunned_data_raises_an_exception(deploy_client):
+def test_parity_request_prunned_data_raises_an_exception(deploy_client):
     """ Interacting with an old block identifier with a pruning client throws. """
     contract_proxy = deploy_rpc_test_contract(deploy_client, 'RpcWithStorageTest')
+    iterations = 1000
 
-    check_block = deploy_client.get_checking_block()
-    startgas = safe_gas_limit(contract_proxy.estimate_gas(check_block, 'waste_storage'))
-    transaction = contract_proxy.transact('waste_storage', startgas)
-    deploy_client.poll(transaction)
-    first_receipt = deploy_client.get_transaction_receipt(transaction)
+    def send_transaction():
+        check_block = deploy_client.get_checking_block()
+        startgas = contract_proxy.estimate_gas(check_block, 'waste_storage', iterations)
+        startgas = safe_gas_limit(startgas)
+        transaction = contract_proxy.transact('waste_storage', startgas, iterations)
+        deploy_client.poll(transaction)
+        return deploy_client.get_transaction_receipt(transaction)
+
+    first_receipt = send_transaction()
     pruned_block_number = first_receipt['blockNumber']
 
     for _ in range(10):
-        check_block = deploy_client.get_checking_block()
-        startgas = safe_gas_limit(contract_proxy.estimate_gas(check_block, 'waste_storage'))
-        transaction = contract_proxy.transact('waste_storage', startgas)
-        deploy_client.poll(transaction)
+        send_transaction()
 
     with pytest.raises(ValueError):
         contract_proxy.contract.functions.const().call(block_identifier=pruned_block_number)