-
Notifications
You must be signed in to change notification settings - Fork 305
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
13 changed files
with
303 additions
and
132 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,95 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "log" | ||
| "time" | ||
|
|
||
| "github.com/celestiaorg/celestia-app/v3/app" | ||
| v2 "github.com/celestiaorg/celestia-app/v3/pkg/appconsts/v2" | ||
| v3 "github.com/celestiaorg/celestia-app/v3/pkg/appconsts/v3" | ||
| "github.com/celestiaorg/celestia-app/v3/test/e2e/testnet" | ||
| ) | ||
|
|
||
| func MajorUpgradeToV3(logger *log.Logger) error { | ||
| numNodes := 4 | ||
| upgradeHeightV3 := int64(20) | ||
| ctx, cancel := context.WithCancel(context.Background()) | ||
| defer cancel() | ||
|
|
||
| logger.Println("Creating testnet") | ||
| testNet, err := testnet.New(ctx, "MajorUpgradeToV3", seed, nil, "test") | ||
| testnet.NoError("failed to create testnet", err) | ||
|
|
||
| defer testNet.Cleanup(ctx) | ||
|
|
||
| // HACKHACK: use a version of celestia-app built from a commit on this PR. | ||
| // This can be removed after the PR is merged to main and we override the | ||
| // upgrade height delay to one block in a new Docker image. | ||
| version := "1a20c01" | ||
|
|
||
| logger.Println("Running major upgrade to v3 test", "version", version) | ||
|
|
||
| consensusParams := app.DefaultConsensusParams() | ||
| consensusParams.Version.AppVersion = v2.Version // Start the test on v2 | ||
| testNet.SetConsensusParams(consensusParams) | ||
|
|
||
| preloader, err := testNet.NewPreloader() | ||
| testnet.NoError("failed to create preloader", err) | ||
|
|
||
| err = preloader.AddImage(ctx, testnet.DockerImageName(version)) | ||
| testnet.NoError("failed to add image", err) | ||
| defer func() { _ = preloader.EmptyImages(ctx) }() | ||
|
|
||
| logger.Println("Creating genesis nodes") | ||
| for i := 0; i < numNodes; i++ { | ||
| err := testNet.CreateGenesisNode(ctx, version, 10000000, 0, testnet.DefaultResources, true) | ||
| testnet.NoError("failed to create genesis node", err) | ||
| } | ||
|
|
||
| logger.Println("Creating txsim") | ||
| endpoints, err := testNet.RemoteGRPCEndpoints(ctx) | ||
| testnet.NoError("failed to get remote gRPC endpoints", err) | ||
| upgradeSchedule := map[int64]uint64{ | ||
| upgradeHeightV3: v3.Version, | ||
| } | ||
|
|
||
| err = testNet.CreateTxClient(ctx, "txsim", version, 1, "100-2000", 100, testnet.DefaultResources, endpoints[0], upgradeSchedule) | ||
| testnet.NoError("failed to create tx client", err) | ||
|
|
||
| logger.Println("Setting up testnet") | ||
| testnet.NoError("Failed to setup testnet", testNet.Setup(ctx)) | ||
| logger.Println("Starting testnet") | ||
| testnet.NoError("Failed to start testnet", testNet.Start(ctx)) | ||
|
|
||
| timer := time.NewTimer(10 * time.Minute) | ||
| defer timer.Stop() | ||
| ticker := time.NewTicker(3 * time.Second) | ||
| defer ticker.Stop() | ||
|
|
||
| logger.Println("waiting for upgrade") | ||
| for _, node := range testNet.Nodes() { | ||
| client, err := node.Client() | ||
| testnet.NoError("failed to get client", err) | ||
|
|
||
| upgradeComplete := false | ||
| lastHeight := int64(0) | ||
| for !upgradeComplete { | ||
| select { | ||
| case <-timer.C: | ||
| return fmt.Errorf("failed to upgrade to v3, last height: %d", lastHeight) | ||
| case <-ticker.C: | ||
| resp, err := client.Header(ctx, nil) | ||
| testnet.NoError("failed to get header", err) | ||
| if resp.Header.Version.App == v3.Version { | ||
| upgradeComplete = true | ||
| } | ||
| logger.Printf("height %v", resp.Header.Height) | ||
| lastHeight = resp.Header.Height | ||
| } | ||
| } | ||
| } | ||
|
|
||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,37 @@ | ||
| package testnet | ||
|
|
||
| import ( | ||
| "io" | ||
| "math/rand" | ||
|
|
||
| "github.com/tendermint/tendermint/crypto" | ||
| "github.com/tendermint/tendermint/crypto/ed25519" | ||
| "github.com/tendermint/tendermint/crypto/secp256k1" | ||
| ) | ||
|
|
||
| type keyGenerator struct { | ||
| random *rand.Rand | ||
| } | ||
|
|
||
| func newKeyGenerator(seed int64) *keyGenerator { | ||
| return &keyGenerator{ | ||
| random: rand.New(rand.NewSource(seed)), //nolint:gosec | ||
| } | ||
| } | ||
|
|
||
| func (g *keyGenerator) Generate(keyType string) crypto.PrivKey { | ||
| seed := make([]byte, ed25519.SeedSize) | ||
|
|
||
| _, err := io.ReadFull(g.random, seed) | ||
| if err != nil { | ||
| panic(err) // this shouldn't happen | ||
| } | ||
| switch keyType { | ||
| case "secp256k1": | ||
| return secp256k1.GenPrivKeySecp256k1(seed) | ||
| case "", "ed25519": | ||
| return ed25519.GenPrivKeyFromSecret(seed) | ||
| default: | ||
| panic("KeyType not supported") // should not make it this far | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.