71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
|
package middlewares
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"net/url"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/containous/traefik/testhelpers"
|
||
|
|
"github.com/vulcand/oxy/roundrobin"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEmptyBackendHandler(t *testing.T) {
|
||
|
|
tests := []struct {
|
||
|
|
amountServer int
|
||
|
|
wantStatusCode int
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
amountServer: 0,
|
||
|
|
wantStatusCode: http.StatusServiceUnavailable,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
amountServer: 1,
|
||
|
|
wantStatusCode: http.StatusOK,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, test := range tests {
|
||
|
|
test := test
|
||
|
|
|
||
|
|
t.Run(fmt.Sprintf("amount servers %d", test.amountServer), func(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
nextHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(http.StatusOK)
|
||
|
|
})
|
||
|
|
handler := NewEmptyBackendHandler(&healthCheckLoadBalancer{test.amountServer}, nextHandler)
|
||
|
|
|
||
|
|
recorder := httptest.NewRecorder()
|
||
|
|
req := httptest.NewRequest(http.MethodGet, "http://localhost", nil)
|
||
|
|
|
||
|
|
handler.ServeHTTP(recorder, req)
|
||
|
|
|
||
|
|
if recorder.Result().StatusCode != test.wantStatusCode {
|
||
|
|
t.Errorf("Received status code %d, wanted %d", recorder.Result().StatusCode, test.wantStatusCode)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type healthCheckLoadBalancer struct {
|
||
|
|
amountServer int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lb *healthCheckLoadBalancer) RemoveServer(u *url.URL) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lb *healthCheckLoadBalancer) UpsertServer(u *url.URL, options ...roundrobin.ServerOption) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lb *healthCheckLoadBalancer) Servers() []*url.URL {
|
||
|
|
servers := make([]*url.URL, lb.amountServer)
|
||
|
|
for i := 0; i < lb.amountServer; i++ {
|
||
|
|
servers = append(servers, testhelpers.MustParseURL("http://localhost"))
|
||
|
|
}
|
||
|
|
return servers
|
||
|
|
}
|