mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-06-21 14:14:30 +00:00
55 lines
1.7 KiB
Go
55 lines
1.7 KiB
Go
package utils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPoolAdd(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
pool := NewPool(2) // 2 is the cert threshold
|
|
timeout1 := Timeout{Round: 1, Signature: []byte{1}}
|
|
timeout2 := Timeout{Round: 1, Signature: []byte{2}}
|
|
timeout3 := Timeout{Round: 1, Signature: []byte{3}}
|
|
thresholdReached, numOfItems, pooledTimeouts := pool.Add(&timeout1)
|
|
assert.NotNil(pooledTimeouts)
|
|
assert.Equal(1, numOfItems)
|
|
assert.False(thresholdReached)
|
|
thresholdReached, numOfItems, pooledTimeouts = pool.Add(&timeout1)
|
|
assert.NotNil(pooledTimeouts)
|
|
assert.False(thresholdReached)
|
|
// Duplicates should not be added
|
|
assert.Equal(1, numOfItems)
|
|
|
|
// Should add the one that is not a duplicates
|
|
thresholdReached, numOfItems, pooledTimeouts = pool.Add(&timeout2)
|
|
assert.True(thresholdReached)
|
|
assert.NotNil(pooledTimeouts)
|
|
assert.Equal(2, numOfItems)
|
|
|
|
// Try to add one more to the same round, but that round threshold has already been reached, hence deleted
|
|
thresholdReached, numOfItems, pooledTimeouts = pool.Add(&timeout3)
|
|
assert.False(thresholdReached)
|
|
assert.NotNil(pooledTimeouts)
|
|
assert.Equal(1, numOfItems)
|
|
|
|
pool = NewPool(3) // 3 is the cert size
|
|
thresholdReached, numOfItems, pooledTimeouts = pool.Add(&timeout1)
|
|
assert.False(thresholdReached)
|
|
assert.NotNil(pooledTimeouts)
|
|
assert.Equal(1, numOfItems)
|
|
|
|
thresholdReached, numOfItems, pooledTimeouts = pool.Add(&timeout2)
|
|
assert.False(thresholdReached)
|
|
assert.Equal(2, numOfItems)
|
|
assert.NotNil(pooledTimeouts)
|
|
pool.Clear()
|
|
|
|
// Pool has been cleared. Start from 0 again
|
|
thresholdReached, numOfItems, pooledTimeouts = pool.Add(&timeout3)
|
|
assert.False(thresholdReached)
|
|
assert.Equal(1, numOfItems)
|
|
assert.NotNil(pooledTimeouts)
|
|
}
|