mirror of
https://github.com/tendermint/tendermint.git
synced 2026-02-10 05:50:19 +00:00
8a991e288c
This continues the push of plumbing contexts through tendermint. I attempted to find all goroutines in the production code (non-test) and made sure that these threads would exit when their contexts were canceled, and I believe this PR does that.
60 lines
1.8 KiB
Go
60 lines
1.8 KiB
Go
package p2p
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
"github.com/tendermint/tendermint/types"
|
|
)
|
|
|
|
func TestRouter_ConstructQueueFactory(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
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(ctx, log.NewNopLogger(), nil, types.NodeInfo{}, nil, 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(ctx, log.NewNopLogger(), nil, types.NodeInfo{}, nil, 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(ctx, log.NewNopLogger(), nil, types.NodeInfo{}, nil, nil, nil, nil, opts)
|
|
require.NoError(t, err)
|
|
q, ok := r.queueFactory(1).(*pqScheduler)
|
|
require.True(t, ok)
|
|
defer q.close()
|
|
})
|
|
t.Run("NonExistant", func(t *testing.T) {
|
|
opts := RouterOptions{QueueType: "fast"}
|
|
_, err := NewRouter(ctx, log.NewNopLogger(), nil, types.NodeInfo{}, nil, 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(ctx)
|
|
require.Error(t, err)
|
|
require.Nil(t, fn)
|
|
})
|
|
}
|