ethereum: add 'make test-upgrade' to simulate contract upgrades against mainnet
This commit is contained in:
Csongor Kiss
Csongor Kiss
parent
3c6e1b07d6
commit
7942393b18
|
|
@ -25,5 +25,9 @@ test: build .env dependencies
|
|||
npm test || (pkill ganache-cli && exit 1)
|
||||
pkill ganache-cli || true
|
||||
|
||||
.PHONY: test-upgrade
|
||||
test-upgrade: build .env dependencies
|
||||
./simulate_upgrades
|
||||
|
||||
clean:
|
||||
rm -rf ganache.log .env node_modules build
|
||||
|
|
|
|||
|
|
@ -0,0 +1,150 @@
|
|||
#!/bin/bash
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
function usage() {
|
||||
cat <<EOF >&2
|
||||
Usage:
|
||||
|
||||
$(basename "$0") [-h] [-m s] [-c s] [-x] [-k] [-d] [-a s] [-l s] -- Simulate an upgrade on a fork of mainnet, and check for any errors.
|
||||
|
||||
where:
|
||||
-h show this help text
|
||||
-m module (bridge, token_bridge, nft_bridge)
|
||||
-c chain name
|
||||
-x run anvil
|
||||
-d don't compile contract first
|
||||
-k keep anvil alive
|
||||
-l file to loge to (by default creates a new tmp file)
|
||||
-a new code address (by default it builds the most recent contract in the repository)
|
||||
EOF
|
||||
exit 1
|
||||
}
|
||||
|
||||
before=$(mktemp)
|
||||
after=$(mktemp)
|
||||
|
||||
### Parse command line options
|
||||
address=""
|
||||
module=""
|
||||
chain_name=""
|
||||
run_anvil=false
|
||||
skip_compile=false
|
||||
keepalive_anvil=false
|
||||
anvil_out=$(mktemp)
|
||||
while getopts ':hm:c:a:xkdl:' option; do
|
||||
case "$option" in
|
||||
h) usage
|
||||
;;
|
||||
m) module=$OPTARG
|
||||
;;
|
||||
a) address=$OPTARG
|
||||
;;
|
||||
c) chain_name=$OPTARG
|
||||
;;
|
||||
x) run_anvil=true
|
||||
;;
|
||||
d) skip_compile=true
|
||||
;;
|
||||
l) anvil_out=$OPTARG
|
||||
;;
|
||||
k) keepalive_anvil=true
|
||||
run_anvil=true
|
||||
;;
|
||||
:) printf "missing argument for -%s\n" "$OPTARG" >&2
|
||||
usage
|
||||
;;
|
||||
\?) printf "illegal option: -%s\n" "$OPTARG" >&2
|
||||
usage
|
||||
;;
|
||||
esac
|
||||
done
|
||||
shift $((OPTIND - 1))
|
||||
|
||||
[ -z "$chain_name" ] && usage
|
||||
[ -z "$module" ] && usage
|
||||
|
||||
CORE=$(worm contract mainnet "$chain_name" Core)
|
||||
echo "core: $CORE"
|
||||
|
||||
|
||||
GUARDIAN_ADDRESS=0xbeFA429d57cD18b7F8A4d91A2da9AB4AF05d0FBe
|
||||
GUARDIAN_SECRET=cfb12303a19cde580bb4dd771639b0d26bc68353645571a8cff516ab2ee113a0
|
||||
|
||||
ANVIL_PID=""
|
||||
|
||||
function clean_up () {
|
||||
ARG=$?
|
||||
[ -n "$ANVIL_PID" ] && kill "$ANVIL_PID"
|
||||
exit $ARG
|
||||
}
|
||||
trap clean_up EXIT
|
||||
|
||||
|
||||
#TODO: make RPC an optional argument
|
||||
HOST="http://localhost"
|
||||
PORT="8545"
|
||||
RPC="$HOST:$PORT"
|
||||
|
||||
if [[ $run_anvil = true ]]; then
|
||||
anvil --port $PORT --fork-url "$(worm rpc mainnet $chain_name)" --mnemonic "myth like bonus scare over problem client lizard pioneer submit female collect" > "$anvil_out" &
|
||||
ANVIL_PID=$!
|
||||
echo "🍴 Forking mainnet..."
|
||||
echo "Anvil logs in $anvil_out"
|
||||
sleep 5
|
||||
# ps | grep "$ANVIL_PID"
|
||||
fi
|
||||
|
||||
MODULE=""
|
||||
SCRIPT=""
|
||||
case "$module" in
|
||||
bridge|core)
|
||||
MODULE=Core
|
||||
SCRIPT="scripts/deploy_core_bridge.js"
|
||||
;;
|
||||
token_bridge)
|
||||
MODULE=TokenBridge
|
||||
SCRIPT="scripts/deploy_token_bridge.js"
|
||||
;;
|
||||
nft_bridge)
|
||||
MODULE=NFTBridge
|
||||
SCRIPT="scripts/deploy_nft_bridge.js"
|
||||
;;
|
||||
*) echo "unknown module $module" >&2
|
||||
usage
|
||||
;;
|
||||
esac
|
||||
|
||||
CONTRACT=$(worm contract mainnet "$chain_name" "$MODULE")
|
||||
|
||||
if [[ -n "$address" ]]; then
|
||||
new_implementation="$address"
|
||||
else
|
||||
if [[ $skip_compile = false ]]; then
|
||||
echo "🛠 Compiling contract..."
|
||||
build_output=$(npm run build) || ( echo "$build_output" && exit 1 )
|
||||
fi
|
||||
printf "⬆️ Deploying implementation..."
|
||||
deploy_output=$(npx truffle exec $SCRIPT --network development) || ( echo "$deploy_output" && exit 1 )
|
||||
new_implementation=$(echo "$deploy_output" | grep "address:" | cut -d' ' -f3)
|
||||
fi
|
||||
printf " %s\n" "$new_implementation"
|
||||
vaa=$(worm generate upgrade -c "$chain_name" -a "$new_implementation" -m $MODULE -g "$GUARDIAN_SECRET")
|
||||
|
||||
echo "💂 Overriding guardian set with $GUARDIAN_ADDRESS"
|
||||
worm evm hijack -g "$GUARDIAN_ADDRESS" -i 0 -a "$CORE" --rpc "$RPC"> /dev/null
|
||||
|
||||
echo "🔍 Querying old contract state"
|
||||
worm evm info -c "$chain_name" -m $MODULE -n devnet -a "$CONTRACT" --rpc "$RPC" | grep -v '"implementation":' > "$before"
|
||||
echo "🤝 Submitting VAA"
|
||||
worm submit "$vaa" -n devnet -a "$CONTRACT" --rpc "$RPC" > /dev/null
|
||||
echo "🔍 Querying new contract state"
|
||||
worm evm info -c "$chain_name" -m $MODULE -n devnet -a "$CONTRACT" --rpc "$RPC" | grep -v '"implementation":' > "$after"
|
||||
|
||||
git diff --no-index "$before" "$after" --exit-code && echo "✅ Upgrade simulation successful" || exit 1
|
||||
|
||||
if [[ $keepalive_anvil = true ]]; then
|
||||
echo "Listening on $RPC"
|
||||
# tail -f "$anvil_out"
|
||||
wait "$ANVIL_PID"
|
||||
fi
|
||||
|
|
@ -0,0 +1,32 @@
|
|||
#!/bin/bash
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
ANVIL_PID=""
|
||||
function clean_up () {
|
||||
ARG=$?
|
||||
[ -n "$ANVIL_PID" ] && kill "$ANVIL_PID"
|
||||
exit $ARG
|
||||
}
|
||||
trap clean_up EXIT
|
||||
|
||||
echo "🍴 Forking mainnet..."
|
||||
anvil --base-fee 0 --fork-url "$(worm rpc mainnet ethereum)" --mnemonic "myth like bonus scare over problem client lizard pioneer submit female collect" > /dev/null &
|
||||
ANVIL_PID=$!
|
||||
|
||||
sleep 10
|
||||
|
||||
echo "========================= Updating core contract #1 ============================"
|
||||
./simulate_upgrade -m bridge -c ethereum -d
|
||||
echo "========================= Updating core contract #2 ============================"
|
||||
./simulate_upgrade -m bridge -c ethereum -d
|
||||
|
||||
echo "===================== Updating token bridge contract #1 ========================"
|
||||
./simulate_upgrade -m token_bridge -c ethereum -d
|
||||
echo "===================== Updating token bridge contract #2 ========================"
|
||||
./simulate_upgrade -m token_bridge -c ethereum -d
|
||||
|
||||
echo "====================== Updating NFT bridge contract #1 ========================="
|
||||
./simulate_upgrade -m nft_bridge -c ethereum -d
|
||||
echo "====================== Updating NFT bridge contract #2 ========================="
|
||||
./simulate_upgrade -m nft_bridge -c ethereum -d
|
||||
Loading…
Reference in New Issue