fixup: Respond to joshua-kim's feedback

ava-labs · May 21, 2024 · b1f49af · b1f49af
1 parent 665efee
commit b1f49af
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 16 deletions.
diff --git a/tests/antithesis/avalanchego/main.go b/tests/antithesis/avalanchego/main.go
@@ -101,8 +101,7 @@ func main() {
 			},
 		}})
 		if err != nil {
-			log.Printf("failed to issue initial funding X-chain baseTx: %s", err)
-			return
+			log.Fatalf("failed to issue initial funding X-chain baseTx: %s", err)
 		}
 		log.Printf("issued initial funding X-chain baseTx %s in %s", baseTx.ID(), time.Since(baseStartTime))
 

diff --git a/tests/antithesis/xsvm/main.go b/tests/antithesis/xsvm/main.go
@@ -77,8 +77,7 @@ func main() {
 			},
 		)
 		if err != nil {
-			log.Printf("failed to issue initial funding transfer: %s", err)
-			return
+			log.Fatalf("failed to issue initial funding transfer: %s", err)
 		}
 		log.Printf("issued initial funding transfer %s in %s", transferTxStatus.TxID, time.Since(baseStartTime))
 

diff --git a/tests/e2e/e2e_test.go b/tests/e2e/e2e_test.go
@@ -38,7 +38,7 @@ var _ = ginkgo.SynchronizedBeforeSuite(func() []byte {
 	// Run only once in the first ginkgo process
 
 	nodes := tmpnet.NewNodesOrPanic(flagVars.NodeCount())
-	subnets := vms.XSVMSubnets(nodes...)
+	subnets := vms.XSVMSubnetsOrPanic(nodes...)
 	return e2e.NewTestEnvironment(
 		flagVars,
 		&tmpnet.Network{

diff --git a/tests/e2e/vms/xsvm.go b/tests/e2e/vms/xsvm.go
@@ -27,10 +27,14 @@ var (
 	subnetBName = "xsvm-b"
 )
 
-func XSVMSubnets(nodes ...*tmpnet.Node) []*tmpnet.Subnet {
+func XSVMSubnetsOrPanic(nodes ...*tmpnet.Node) []*tmpnet.Subnet {
+	key, err := secp256k1.NewPrivateKey()
+	if err != nil {
+		panic(err)
+	}
 	return []*tmpnet.Subnet{
-		subnet.NewXSVMOrPanic(subnetAName, nil /* key, will be generated */, nodes...),
-		subnet.NewXSVMOrPanic(subnetBName, nil /* key, will be generated */, nodes...),
+		subnet.NewXSVMOrPanic(subnetAName, key, nodes...),
+		subnet.NewXSVMOrPanic(subnetBName, key, nodes...),
 	}
 }
 

diff --git a/tests/fixture/subnet/xsvm.go b/tests/fixture/subnet/xsvm.go
@@ -18,14 +18,6 @@ func NewXSVMOrPanic(name string, key *secp256k1.PrivateKey, nodes ...*tmpnet.Nod
 		panic("a subnet must be validated by at least one node")
 	}
 
-	if key == nil {
-		var err error
-		key, err = secp256k1.NewPrivateKey()
-		if err != nil {
-			panic(err)
-		}
-	}
-
 	genesisBytes, err := genesis.Codec.Marshal(genesis.CodecVersion, &genesis.Genesis{
 		Timestamp: time.Now().Unix(),
 		Allocations: []genesis.Allocation{