cache_test.go 3.92 KB
Newer Older
1 2 3 4 5 6
package l1

import (
	"math/rand"
	"testing"

7
	"github.com/ethereum-optimism/optimism/op-program/client/l1/test"
8
	"github.com/ethereum-optimism/optimism/op-service/eth"
Sabnock01's avatar
Sabnock01 committed
9
	"github.com/ethereum-optimism/optimism/op-service/testutils"
10
	"github.com/ethereum/go-ethereum/common"
11
	"github.com/ethereum/go-ethereum/crypto"
12 13 14 15 16 17 18 19
	"github.com/stretchr/testify/require"
)

// Should implement Oracle
var _ Oracle = (*CachingOracle)(nil)

func TestCachingOracle_HeaderByBlockHash(t *testing.T) {
	rng := rand.New(rand.NewSource(1))
20
	stub := test.NewStubOracle(t)
21 22 23 24
	oracle := NewCachingOracle(stub)
	block := testutils.RandomBlockInfo(rng)

	// Initial call retrieves from the stub
25
	stub.Blocks[block.Hash()] = block
26 27 28 29
	result := oracle.HeaderByBlockHash(block.Hash())
	require.Equal(t, block, result)

	// Later calls should retrieve from cache
30
	delete(stub.Blocks, block.Hash())
31 32 33 34 35 36
	result = oracle.HeaderByBlockHash(block.Hash())
	require.Equal(t, block, result)
}

func TestCachingOracle_TransactionsByBlockHash(t *testing.T) {
	rng := rand.New(rand.NewSource(1))
37
	stub := test.NewStubOracle(t)
38 39 40 41
	oracle := NewCachingOracle(stub)
	block, _ := testutils.RandomBlock(rng, 3)

	// Initial call retrieves from the stub
42 43
	stub.Blocks[block.Hash()] = eth.BlockToInfo(block)
	stub.Txs[block.Hash()] = block.Transactions()
44
	actualBlock, actualTxs := oracle.TransactionsByBlockHash(block.Hash())
45
	require.Equal(t, eth.BlockToInfo(block), actualBlock)
46 47 48
	require.Equal(t, block.Transactions(), actualTxs)

	// Later calls should retrieve from cache
49 50
	delete(stub.Blocks, block.Hash())
	delete(stub.Txs, block.Hash())
51
	actualBlock, actualTxs = oracle.TransactionsByBlockHash(block.Hash())
52
	require.Equal(t, eth.BlockToInfo(block), actualBlock)
53 54 55 56 57
	require.Equal(t, block.Transactions(), actualTxs)
}

func TestCachingOracle_ReceiptsByBlockHash(t *testing.T) {
	rng := rand.New(rand.NewSource(1))
58
	stub := test.NewStubOracle(t)
59 60 61 62
	oracle := NewCachingOracle(stub)
	block, rcpts := testutils.RandomBlock(rng, 3)

	// Initial call retrieves from the stub
63 64
	stub.Blocks[block.Hash()] = eth.BlockToInfo(block)
	stub.Rcpts[block.Hash()] = rcpts
65
	actualBlock, actualRcpts := oracle.ReceiptsByBlockHash(block.Hash())
66
	require.Equal(t, eth.BlockToInfo(block), actualBlock)
67 68 69
	require.EqualValues(t, rcpts, actualRcpts)

	// Later calls should retrieve from cache
70 71
	delete(stub.Blocks, block.Hash())
	delete(stub.Rcpts, block.Hash())
72
	actualBlock, actualRcpts = oracle.ReceiptsByBlockHash(block.Hash())
73
	require.Equal(t, eth.BlockToInfo(block), actualBlock)
74 75
	require.EqualValues(t, rcpts, actualRcpts)
}
76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95

func TestCachingOracle_GetBlobs(t *testing.T) {
	stub := test.NewStubOracle(t)
	oracle := NewCachingOracle(stub)

	l1BlockRef := eth.L1BlockRef{Time: 0}
	indexedBlobHash := eth.IndexedBlobHash{Hash: [32]byte{0xFA, 0xCA, 0xDE}, Index: 0}
	blob := eth.Blob{0xFF}

	// Initial call retrieves from the stub
	stub.Blobs[l1BlockRef] = make(map[eth.IndexedBlobHash]*eth.Blob)
	stub.Blobs[l1BlockRef][indexedBlobHash] = &blob
	actualBlob := oracle.GetBlob(l1BlockRef, indexedBlobHash)
	require.Equal(t, &blob, actualBlob)

	// Later calls should retrieve from cache
	delete(stub.Blobs[l1BlockRef], indexedBlobHash)
	actualBlob = oracle.GetBlob(l1BlockRef, indexedBlobHash)
	require.Equal(t, &blob, actualBlob)
}
96

97
func TestCachingOracle_Precompile(t *testing.T) {
98 99 100 101
	stub := test.NewStubOracle(t)
	oracle := NewCachingOracle(stub)

	input := []byte{0x01, 0x02, 0x03, 0x04}
102 103
	output := []byte{0x0a, 0x0b, 0x0c, 0x0d}
	addr := common.Address{0x1}
104 105

	// Initial call retrieves from the stub
106 107 108 109
	stub.PcmpResults[crypto.Keccak256Hash(append(addr.Bytes(), input...))] = output
	actualResult, actualStatus := oracle.Precompile(addr, input)
	require.True(t, actualStatus)
	require.EqualValues(t, output, actualResult)
110 111

	// Later calls should retrieve from cache
112 113 114 115
	delete(stub.PcmpResults, crypto.Keccak256Hash(append(addr.Bytes(), input...)))
	actualResult, actualStatus = oracle.Precompile(addr, input)
	require.True(t, actualStatus)
	require.EqualValues(t, output, actualResult)
116
}