43 lines
1.6 KiB
Go
43 lines
1.6 KiB
Go
package httpserver
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"git.yuntuo.hk/k2safe/orbit/platform/events"
|
|
"git.yuntuo.hk/k2safe/orbit/pulse/worker"
|
|
)
|
|
|
|
func TestHubOptionsAreBoundedAndApplied(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
workerOptions := resolveHubOptions([]HubOptions{{ConnectionLimit: 7, DispatchWorkers: 3, DispatchCapacity: 11}})
|
|
if workerOptions.ConnectionLimit != 7 || workerOptions.DispatchWorkers != 3 || workerOptions.DispatchCapacity != 11 {
|
|
t.Fatalf("worker hub options not resolved: %+v", workerOptions)
|
|
}
|
|
workerHub := NewWorkerHub(ctx, worker.New(nil), workerOptions)
|
|
if cap(workerHub.slots) != 7 {
|
|
t.Fatalf("worker connection limit not applied: %d", cap(workerHub.slots))
|
|
}
|
|
eventOptions := resolveHubOptions([]HubOptions{{ConnectionLimit: 5, DispatchWorkers: 2, DispatchCapacity: 9}})
|
|
eventHub := NewEventHub(ctx, events.New(nil), eventOptions)
|
|
if cap(eventHub.slots) != 5 {
|
|
t.Fatalf("event connection limit not applied: %d", cap(eventHub.slots))
|
|
}
|
|
workerHub.Shutdown(context.Background())
|
|
eventHub.Shutdown(context.Background())
|
|
}
|
|
|
|
func TestHubOptionsRejectInvalidValues(t *testing.T) {
|
|
for _, options := range []HubOptions{{ConnectionLimit: -1}, {ConnectionLimit: maxHubConnectionLimit + 1}, {DispatchWorkers: -1}, {DispatchWorkers: maxHubDispatchWorkers + 1}, {DispatchCapacity: -1}, {DispatchCapacity: maxHubDispatchCapacity + 1}} {
|
|
t.Run("invalid", func(t *testing.T) {
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("invalid hub options did not panic")
|
|
}
|
|
}()
|
|
resolveHubOptions([]HubOptions{options})
|
|
})
|
|
}
|
|
}
|