optimism/op-devstack/stack/context_test.go at develop · binarycryptodev/optimism · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package stack

import (
	"context"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/ethereum/go-ethereum/log"

	"github.com/ethereum-optimism/optimism/op-service/eth"
	"github.com/ethereum-optimism/optimism/op-service/tri"
)

func TestContext(t *testing.T) {
	ctx := context.Background()
	chainA := eth.ChainIDFromUInt64(900)
	chainB := eth.ChainIDFromUInt64(901)
	t.Run("chainID", func(t *testing.T) {
		require.Equal(t, eth.ChainID{}, ChainIDFromContext(ctx), "none")
		require.Equal(t, chainA, ChainIDFromContext(ContextWithChainID(ctx, chainA)), "lookup")
		require.Equal(t, chainB, ChainIDFromContext(ContextWithChainID(ContextWithChainID(ctx, chainA), chainB)), "priority")
	})
	t.Run("kind", func(t *testing.T) {
		require.Equal(t, Kind(""), KindFromContext(ctx), "none")
		require.Equal(t, L2BatcherKind, KindFromContext(ContextWithKind(ctx, L2BatcherKind)), "lookup")
		require.Equal(t, L2ProposerKind, KindFromContext(ContextWithKind(ContextWithKind(ctx, L2BatcherKind), L2ProposerKind)), "priority")
	})
	t.Run("id", func(t *testing.T) {
		require.Equal(t, L2BatcherID{}, IDFromContext[L2BatcherID](ctx), "none")
		require.Equal(t, SuperchainID(""), IDFromContext[SuperchainID](ctx), "none")
		id1 := L2BatcherID{
			key:     "batcherA",
			chainID: chainA,
		}
		ctx1 := ContextWithID(ctx, id1)
		require.Equal(t, L2BatcherKind, KindFromContext(ctx1), "lookup kind")
		require.Equal(t, chainA, ChainIDFromContext(ctx1), "lookup chainID")
		require.Equal(t, id1, IDFromContext[L2BatcherID](ctx1), "lookup ID")
		// now overlay another different kind of ID on top
		id2 := SuperchainID("foobar")
		ctx2 := ContextWithID(ctx1, id2)
		require.Equal(t, SuperchainKind, KindFromContext(ctx2), "lookup kind")
		require.Equal(t, chainA, ChainIDFromContext(ctx2), "chainID still preserved")
		require.Equal(t, id2, IDFromContext[SuperchainID](ctx2), "lookup ID")
		require.Equal(t, L2BatcherID{}, IDFromContext[L2BatcherID](ctx2), "batcher ID not available")
	})
}

func TestLogFilter(t *testing.T) {
	ctx := context.Background()
	chainA := eth.ChainIDFromUInt64(900)
	chainB := eth.ChainIDFromUInt64(901)
	t.Run("chainID", func(t *testing.T) {
		fn := ChainIDSelector(chainA).Mute()
		require.Equal(t, tri.Undefined, fn(ctx, log.LevelDebug), "regular context should be false")
		require.Equal(t, tri.False, fn(ContextWithChainID(ctx, chainA), log.LevelDebug), "detected chain should be muted")
		require.Equal(t, tri.Undefined, fn(ContextWithChainID(ctx, chainB), log.LevelDebug), "different chain should be shown")
	})
	t.Run("kind", func(t *testing.T) {
		fn := KindSelector(L2BatcherKind).Mute()
		require.Equal(t, tri.Undefined, fn(ctx, log.LevelDebug), "regular context should be false")
		require.Equal(t, tri.False, fn(ContextWithKind(ctx, L2BatcherKind), log.LevelDebug), "detected kind should be muted")
		require.Equal(t, tri.Undefined, fn(ContextWithKind(ctx, L2ProposerKind), log.LevelDebug), "different kind should be shown")
	})
	t.Run("id", func(t *testing.T) {
		id1 := L2BatcherID{
			key:     "batcherA",
			chainID: chainA,
		}
		fn := IDSelector(id1).Mute()
		require.Equal(t, tri.Undefined, fn(ctx, log.LevelDebug), "regular context should be false")
		require.Equal(t, tri.False, fn(ContextWithID(ctx, id1), log.LevelDebug), "detected id should be muted")
		id2 := SuperchainID("foobar")
		require.Equal(t, tri.Undefined, fn(ContextWithID(ctx, id2), log.LevelDebug), "different id should be shown")
	})
}