test: ensure correct tx removal

test/dummy_test.go

@@ -1,7 +1,11 @@
 package test
 
 import (
+	"context"
+	"github.com/rollkit/go-execution/types"
+	"github.com/stretchr/testify/require"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/suite"
 )
@@ -17,3 +21,37 @@ func (s *DummyTestSuite) SetupTest() {
 func TestDummySuite(t *testing.T) {
 	suite.Run(t, new(DummyTestSuite))
 }
+
+func TestTxRemoval(t *testing.T) {
+	exec := NewDummyExecutor()
+	tx1 := types.Tx([]byte{1, 2, 3})
+	tx2 := types.Tx([]byte{3, 2, 1})
+
+	exec.InjectTx(tx1)
+	exec.InjectTx(tx2)
+
+	// first execution of GetTxs - nothing special
+	txs, err := exec.GetTxs(context.Background())
+	require.NoError(t, err)
+	require.Len(t, txs, 2)
+	require.Contains(t, txs, tx1)
+	require.Contains(t, txs, tx2)
+
+	// ExecuteTxs was not called, so 2 txs should still be returned
+	txs, err = exec.GetTxs(context.Background())
+	require.NoError(t, err)
+	require.Len(t, txs, 2)
+	require.Contains(t, txs, tx1)
+	require.Contains(t, txs, tx2)
+
+	state, _, err := exec.ExecuteTxs(context.Background(), []types.Tx{tx1}, 1, time.Now(), nil)
+	require.NoError(t, err)
+	require.NotEmpty(t, state)
+
+	// ExecuteTxs was called, 1 tx remaining in mempool
+	txs, err = exec.GetTxs(context.Background())
+	require.NoError(t, err)
+	require.Len(t, txs, 1)
+	require.NotContains(t, txs, tx1)
+	require.Contains(t, txs, tx2)
+}