hf test go: support passing in fork method flag

Author: glyh

scripts/hardfork/build-and-test.sh

@@ -13,9 +13,10 @@
 set -eux -o pipefail
 
 PREFORK=""
+FORK_METHOD="legacy"
 
 usage() {
-  echo "Usage: $0 --fork-from <PREFORK>"
+  echo "Usage: $0 --fork-from <PREFORK> --fork-method <FORK_METHOD>"
   exit 1
 }
 
@@ -31,6 +32,23 @@ while [[ $# -gt 0 ]]; do
       PREFORK="$2"
       shift 2
       ;;
+    --fork-method)
+      # ensure value exists
+      if [[ $# -lt 2 ]]; then
+        echo "Error: --fork-from requires an argument."
+        usage
+      fi
+      case "$2" in
+        legacy|advanced-generate-hf-config)
+          FORK_METHOD="$2"
+          ;;
+        *)
+          echo "Error: --fork-method must be either 'legacy' or 'advanced-generate-hf-config'."
+          usage
+          ;;
+      esac
+      shift 2
+      ;;
     --help|-h)
       usage
       ;;
@@ -51,6 +69,11 @@ if [[ -z "$PREFORK" ]]; then
   usage
 fi
 
+if [[ "$FORK_METHOD" == "advanced-generate-hf-config" ]]; then
+  echo "Error: unimplemented fork mode 'advanced-generate-hf-config'"
+  usage
+fi
+
 NIX_OPTS=( --accept-flake-config --experimental-features 'nix-command flakes' )
 
 if [[ -n "${NIX_CACHE_NAR_SECRET:-}" ]]; then
@@ -152,6 +175,7 @@ hardfork_test/bin/hardfork_test \
   --slot-tx-end "$SLOT_TX_END" \
   --slot-chain-end "$SLOT_CHAIN_END" \
   --script-dir "$SCRIPT_DIR" \
-  --root "$NETWORK_ROOT"
+  --root "$NETWORK_ROOT" \
+  --fork-method "$FORK_METHOD"
 
 popd

src/app/hardfork_test/src/internal/app/root.go

@@ -85,6 +85,7 @@ func init() {
 	rootCmd.Flags().IntVar(&cfg.ForkEarliestBlockMaxRetries, "fork-earliest-block-max-retries", cfg.ForkEarliestBlockMaxRetries, "Maximum number of retries to wait for earliest block in fork network")
 	rootCmd.Flags().IntVar(&cfg.HTTPClientTimeoutSeconds, "http-timeout", cfg.HTTPClientTimeoutSeconds, "HTTP client timeout in seconds for GraphQL requests")
 	rootCmd.Flags().IntVar(&cfg.ClientMaxRetries, "client-max-retries", cfg.ClientMaxRetries, "Maximum number of retries for client requests")
+	rootCmd.Flags().Var(&cfg.ForkMethod, "mode", "The implementation of fork")
 
 	// Mark required flags
 	rootCmd.MarkFlagRequired("main-mina-exe")

src/app/hardfork_test/src/internal/config/config.go

@@ -61,6 +61,8 @@ type Config struct {
 	ForkEarliestBlockMaxRetries   int // Max retries to wait for earliest block in fork network
 	HTTPClientTimeoutSeconds      int // HTTP client timeout for GraphQL requests
 	ClientMaxRetries              int // Max number of retries for client requests
+
+	ForkMethod ForkMethod
 }
 
 // DefaultConfig returns the default configuration with values

src/app/hardfork_test/src/internal/config/fork_method.go

@@ -0,0 +1,48 @@
+package config
+
+import (
+	"fmt"
+	"sort"
+	"strings"
+)
+
+type ForkMethod int
+
+const (
+	Legacy ForkMethod = iota
+)
+
+var forkMethodToString = map[ForkMethod]string{
+	Legacy: "legacy",
+}
+
+var stringToForkMethod = map[string]ForkMethod{
+	"legacy": Legacy,
+}
+
+func (m *ForkMethod) String() string {
+	if s, ok := forkMethodToString[*m]; ok {
+		return s
+	}
+	return "unknown"
+}
+
+func validKeys(m map[string]ForkMethod) string {
+	keys := make([]string, 0, len(m))
+	for k := range m {
+		keys = append(keys, k)
+	}
+	sort.Strings(keys)
+	return strings.Join(keys, "|")
+}
+
+func (m *ForkMethod) Set(s string) error {
+	v, ok := stringToForkMethod[s]
+	if !ok {
+		return fmt.Errorf("invalid mode %q (valid: %s)", s, validKeys(stringToForkMethod))
+	}
+	*m = v
+	return nil
+}
+
+func (m *ForkMethod) Type() string { return "fork method" }