From 4fbe4e1dbbf27cdc24e03241353c385af5939ad0 Mon Sep 17 00:00:00 2001 From: Andrea Franz Date: Thu, 30 Jul 2026 11:25:09 +0000 Subject: [PATCH] test(amm-ui): bootstrap deterministic test wallet in testnet setup --- apps/amm/tests/testnet/setup-amm-testnet.sh | 214 +++++++++++++------- 1 file changed, 142 insertions(+), 72 deletions(-) diff --git a/apps/amm/tests/testnet/setup-amm-testnet.sh b/apps/amm/tests/testnet/setup-amm-testnet.sh index 92b88ac..7f92174 100755 --- a/apps/amm/tests/testnet/setup-amm-testnet.sh +++ b/apps/amm/tests/testnet/setup-amm-testnet.sh @@ -8,27 +8,30 @@ # AMM UI swap test (apps/amm/tests/swap.mjs) exercises; run it once to stand up # a swappable pool, then launch the UI / run the test. # +# DETERMINISTIC TEST WALLET: by default the script bootstraps an ISOLATED wallet +# (git-ignored, under this folder) by restoring it from a fixed BIP-39 mnemonic +# and creating its accounts in a fixed order. The wallet's BIP-32 key tree makes +# those account ids reproducible across machines, so the whole team gets the +# same token/holding ids — and the script auto-writes apps/amm/amm-tokens.json +# from them for the UI. It never touches your personal ~/.lee/wallet. +# # Program IDs and all AMM PDAs (config, pool, vaults, LP, tick) are DERIVED at # runtime from the deployed binaries + the token definition accounts, so this # script stays correct even if you rebuild the guest binaries (new image id => -# new program id => new PDAs). Only the input accounts below are hand-picked. +# new program id => new PDAs). # -# TEST WALLET (isolated): by default this script points the wallet at a -# dedicated, git-ignored home under this folder (apps/amm/tests/testnet/.wallet) -# so it never touches your personal ~/.lee/wallet. Override with TEST_WALLET_HOME. -# -# Prerequisites (all managed by you, outside this script): +# Prerequisites (managed by you, outside this script): # - `wallet` and `spel` on PATH (from the SPEL toolchain) -# - a wallet initialized at TEST_WALLET_HOME, funded and synced, that OWNS the -# account ids configured below (see "TEST ACCOUNTS" — these must be derived -# from / created in the test wallet; see docs/testnet-runbook.md §2) +# - a reachable, funded sequencer (set TEST_SEQUENCER_ADDR, or pre-configure +# the wallet). Account creation is local, but deploys/mints need funds. # - `cargo` (to run the amm_pdas example) # - the guest .bin files already built (`make build-programs`) # - the IDL files under artifacts/ (`make idl`) # # Usage (from anywhere in the repo): # apps/amm/tests/testnet/setup-amm-testnet.sh -# TEST_WALLET_HOME=/tmp/lez-test-wallet apps/amm/tests/testnet/setup-amm-testnet.sh +# TEST_SEQUENCER_ADDR=http://127.0.0.1:8080 apps/amm/tests/testnet/setup-amm-testnet.sh +# FORCE_BOOTSTRAP=1 ... # re-restore the test wallet (rewrites its storage) # set -euo pipefail @@ -39,23 +42,31 @@ REPO_ROOT="${REPO_ROOT:-$(git -C "$SCRIPT_DIR" rev-parse --show-toplevel 2>/dev/ cd "$REPO_ROOT" ############################################################################### -# TEST WALLET — isolated from your personal wallet +# TEST WALLET — isolated + deterministic ############################################################################### # Dedicated wallet home for tests. Kept out of git (see .gitignore). Exported -# under both env var names so whichever the installed `wallet`/`spel` expects -# is satisfied. +# under both env var names so whichever the installed `wallet`/`spel` expects is +# satisfied (the current wallet uses LEE_WALLET_HOME_DIR). TEST_WALLET_HOME="${TEST_WALLET_HOME:-$SCRIPT_DIR/.wallet}" export LEE_WALLET_HOME_DIR="$TEST_WALLET_HOME" export NSSA_WALLET_HOME_DIR="$TEST_WALLET_HOME" -# The deterministic test mnemonic. A wallet restored from this seed always -# yields the same account ids, which is why the ids below can be pinned and -# shared with the team. Keep in sync with the wallet at TEST_WALLET_HOME. +# The deterministic test seed. A wallet restored from this mnemonic yields the +# same account ids every time, which is why they can be shared/pinned. TEST_MNEMONIC="${TEST_MNEMONIC:-test test test test test test test test test test test junk}" +TEST_WALLET_PASSWORD="${TEST_WALLET_PASSWORD:-test}" +TEST_WALLET_DEPTH="${TEST_WALLET_DEPTH:-3}" +# Optional: point the test wallet's config at your sequencer. Leave empty to use +# whatever the wallet home is already configured with. +TEST_SEQUENCER_ADDR="${TEST_SEQUENCER_ADDR:-}" + +# Deterministic accounts, created in THIS fixed order after a fresh restore so +# their ids are reproducible. Resolved to ids at runtime via `wallet account id`. +ACCOUNT_LABELS=(token-a-def token-a-holding token-b-def token-b-holding lp-holding) ############################################################################### -# CONFIG — edit these +# CONFIG — non-account parameters (edit freely) ############################################################################### # --- Program binaries (docker release builds; image ids must match deployment) --- @@ -67,38 +78,20 @@ TWAP_BIN="programs/twap_oracle/methods/guest/target/riscv32im-risc0-zkvm-elf/doc TOKEN_IDL="artifacts/token-idl.json" AMM_IDL="artifacts/amm-idl.json" -# --- TEST ACCOUNTS (must be owned by the wallet at TEST_WALLET_HOME) ---------- -# These are the deterministic accounts the test suite relies on. See the -# "Deterministic test accounts" note at the top of the file / the PR discussion: -# once the test wallet is restored from TEST_MNEMONIC, capture the ids it derives -# and pin them here so amm-tokens.json and swap.mjs can reference known accounts. - -# --- Token A --- -TOKEN_A_NAME="TOKEN A" -TOKEN_A_SUPPLY="1000000000000000000000" -TOKEN_A_DEF="6gQMF38wEDPEnUamLmNqN8opss5xvr9w55eddddMmpEK" # --definition-target-account -TOKEN_A_HOLDING="EF6c3VtgzUzpxfEKpvR3ZxsCch8y5YqazmWQu6Bybvyq" # --holding-target-account -TOKEN_A_MINT_AUTH="EF6c3VtgzUzpxfEKpvR3ZxsCch8y5YqazmWQu6Bybvyq" - -# --- Token B --- -TOKEN_B_NAME="TOKEN B" -TOKEN_B_SUPPLY="1000000000000000000000" -TOKEN_B_DEF="Hw65ZimauFtHtvmDvWb24aWi1E6uRCifdyEn46Ei8wAc" -TOKEN_B_HOLDING="L3hr1xABT3gkSYR1Fztu2smyJ1ftXcivjM8LguMjAsp" -TOKEN_B_MINT_AUTH="L3hr1xABT3gkSYR1Fztu2smyJ1ftXcivjM8LguMjAsp" - -# --- AMM / pool inputs --- -AMM_AUTHORITY="EF6c3VtgzUzpxfEKpvR3ZxsCch8y5YqazmWQu6Bybvyq" -USER_HOLDING_A="$TOKEN_A_HOLDING" -USER_HOLDING_B="$TOKEN_B_HOLDING" -USER_HOLDING_LP="7taJs6YpzBcWc5jiZcnZ817nU1G5WnQtBHskbcMDUc6A" -CLOCK_ACCOUNT="4BdcjoXkq786TMWcBGGHqcxeLYMZmn17rL4eM9ZyRWNU" +# --- Token metadata --- +TOKEN_A_NAME="TOKEN A"; TOKEN_A_SYMBOL="TKA"; TOKEN_A_SUPPLY="1000000000000000000000"; TOKEN_A_DECIMALS=18 +TOKEN_B_NAME="TOKEN B"; TOKEN_B_SYMBOL="TKB"; TOKEN_B_SUPPLY="1000000000000000000000"; TOKEN_B_DECIMALS=18 +# --- Pool inputs --- +CLOCK_ACCOUNT="4BdcjoXkq786TMWcBGGHqcxeLYMZmn17rL4eM9ZyRWNU" # canonical LEZ system clock POOL_TOKEN_A_AMOUNT="10000" POOL_TOKEN_B_AMOUNT="10000" POOL_FEES="1" POOL_DEADLINE="18446744073709551615" +# Where the UI token config is written (git-ignored; consumed via TOKENS_CONFIG). +TOKENS_CONFIG_OUT="apps/amm/amm-tokens.json" + ############################################################################### # Helpers ############################################################################### @@ -136,9 +129,6 @@ run_tx() { die "command exited with status $rc — $desc" fi - # The confirmation line printed by spel is: - # ✅ Transaction confirmed — included in a block. - # Match on the two stable substrings to be robust to emoji/dash encoding. if grep -q "Transaction confirmed" "$tmp" && grep -q "included in a block" "$tmp"; then log "${GRN}✅ CONFIRMED — included in a block: ${desc}${RST}" rm -f "$tmp" @@ -148,7 +138,7 @@ run_tx() { rm -f "$tmp" if [ "$mode" = "soft" ]; then log "${YEL}⚠ no '✅ Transaction confirmed — included in a block.' marker for: ${desc}" - log " (continuing — 'wallet deploy-program' may not print the spel marker; verify manually)${RST}" + log " (continuing — some commands don't print the spel marker; verify manually)${RST}" return 0 fi die "NOT CONFIRMED — expected '✅ Transaction confirmed — included in a block.' for: $desc" @@ -171,8 +161,47 @@ program_id() { printf '%s' "$pid" } +# Resolve a wallet account id (bare base58) from its label. Deterministic under +# the test mnemonic. Returns non-zero if the label isn't registered yet. +acct_id() { + local label="$1" out + out="$(wallet account id --account-id "$label" 2>/dev/null)" || return 1 + printf '%s' "$out" | grep -oE '[1-9A-HJ-NP-Za-km-z]{32,44}' | head -n1 +} + +# Restore the isolated test wallet from the fixed mnemonic and register the +# deterministic accounts. `restore-keys` REWRITES storage (safe here — it's a +# throwaway test home). Idempotent: skips accounts that already resolve. +bootstrap_test_wallet() { + sec "Bootstrap deterministic test wallet" + kv "wallet home" "$TEST_WALLET_HOME" + mkdir -p "$TEST_WALLET_HOME" + + if [ -n "$TEST_SEQUENCER_ADDR" ]; then + log "${DIM}\$ wallet config set sequencer_addr $TEST_SEQUENCER_ADDR${RST}" + wallet config set sequencer_addr "$TEST_SEQUENCER_ADDR" \ + || log "${YEL}⚠ 'wallet config set sequencer_addr' failed — configure the wallet manually.${RST}" + fi + + # restore-keys reads the mnemonic then the password from stdin. + log "${DIM}\$ printf '\\n\\n' | wallet restore-keys --depth $TEST_WALLET_DEPTH${RST}" + printf '%s\n%s\n' "$TEST_MNEMONIC" "$TEST_WALLET_PASSWORD" \ + | wallet restore-keys --depth "$TEST_WALLET_DEPTH" \ + || die "wallet restore-keys failed" + + local label + for label in "${ACCOUNT_LABELS[@]}"; do + if acct_id "$label" >/dev/null 2>&1; then + kv "exists" "$label" + else + log "${DIM}\$ wallet account new public --label $label${RST}" + wallet account new public --label "$label" || die "failed to create account: $label" + fi + done +} + ############################################################################### -# 0. Preflight +# 0. Preflight + wallet bootstrap ############################################################################### sec "Preflight" require_cmd wallet @@ -181,26 +210,47 @@ require_cmd cargo require_file "$TOKEN_BIN"; require_file "$AMM_BIN"; require_file "$TWAP_BIN" require_file "$TOKEN_IDL"; require_file "$AMM_IDL" kv "repo root" "$REPO_ROOT" -kv "test wallet home" "$TEST_WALLET_HOME" -if [ ! -d "$TEST_WALLET_HOME" ]; then - die "test wallet home not found: $TEST_WALLET_HOME - Initialize a wallet there from the test mnemonic, then create/pin the TEST - ACCOUNTS above. See docs/testnet-runbook.md §2. Mnemonic: - \"$TEST_MNEMONIC\"" +kv "token bin" "$TOKEN_BIN"; kv "amm bin" "$AMM_BIN"; kv "twap bin" "$TWAP_BIN" + +if [ ! -d "$TEST_WALLET_HOME" ] || [ "${FORCE_BOOTSTRAP:-0}" = "1" ]; then + bootstrap_test_wallet +else + kv "test wallet" "reusing $TEST_WALLET_HOME (FORCE_BOOTSTRAP=1 to re-restore)" fi -kv "token bin" "$TOKEN_BIN" -kv "amm bin" "$AMM_BIN" -kv "twap bin" "$TWAP_BIN" ############################################################################### -# 1. Deploy programs +# 1. Resolve the deterministic test accounts +############################################################################### +sec "Resolve deterministic test accounts (from the test mnemonic)" +TOKEN_A_DEF="$(acct_id token-a-def)" || die "token-a-def not registered — run with FORCE_BOOTSTRAP=1" +TOKEN_A_HOLDING="$(acct_id token-a-holding)" || die "token-a-holding not registered" +TOKEN_B_DEF="$(acct_id token-b-def)" || die "token-b-def not registered" +TOKEN_B_HOLDING="$(acct_id token-b-holding)" || die "token-b-holding not registered" +USER_HOLDING_LP="$(acct_id lp-holding)" || die "lp-holding not registered" +for v in TOKEN_A_DEF TOKEN_A_HOLDING TOKEN_B_DEF TOKEN_B_HOLDING USER_HOLDING_LP; do + [ -n "${!v}" ] || die "failed to resolve account id for $v" +done + +# Derived roles (the input holding signs; mint authority == holding; authority is the A holding). +TOKEN_A_MINT_AUTH="$TOKEN_A_HOLDING"; TOKEN_B_MINT_AUTH="$TOKEN_B_HOLDING" +AMM_AUTHORITY="$TOKEN_A_HOLDING" +USER_HOLDING_A="$TOKEN_A_HOLDING"; USER_HOLDING_B="$TOKEN_B_HOLDING" + +kv "token-a-def" "$TOKEN_A_DEF" +kv "token-a-holding" "$TOKEN_A_HOLDING" +kv "token-b-def" "$TOKEN_B_DEF" +kv "token-b-holding" "$TOKEN_B_HOLDING" +kv "lp-holding" "$USER_HOLDING_LP" + +############################################################################### +# 2. Deploy programs ############################################################################### run_tx soft "deploy token program" -- wallet deploy-program "$TOKEN_BIN" run_tx soft "deploy amm program" -- wallet deploy-program "$AMM_BIN" run_tx soft "deploy twap_oracle program" -- wallet deploy-program "$TWAP_BIN" ############################################################################### -# 2. Derive program IDs +# 3. Derive program IDs ############################################################################### sec "Program IDs (derived from the deployed binaries)" TOKEN_PID="$(program_id "$TOKEN_BIN")"; kv "token program id" "$TOKEN_PID" @@ -208,7 +258,7 @@ AMM_PID="$(program_id "$AMM_BIN")"; kv "amm program id" "$AMM_PID" TWAP_PID="$(program_id "$TWAP_BIN")"; kv "twap program id" "$TWAP_PID" ############################################################################### -# 3. Create token definitions (mint supply to the holding accounts) +# 4. Create token definitions (mint supply to the holding accounts) ############################################################################### run_tx strict "create fungible definition: $TOKEN_A_NAME" -- \ spel --idl "$TOKEN_IDL" --program "$TOKEN_BIN" -- new-fungible-definition \ @@ -225,7 +275,7 @@ run_tx strict "create fungible definition: $TOKEN_B_NAME" -- \ --mint-authority "$TOKEN_B_MINT_AUTH" ############################################################################### -# 4. Verify token definitions & holdings +# 5. Verify token definitions & holdings ############################################################################### inspect "$TOKEN_IDL" "$TOKEN_A_DEF" "TokenDefinition" inspect "$TOKEN_IDL" "$TOKEN_A_HOLDING" "TokenHolding" @@ -233,7 +283,7 @@ inspect "$TOKEN_IDL" "$TOKEN_B_DEF" "TokenDefinition" inspect "$TOKEN_IDL" "$TOKEN_B_HOLDING" "TokenHolding" ############################################################################### -# 5. Derive AMM PDAs from the program ids + token pair +# 6. Derive AMM PDAs from the program ids + token pair ############################################################################### sec "Deriving AMM PDAs (amm_pdas example)" log "${DIM}\$ cargo run -q -p amm_program --example amm_pdas -- $AMM_PID $TWAP_PID $TOKEN_A_DEF $TOKEN_B_DEF${RST}" @@ -262,13 +312,9 @@ kv "vault_b" "$VAULT_B" kv "pool_definition_lp" "$POOL_LP" kv "lp_lock_holding" "$LP_LOCK" kv "current_tick_account" "$TICK" -kv "user_holding_a" "$USER_HOLDING_A" -kv "user_holding_b" "$USER_HOLDING_B" -kv "user_holding_lp" "$USER_HOLDING_LP" -kv "clock" "$CLOCK_ACCOUNT" ############################################################################### -# 6. Initialize the AMM +# 7. Initialize the AMM ############################################################################### run_tx strict "initialize AMM config" -- \ spel --idl "$AMM_IDL" --program "$AMM_BIN" -- initialize \ @@ -278,7 +324,7 @@ run_tx strict "initialize AMM config" -- \ --authority "$AMM_AUTHORITY" ############################################################################### -# 7. Create the pool (seed initial liquidity) +# 8. Create the pool (seed initial liquidity) ############################################################################### run_tx strict "create pool + seed liquidity" -- \ spel --idl "$AMM_IDL" --program "$AMM_BIN" -- new-definition \ @@ -299,16 +345,40 @@ run_tx strict "create pool + seed liquidity" -- \ --deadline "$POOL_DEADLINE" ############################################################################### -# 8. Verify the pool +# 9. Verify the pool ############################################################################### inspect "$AMM_IDL" "$POOL" "PoolDefinition" +############################################################################### +# 10. Write the UI token config from the deterministic accounts +############################################################################### +sec "Write UI token config -> $TOKENS_CONFIG_OUT" +cat > "$TOKENS_CONFIG_OUT" <