45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
|
package balance
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math/big"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_simpleCacheAll(t *testing.T) {
|
||
|
|
cache := newSimpleCache()
|
||
|
|
|
||
|
|
// init args
|
||
|
|
block := big.NewInt(1)
|
||
|
|
chainID := uint64(1)
|
||
|
|
account := common.Address{1}
|
||
|
|
balance := big.NewInt(1)
|
||
|
|
|
||
|
|
// Test balance
|
||
|
|
cache.AddBalance(account, chainID, block, balance)
|
||
|
|
require.Equal(t, 1, cache.BalanceSize(account, chainID))
|
||
|
|
require.Equal(t, 0, cache.NonceSize(account, chainID))
|
||
|
|
|
||
|
|
balRes := cache.GetBalance(account, chainID, block)
|
||
|
|
require.Equal(t, balance, balRes)
|
||
|
|
|
||
|
|
// Test nonce
|
||
|
|
cache = newSimpleCache()
|
||
|
|
nonce := int64(2)
|
||
|
|
cache.AddNonce(account, chainID, block, &nonce)
|
||
|
|
require.Equal(t, 1, cache.NonceSize(account, chainID))
|
||
|
|
require.Equal(t, 0, cache.BalanceSize(account, chainID))
|
||
|
|
|
||
|
|
nonceRes := cache.GetNonce(account, chainID, block)
|
||
|
|
require.Equal(t, nonce, *nonceRes)
|
||
|
|
|
||
|
|
// Test nonceRange size after adding nonce
|
||
|
|
cache = newSimpleCache()
|
||
|
|
cache.AddNonce(account, chainID, block, &nonce)
|
||
|
|
require.Equal(t, 1, cache.nonceRangeCache.size(account, chainID))
|
||
|
|
require.Equal(t, 1, len(cache.nonceRangeCache.sortedRanges))
|
||
|
|
}
|