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

Simplify e2e test Close

parent 99c75b47
package op_e2e
import (
"context"
"math"
"math/big"
"testing"
......@@ -30,7 +29,7 @@ func TestERC20BridgeDeposits(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
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)
......
......@@ -214,11 +214,13 @@ type System struct {
Mocknet mocknet.Mocknet
}
func (sys *System) Close(ctx context.Context) {
func (sys *System) Close() {
if sys.L2OutputSubmitter != nil {
sys.L2OutputSubmitter.Stop()
}
if sys.BatchSubmitter != nil {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
sys.BatchSubmitter.StopIfRunning(ctx)
}
......
......@@ -73,7 +73,7 @@ func TestL2OutputSubmitter(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l1Client := sys.Clients["l1"]
......@@ -146,7 +146,7 @@ func TestSystemE2E(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
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)
......@@ -261,7 +261,7 @@ func TestConfirmationDepth(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
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)
......@@ -320,7 +320,7 @@ func TestPendingGasLimit(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
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)
......@@ -365,7 +365,7 @@ func TestFinalize(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l2Seq := sys.Clients["sequencer"]
......@@ -392,7 +392,7 @@ func TestMintOnRevertedDeposit(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l1Client := sys.Clients["l1"]
l2Verif := sys.Clients["verifier"]
......@@ -475,11 +475,7 @@ func TestMissingBatchE2E(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer func() {
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
sys.Close(ctx)
}()
defer sys.Close()
l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"]
......@@ -613,7 +609,7 @@ func TestSystemMockP2P(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"]
......@@ -701,7 +697,7 @@ func TestSystemRPCAltSync(t *testing.T) {
},
})
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"]
......@@ -813,7 +809,7 @@ func TestSystemDenseTopology(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"]
......@@ -878,7 +874,7 @@ func TestL1InfoContract(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l1Client := sys.Clients["l1"]
l2Seq := sys.Clients["sequencer"]
......@@ -1007,7 +1003,7 @@ func TestWithdrawals(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l1Client := sys.Clients["l1"]
l2Seq := sys.Clients["sequencer"]
......@@ -1214,7 +1210,7 @@ func TestFees(t *testing.T) {
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l2Seq := sys.Clients["sequencer"]
l2Verif := sys.Clients["verifier"]
......@@ -1360,7 +1356,7 @@ func TestStopStartSequencer(t *testing.T) {
cfg := DefaultSystemConfig(t)
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
l2Seq := sys.Clients["sequencer"]
rollupNode := sys.RollupNodes["sequencer"]
......@@ -1404,7 +1400,7 @@ func TestStopStartBatcher(t *testing.T) {
cfg := DefaultSystemConfig(t)
sys, err := cfg.Start()
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())
require.Nil(t, err)
......
......@@ -49,7 +49,7 @@ func TestGasPriceOracleFeeUpdates(t *testing.T) {
cfg := DefaultSystemConfig(t)
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair.
l1Client := sys.Clients["l1"]
......@@ -136,7 +136,7 @@ func TestL2SequencerRPCDepositTx(t *testing.T) {
cfg := DefaultSystemConfig(t)
sys, err := cfg.Start()
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair.
l2Seq := sys.Clients["sequencer"]
......@@ -250,7 +250,7 @@ func TestMixedDepositValidity(t *testing.T) {
cfg := DefaultSystemConfig(t)
sys, testAccounts, err := startConfigWithTestAccounts(&cfg, accountUsedToDeposit)
require.Nil(t, err, "Error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair.
l1Client := sys.Clients["l1"]
......@@ -430,7 +430,7 @@ func TestMixedWithdrawalValidity(t *testing.T) {
cfg.DeployConfig.FinalizationPeriodSeconds = 6
sys, err := cfg.Start()
require.NoError(t, err, "error starting up system")
defer sys.Close(context.Background())
defer sys.Close()
// Obtain our sequencer, verifier, and transactor keypair.
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