prysm-pulse/testing/spectest/utils/config.go
terence 5a66807989
Update to V5 (#13622)
* First take at updating everything to v5

* Patch gRPC gateway to use prysm v5

Fix patch

* Update go ssz

---------

Co-authored-by: Preston Van Loon <pvanloon@offchainlabs.com>
2024-02-15 05:46:47 +00:00

29 lines
842 B
Go

// Package utils allows for easy switching of chain configuration parameters in spec conformity unit tests.
package utils
import (
"errors"
"fmt"
"testing"
"github.com/prysmaticlabs/prysm/v5/config/params"
)
// SetConfig sets the global params for spec tests depending on the option chosen.
// Provides reset function allowing to get back to the previous configuration at the end of a test.
func SetConfig(t testing.TB, config string) error {
params.SetupTestConfigCleanup(t)
switch config {
case "minimal":
params.OverrideBeaconConfig(params.MinimalSpecConfig().Copy())
return nil
case "mainnet":
params.OverrideBeaconConfig(params.MainnetConfig().Copy())
return nil
case "":
return errors.New("no config provided")
default:
return fmt.Errorf("did not receive a valid config, instead received this %s", config)
}
}