Commit 8b87c2d2 authored by Brian Bland's avatar Brian Bland

Simplify e2e test Close

parent 99c75b47
package op_e2e package op_e2e
import ( import (
"context"
"math" "math"
"math/big" "math/big"
"testing" "testing"
...@@ -30,7 +29,7 @@ func TestERC20BridgeDeposits(t *testing.T) { ...@@ -30,7 +29,7 @@ func TestERC20BridgeDeposits(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
log := testlog.Logger(t, log.LvlInfo) log := testlog.Logger(t, log.LvlInfo)
log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time) log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time)
......
...@@ -214,11 +214,13 @@ type System struct { ...@@ -214,11 +214,13 @@ type System struct {
Mocknet mocknet.Mocknet Mocknet mocknet.Mocknet
} }
func (sys *System) Close(ctx context.Context) { func (sys *System) Close() {
if sys.L2OutputSubmitter != nil { if sys.L2OutputSubmitter != nil {
sys.L2OutputSubmitter.Stop() sys.L2OutputSubmitter.Stop()
} }
if sys.BatchSubmitter != nil { if sys.BatchSubmitter != nil {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
sys.BatchSubmitter.StopIfRunning(ctx) sys.BatchSubmitter.StopIfRunning(ctx)
} }
......
...@@ -73,7 +73,7 @@ func TestL2OutputSubmitter(t *testing.T) { ...@@ -73,7 +73,7 @@ func TestL2OutputSubmitter(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l1Client := sys.Clients["l1"] l1Client := sys.Clients["l1"]
...@@ -146,7 +146,7 @@ func TestSystemE2E(t *testing.T) { ...@@ -146,7 +146,7 @@ func TestSystemE2E(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
log := testlog.Logger(t, log.LvlInfo) log := testlog.Logger(t, log.LvlInfo)
log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time) log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time)
...@@ -261,7 +261,7 @@ func TestConfirmationDepth(t *testing.T) { ...@@ -261,7 +261,7 @@ func TestConfirmationDepth(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
log := testlog.Logger(t, log.LvlInfo) log := testlog.Logger(t, log.LvlInfo)
log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time) log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time)
...@@ -320,7 +320,7 @@ func TestPendingGasLimit(t *testing.T) { ...@@ -320,7 +320,7 @@ func TestPendingGasLimit(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
log := testlog.Logger(t, log.LvlInfo) log := testlog.Logger(t, log.LvlInfo)
log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time) log.Info("genesis", "l2", sys.RollupConfig.Genesis.L2, "l1", sys.RollupConfig.Genesis.L1, "l2_time", sys.RollupConfig.Genesis.L2Time)
...@@ -365,7 +365,7 @@ func TestFinalize(t *testing.T) { ...@@ -365,7 +365,7 @@ func TestFinalize(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
...@@ -392,7 +392,7 @@ func TestMintOnRevertedDeposit(t *testing.T) { ...@@ -392,7 +392,7 @@ func TestMintOnRevertedDeposit(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l1Client := sys.Clients["l1"] l1Client := sys.Clients["l1"]
l2Verif := sys.Clients["verifier"] l2Verif := sys.Clients["verifier"]
...@@ -475,11 +475,7 @@ func TestMissingBatchE2E(t *testing.T) { ...@@ -475,11 +475,7 @@ func TestMissingBatchE2E(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer func() { defer sys.Close()
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
sys.Close(ctx)
}()
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"] l2Verif := sys.Clients["verifier"]
...@@ -613,7 +609,7 @@ func TestSystemMockP2P(t *testing.T) { ...@@ -613,7 +609,7 @@ func TestSystemMockP2P(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"] l2Verif := sys.Clients["verifier"]
...@@ -701,7 +697,7 @@ func TestSystemRPCAltSync(t *testing.T) { ...@@ -701,7 +697,7 @@ func TestSystemRPCAltSync(t *testing.T) {
}, },
}) })
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"] l2Verif := sys.Clients["verifier"]
...@@ -813,7 +809,7 @@ func TestSystemDenseTopology(t *testing.T) { ...@@ -813,7 +809,7 @@ func TestSystemDenseTopology(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"] l2Verif := sys.Clients["verifier"]
...@@ -878,7 +874,7 @@ func TestL1InfoContract(t *testing.T) { ...@@ -878,7 +874,7 @@ func TestL1InfoContract(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l1Client := sys.Clients["l1"] l1Client := sys.Clients["l1"]
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
...@@ -1007,7 +1003,7 @@ func TestWithdrawals(t *testing.T) { ...@@ -1007,7 +1003,7 @@ func TestWithdrawals(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l1Client := sys.Clients["l1"] l1Client := sys.Clients["l1"]
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
...@@ -1214,7 +1210,7 @@ func TestFees(t *testing.T) { ...@@ -1214,7 +1210,7 @@ func TestFees(t *testing.T) {
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"] l2Verif := sys.Clients["verifier"]
...@@ -1360,7 +1356,7 @@ func TestStopStartSequencer(t *testing.T) { ...@@ -1360,7 +1356,7 @@ func TestStopStartSequencer(t *testing.T) {
cfg := DefaultSystemConfig(t) cfg := DefaultSystemConfig(t)
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
rollupNode := sys.RollupNodes["sequencer"] rollupNode := sys.RollupNodes["sequencer"]
...@@ -1404,7 +1400,7 @@ func TestStopStartBatcher(t *testing.T) { ...@@ -1404,7 +1400,7 @@ func TestStopStartBatcher(t *testing.T) {
cfg := DefaultSystemConfig(t) cfg := DefaultSystemConfig(t)
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
rollupRPCClient, err := rpc.DialContext(context.Background(), sys.RollupNodes["verifier"].HTTPEndpoint()) rollupRPCClient, err := rpc.DialContext(context.Background(), sys.RollupNodes["verifier"].HTTPEndpoint())
require.Nil(t, err) require.Nil(t, err)
......
...@@ -49,7 +49,7 @@ func TestGasPriceOracleFeeUpdates(t *testing.T) { ...@@ -49,7 +49,7 @@ func TestGasPriceOracleFeeUpdates(t *testing.T) {
cfg := DefaultSystemConfig(t) cfg := DefaultSystemConfig(t)
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair. // Obtain our sequencer, verifier, and transactor keypair.
l1Client := sys.Clients["l1"] l1Client := sys.Clients["l1"]
...@@ -136,7 +136,7 @@ func TestL2SequencerRPCDepositTx(t *testing.T) { ...@@ -136,7 +136,7 @@ func TestL2SequencerRPCDepositTx(t *testing.T) {
cfg := DefaultSystemConfig(t) cfg := DefaultSystemConfig(t)
sys, err := cfg.Start() sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair. // Obtain our sequencer, verifier, and transactor keypair.
l2Seq := sys.Clients["sequencer"] l2Seq := sys.Clients["sequencer"]
...@@ -250,7 +250,7 @@ func TestMixedDepositValidity(t *testing.T) { ...@@ -250,7 +250,7 @@ func TestMixedDepositValidity(t *testing.T) {
cfg := DefaultSystemConfig(t) cfg := DefaultSystemConfig(t)
sys, testAccounts, err := startConfigWithTestAccounts(&cfg, accountUsedToDeposit) sys, testAccounts, err := startConfigWithTestAccounts(&cfg, accountUsedToDeposit)
require.Nil(t, err, "Error starting up system") require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair. // Obtain our sequencer, verifier, and transactor keypair.
l1Client := sys.Clients["l1"] l1Client := sys.Clients["l1"]
...@@ -430,7 +430,7 @@ func TestMixedWithdrawalValidity(t *testing.T) { ...@@ -430,7 +430,7 @@ func TestMixedWithdrawalValidity(t *testing.T) {
cfg.DeployConfig.FinalizationPeriodSeconds = 6 cfg.DeployConfig.FinalizationPeriodSeconds = 6
sys, err := cfg.Start() sys, err := cfg.Start()
require.NoError(t, err, "error starting up system") require.NoError(t, err, "error starting up system")
defer sys.Close(context.Background()) defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair. // Obtain our sequencer, verifier, and transactor keypair.
l1Client := sys.Clients["l1"] l1Client := sys.Clients["l1"]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment