WIP

Author: QuentinI

espresso/devnet-tests/batcher_restart_test.go

@@ -6,13 +6,25 @@ 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()
 
 	d := NewDevnet(ctx, t)
+	d.tee = tee
 	require.NoError(t, d.Up())
 	defer func() {
 		require.NoError(t, d.Down())

espresso/devnet-tests/devnet_tools.go

@@ -27,6 +27,7 @@ import (
 
 type Devnet struct {
 	ctx         context.Context
+	tee         bool
 	secrets     secrets.Secrets
 	outageTime  time.Duration
 	successTime time.Duration
@@ -66,9 +67,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",
 	)
 	buf := new(bytes.Buffer)
 	cmd.Stderr = buf