Test batcher restart in enclave

Author: QuentinI

espresso/devnet-tests/batcher_restart_test.go

@@ -6,9 +6,20 @@ import (
 
 	"github.com/ethereum/go-ethereum"
 	"github.com/stretchr/testify/require"
+
+	env "github.com/ethereum-optimism/optimism/espresso/environment"
 )
 
 func TestBatcherRestart(t *testing.T) {
+	testRestart(t, false)
+}
+
+func TestEnclaveRestart(t *testing.T) {
+	env.RunOnlyWithEnclave(t)
+	testRestart(t, true)
+}
+
+func testRestart(t *testing.T, tee bool) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
@@ -18,6 +29,8 @@ func TestBatcherRestart(t *testing.T) {
 		require.NoError(t, d.Down())
 	}()
 
+	d.tee = tee
+
 	// Send a transaction just to check that everything has started up ok.
 	require.NoError(t, d.RunSimpleL2Burn())
 

espresso/devnet-tests/devnet_tools.go

@@ -33,6 +33,7 @@ import (
 
 type Devnet struct {
 	ctx         context.Context
+	tee         bool
 	secrets     secrets.Secrets
 	outageTime  time.Duration
 	successTime time.Duration
@@ -81,9 +82,16 @@ func NewDevnet(ctx context.Context, t *testing.T) *Devnet {
 }
 
 func (d *Devnet) Up() (err error) {
+	var profile string
+	if d.tee {
+		profile = "tee"
+	} else {
+		profile = "default"
+	}
+
 	cmd := exec.CommandContext(
 		d.ctx,
-		"docker", "compose", "up", "-d",
+		"docker", "compose", "--profile", profile, "up", "-d",
 	)
 	cmd.Env = append(
 		cmd.Env,