You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.9 KiB
63 lines
1.9 KiB
package p2p
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
"github.com/tendermint/tendermint/types"
|
|
)
|
|
|
|
func TestRouter_ConstructQueueFactory(t *testing.T) {
|
|
t.Run("ValidateOptionsPopulatesDefaultQueue", func(t *testing.T) {
|
|
opts := RouterOptions{}
|
|
require.NoError(t, opts.Validate())
|
|
require.Equal(t, "fifo", opts.QueueType)
|
|
})
|
|
t.Run("Default", func(t *testing.T) {
|
|
require.Zero(t, os.Getenv("TM_P2P_QUEUE"))
|
|
opts := RouterOptions{}
|
|
r, err := NewRouter(log.NewNopLogger(), nil, types.NodeInfo{}, nil, nil, nil, opts)
|
|
require.NoError(t, err)
|
|
_, ok := r.queueFactory(1).(*fifoQueue)
|
|
require.True(t, ok)
|
|
})
|
|
t.Run("Fifo", func(t *testing.T) {
|
|
opts := RouterOptions{QueueType: queueTypeFifo}
|
|
r, err := NewRouter(log.NewNopLogger(), nil, types.NodeInfo{}, nil, nil, nil, opts)
|
|
require.NoError(t, err)
|
|
_, ok := r.queueFactory(1).(*fifoQueue)
|
|
require.True(t, ok)
|
|
})
|
|
t.Run("Priority", func(t *testing.T) {
|
|
opts := RouterOptions{QueueType: queueTypePriority}
|
|
r, err := NewRouter(log.NewNopLogger(), nil, types.NodeInfo{}, nil, nil, nil, opts)
|
|
require.NoError(t, err)
|
|
q, ok := r.queueFactory(1).(*pqScheduler)
|
|
require.True(t, ok)
|
|
defer q.close()
|
|
})
|
|
t.Run("WDRR", func(t *testing.T) {
|
|
opts := RouterOptions{QueueType: queueTypeWDRR}
|
|
r, err := NewRouter(log.NewNopLogger(), nil, types.NodeInfo{}, nil, nil, nil, opts)
|
|
require.NoError(t, err)
|
|
q, ok := r.queueFactory(1).(*wdrrScheduler)
|
|
require.True(t, ok)
|
|
defer q.close()
|
|
})
|
|
t.Run("NonExistant", func(t *testing.T) {
|
|
opts := RouterOptions{QueueType: "fast"}
|
|
_, err := NewRouter(log.NewNopLogger(), nil, types.NodeInfo{}, nil, nil, nil, opts)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), "fast")
|
|
})
|
|
t.Run("InternalsSafeWhenUnspecified", func(t *testing.T) {
|
|
r := &Router{}
|
|
require.Zero(t, r.options.QueueType)
|
|
|
|
fn, err := r.createQueueFactory()
|
|
require.Error(t, err)
|
|
require.Nil(t, fn)
|
|
})
|
|
}
|