status-go/services/wallet/blockchainstate/blockchainstate_test.go

47 lines
1.1 KiB
Go
Raw Normal View History

package blockchainstate
2023-09-04 10:18:46 +00:00
import (
"testing"
"time"
"github.com/stretchr/testify/require"
)
var mockupTime = time.Unix(946724400, 0) // 2000-01-01 12:00:00
func mockupSince(t time.Time) time.Duration {
return mockupTime.Sub(t)
}
func setupTestState(t *testing.T) (s *BlockChainState) {
state := NewBlockChainState()
2023-09-04 10:18:46 +00:00
state.sinceFn = mockupSince
return state
}
func TestEstimateLatestBlockNumber(t *testing.T) {
state := setupTestState(t)
state.setLatestBlockDataForChain(1, LatestBlockData{
blockNumber: uint64(100),
timestamp: mockupTime.Add(-31 * time.Second),
blockDuration: 10 * time.Second,
})
state.setLatestBlockDataForChain(2, LatestBlockData{
blockNumber: uint64(200),
timestamp: mockupTime.Add(-5 * time.Second),
blockDuration: 12 * time.Second,
})
val, ok := state.estimateLatestBlockNumber(1)
require.True(t, ok)
require.Equal(t, uint64(103), val)
val, ok = state.estimateLatestBlockNumber(2)
require.True(t, ok)
require.Equal(t, uint64(200), val)
val, ok = state.estimateLatestBlockNumber(3)
require.False(t, ok)
require.Equal(t, uint64(0), val)
}