ginuerzh
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package conn
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestConnLimitGenerator(t *testing.T) {
|
|
g := NewConnLimitGenerator(5)
|
|
lim := g.Limiter()
|
|
if lim == nil {
|
|
t.Fatal("Limiter() should not be nil for non-zero limit")
|
|
}
|
|
if lim.Limit() != 5 {
|
|
t.Fatalf("expected limit 5, got %d", lim.Limit())
|
|
}
|
|
|
|
// Each Limiter() call creates a new instance.
|
|
lim2 := g.Limiter()
|
|
if lim2 == lim {
|
|
t.Fatal("each Limiter() call should create a new instance")
|
|
}
|
|
}
|
|
|
|
func TestConnLimitGenerator_Zero(t *testing.T) {
|
|
g := NewConnLimitGenerator(0)
|
|
if g == nil {
|
|
t.Fatal("generator should not be nil")
|
|
}
|
|
if lim := g.Limiter(); lim != nil {
|
|
t.Fatal("Limiter() should be nil for zero limit")
|
|
}
|
|
}
|
|
|
|
func TestConnLimitGenerator_Negative(t *testing.T) {
|
|
g := NewConnLimitGenerator(-5)
|
|
if lim := g.Limiter(); lim != nil {
|
|
t.Fatal("Limiter() should be nil for negative limit")
|
|
}
|
|
}
|
|
|
|
func TestConnLimitGenerator_NilReceiver(t *testing.T) {
|
|
var g *connLimitGenerator
|
|
if lim := g.Limiter(); lim != nil {
|
|
t.Fatal("nil receiver Limiter() should return nil")
|
|
}
|
|
}
|
|
|
|
func TestConnLimitSingleGenerator(t *testing.T) {
|
|
g := NewConnLimitSingleGenerator(5)
|
|
lim := g.Limiter()
|
|
if lim == nil {
|
|
t.Fatal("Limiter() should not be nil")
|
|
}
|
|
if lim.Limit() != 5 {
|
|
t.Fatalf("expected limit 5, got %d", lim.Limit())
|
|
}
|
|
|
|
// Same instance every call.
|
|
lim2 := g.Limiter()
|
|
if lim2 != lim {
|
|
t.Fatal("single generator should return same instance")
|
|
}
|
|
}
|
|
|
|
func TestConnLimitSingleGenerator_Zero(t *testing.T) {
|
|
g := NewConnLimitSingleGenerator(0)
|
|
if lim := g.Limiter(); lim != nil {
|
|
t.Fatal("Limiter() should be nil for zero limit")
|
|
}
|
|
}
|
|
|
|
func TestConnLimitSingleGenerator_Negative(t *testing.T) {
|
|
g := NewConnLimitSingleGenerator(-1)
|
|
if lim := g.Limiter(); lim != nil {
|
|
t.Fatal("Limiter() should be nil for negative limit")
|
|
}
|
|
}
|
|
|
|
func TestConnLimitSingleGenerator_NilReceiver(t *testing.T) {
|
|
var g *connLimitSingleGenerator
|
|
if lim := g.Limiter(); lim != nil {
|
|
t.Fatal("nil receiver Limiter() should return nil")
|
|
}
|
|
}
|