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

@@ -34,6 +34,7 @@ import (
 
 type Devnet struct {
 	ctx           context.Context
+	tee           bool
 	secrets       secrets.Secrets
 	outageTime    time.Duration
 	successTime   time.Duration
@@ -106,10 +107,15 @@ func (d *Devnet) Up() (err error) {
 		// up any existing state.
 		return fmt.Errorf("devnet is already running, this should be a clean state; please shut it down first")
 	}
-
+	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,