test: deposit and withdraw on honeypot

Author: mpolitzer

prt/tests/rollups/dave/sender.lua

@@ -80,31 +80,10 @@ function Sender:new(input_box_address, dave_app_factory_address, app_contract_ad
     return sender
 end
 
-local cast_impersonate_template = [[
-cast rpc anvil_impersonateAccount %s --rpc-url "%s" 2>&1
-]]
 local cast_send_template = [[
 cast send --from "%s" --rpc-url "%s" --value "%s" "%s" "%s" %s --unlocked 2>&1
 ]]
 function Sender:_send_tx(tournament_address, sender, sig, args, value)
-    -- impersonate
-    do
-        local cmd = string.format(
-            cast_impersonate_template,
-            sender,
-            self.endpoint
-        )
-
-        local handle = io.popen(cmd)
-        assert(handle)
-
-        local ret = handle:read "*a"
-        handle:close()
-        if ret:find "Error" then
-            error(string.format("impersonate %s reverted:\n%s", sender, ret))
-        end
-    end
-
     value = value or bint.zero()
 
     local quoted_args = quote_args(args)

prt/tests/rollups/justfile

@@ -41,7 +41,7 @@ test-yield-all:
   just test {{YIELD_DIR}} bad_commitment
 
 test-honeypot-ci:
-  just test {{HONEYPOT_DIR}} simple_honeypot
+  just test {{HONEYPOT_DIR}} honeypot
 
 test-yield-ci:
   just test {{YIELD_DIR}} simple

prt/tests/rollups/test_cases/honeypot.lua

@@ -0,0 +1,58 @@
+require "setup_path"
+
+local Hash = require "cryptography.hash"
+local env = require "test_env"
+
+local CONFIG_ERC20_PORTAL_ADDRESS = "ACA6586A0Cf05bD831f2501E7B4aea550dA6562D"
+local CONFIG_ERC20_WITHDRAWAL_ADDRESS = "70997970C51812dc3A010C7d01b50e0d17dc79C8"
+local CONFIG_ERC20_TOKEN_ADDRESS = "c6e7DF5E7b4f2A278906862b61205850344D4e7d"
+local ETH10K = "0x21e19e0c9bab2400000"
+
+local function exec(fmt, ...)
+    local cmd = string.format(fmt, ...)
+    local reader = io.popen(cmd)
+    assert(reader, "`popen` returned nil reader")
+
+    local data = reader:read("*a")
+    local success, _, code = reader:close()
+    assert(success, string.format("command [[%s]] failed on close:\n%d", cmd, code))
+
+    return data:gsub("\n$", "") -- remove trailing newline from data
+end
+
+local function set_balance(endpoint, account, value)
+    return exec([[cast rpc -r "%s" anvil_setBalance "%s" "%s"]], endpoint, account, value)
+end
+
+local function auto_impersonate(endpoint, yes)
+    return exec([[cast rpc -r "%s" anvil_autoImpersonateAccount %s]], endpoint, yes)
+end
+
+local deposit_payload = exec(
+	[[cast abi-encode --packed '(address,address,uint256,bytes)' "%s" "%s" "%s" "%s"]],
+	CONFIG_ERC20_TOKEN_ADDRESS,
+	CONFIG_ERC20_PORTAL_ADDRESS,
+	"1000", "0x"
+)
+
+local valid_deposit_inpuit = { sender = CONFIG_ERC20_PORTAL_ADDRESS, payload = deposit_payload }
+local invalid_deposit_input = { sender = CONFIG_ERC20_WITHDRAWAL_ADDRESS, payload = deposit_payload }
+local withdrawal_input = { sender = CONFIG_ERC20_WITHDRAWAL_ADDRESS, payload = "0x" }
+
+-- Main Execution
+env.spawn_blockchain({valid_deposit_inpuit, invalid_deposit_input, withdrawal_input}, function ()
+	set_balance(env.blockchain.endpoint, CONFIG_ERC20_PORTAL_ADDRESS, ETH10K)
+	auto_impersonate(env.blockchain.endpoint, "true")
+end)
+
+-- Spawn Dave node
+env.spawn_node()
+
+-- advance such that epoch 0 is finished
+local sealed_epoch = env.roll_epoch()
+
+-- run epoch 1
+env.run_epoch(sealed_epoch, {
+    -- ustep + reset
+    { hash = Hash.zero, meta_cycle = 1 << 44 }
+}, { env.sample_inputs[1], env.sample_inputs[1], env.sample_inputs[1] })

prt/tests/rollups/test_env.lua

@@ -49,7 +49,7 @@ local Env = {
     -- consensus_address = false,
 }
 
-function Env.spawn_blockchain(inputs)
+function Env.spawn_blockchain(inputs, run_before_inputs)
     inputs = inputs or {}
 
     local blockchain = Blockchain:new(ANVIL_PATH)
@@ -58,6 +58,9 @@ function Env.spawn_blockchain(inputs)
     Env.app_address = Env.reader.app_address
     Env.consensus_address = Env.reader.consensus_address
     Env.sender = Sender:new(INPUT_BOX_ADDRESS, DAVE_APP_FACTORY_ADDRESS, Env.app_address, blockchain.pks[1], blockchain.endpoint)
+    if run_before_inputs then
+        run_before_inputs()
+    end
     Env.sender:tx_add_inputs(inputs)
     Env.sender:tx_new_dave_app(TEMPLATE_MACHINE_HASH, SALT)
     Env.sender:advance_blocks(2)