80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
|
package mirror
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"sync/atomic"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/containous/traefik/v2/pkg/safe"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMirroringOn100(t *testing.T) {
|
||
|
|
var countMirror1, countMirror2 int32
|
||
|
|
handler := http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
rw.WriteHeader(http.StatusOK)
|
||
|
|
})
|
||
|
|
pool := safe.NewPool(context.Background())
|
||
|
|
mirror := New(handler, pool)
|
||
|
|
err := mirror.AddMirror(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
atomic.AddInt32(&countMirror1, 1)
|
||
|
|
}), 10)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
err = mirror.AddMirror(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
atomic.AddInt32(&countMirror2, 1)
|
||
|
|
}), 50)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
for i := 0; i < 100; i++ {
|
||
|
|
mirror.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest(http.MethodGet, "/", nil))
|
||
|
|
}
|
||
|
|
|
||
|
|
pool.Stop()
|
||
|
|
|
||
|
|
val1 := atomic.LoadInt32(&countMirror1)
|
||
|
|
val2 := atomic.LoadInt32(&countMirror2)
|
||
|
|
assert.Equal(t, 10, int(val1))
|
||
|
|
assert.Equal(t, 50, int(val2))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMirroringOn10(t *testing.T) {
|
||
|
|
var countMirror1, countMirror2 int32
|
||
|
|
handler := http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
rw.WriteHeader(http.StatusOK)
|
||
|
|
})
|
||
|
|
pool := safe.NewPool(context.Background())
|
||
|
|
mirror := New(handler, pool)
|
||
|
|
err := mirror.AddMirror(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
atomic.AddInt32(&countMirror1, 1)
|
||
|
|
}), 10)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
err = mirror.AddMirror(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
atomic.AddInt32(&countMirror2, 1)
|
||
|
|
}), 50)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
mirror.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest(http.MethodGet, "/", nil))
|
||
|
|
}
|
||
|
|
|
||
|
|
pool.Stop()
|
||
|
|
|
||
|
|
val1 := atomic.LoadInt32(&countMirror1)
|
||
|
|
val2 := atomic.LoadInt32(&countMirror2)
|
||
|
|
assert.Equal(t, 1, int(val1))
|
||
|
|
assert.Equal(t, 5, int(val2))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInvalidPercent(t *testing.T) {
|
||
|
|
mirror := New(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {}), safe.NewPool(context.Background()))
|
||
|
|
err := mirror.AddMirror(nil, -1)
|
||
|
|
assert.Error(t, err)
|
||
|
|
|
||
|
|
err = mirror.AddMirror(nil, 101)
|
||
|
|
assert.Error(t, err)
|
||
|
|
}
|