hore(target_chains/starknet): automate test data generation (#1587)

pyth-network · May 21, 2024 · 74f02c4 · 74f02c4
1 parent f4cc195
commit 74f02c4
Show file tree

Hide file tree

Showing 10 changed files with 815 additions and 783 deletions.
diff --git a/.github/workflows/ci-starknet-tools.yml b/.github/workflows/ci-starknet-tools.yml
@@ -21,9 +21,12 @@ jobs:
         run: cargo fmt --manifest-path ./target_chains/starknet/tools/test_vaas/Cargo.toml -- --check
       - name: Run clippy
         run: cargo clippy --manifest-path ./target_chains/starknet/tools/test_vaas/Cargo.toml --all-targets -- --deny warnings
-      - name: Run wormhole_mainnet_upgrades binary
-        run: cargo run --manifest-path ./target_chains/starknet/tools/test_vaas/Cargo.toml --bin wormhole_mainnet_upgrades
       - name: Run generate_keypair binary
         run: cargo run --manifest-path ./target_chains/starknet/tools/test_vaas/Cargo.toml --bin generate_keypair
-      - name: Run generate_wormhole_vaas binary
-        run: cargo run --manifest-path ./target_chains/starknet/tools/test_vaas/Cargo.toml --bin generate_wormhole_vaas
+      - name: Check test data
+        run: |
+          cargo run --manifest-path ./target_chains/starknet/tools/test_vaas/Cargo.toml --bin generate_test_data > /tmp/data.cairo
+          if ! diff ./target_chains/starknet/contracts/tests/data.cairo /tmp/data.cairo; then
+            >&2 echo "Re-run generate_test_data to update data.cairo"
+            exit 1
+          fi
diff --git a/target_chains/starknet/contracts/README.md b/target_chains/starknet/contracts/README.md
@@ -13,14 +13,12 @@ The `.tool-versions` file in this directory specifies the tool versions used by
 
 Run `snforge test` in this directory to run the contract unit tests and integration tests.
 
-Some tests contain input data that was generated with `test_vaas` tool.
-To run it, use the following commands from the repository root (requires Rust installation):
+The `tests/data.cairo` file contains input data that was generated with `test_vaas` tool.
+To regenerate it, run the following commands from the repository root (requires Rust installation):
 
 ```
 cd target_chains/starknet/tools/test_vaas
-cargo run --bin re_sign_price_updates
-cargo run --bin generate_wormhole_vaas
-cargo run --bin wormhole_mainnet_upgrades
+cargo run --bin generate_test_data > ../../contracts/tests/data.cairo
 ```
 
 ## Formatting

diff --git a/target_chains/starknet/contracts/tests/data.cairo b/target_chains/starknet/contracts/tests/data.cairo
diff --git a/target_chains/starknet/contracts/tests/pyth.cairo b/target_chains/starknet/contracts/tests/pyth.cairo
diff --git a/target_chains/starknet/contracts/tests/wormhole.cairo b/target_chains/starknet/contracts/tests/wormhole.cairo
diff --git a/target_chains/starknet/tools/test_vaas/src/bin/generate_test_data.rs b/target_chains/starknet/tools/test_vaas/src/bin/generate_test_data.rs
diff --git a/target_chains/starknet/tools/test_vaas/src/bin/generate_wormhole_vaas.rs b/target_chains/starknet/tools/test_vaas/src/bin/generate_wormhole_vaas.rs
diff --git a/target_chains/starknet/tools/test_vaas/src/bin/re_sign_price_updates.rs b/target_chains/starknet/tools/test_vaas/src/bin/re_sign_price_updates.rs