prysm-pulse/testing/spectest/utils/config.go
terencechain d17996f8b0
Update to V4 🚀 (#12134)
* Update V3 from V4

* Fix build v3 -> v4

* Update ssz

* Update beacon_chain.pb.go

* Fix formatter import

* Update update-mockgen.sh comment to v4

* Fix conflicts. Pass build and tests

* Fix test
2023-03-17 18:52:56 +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/v4/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)
}
}