Dynamic Configuration Refactoring

server/router/route_appender_aggregator_test.go

@@ -0,0 +1,116 @@
+package router
+
+import (
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/containous/alice"
+	"github.com/containous/mux"
+	"github.com/containous/traefik/config/static"
+	"github.com/containous/traefik/ping"
+	"github.com/stretchr/testify/assert"
+)
+
+type ChainBuilderMock struct {
+	middles map[string]alice.Constructor
+}
+
+func (c *ChainBuilderMock) BuildChain(ctx context.Context, middles []string) (*alice.Chain, error) {
+	chain := alice.New()
+
+	for _, mName := range middles {
+		if constructor, ok := c.middles[mName]; ok {
+			chain = chain.Append(constructor)
+		}
+	}
+
+	return &chain, nil
+}
+
+func TestNewRouteAppenderAggregator(t *testing.T) {
+	testCases := []struct {
+		desc       string
+		staticConf static.Configuration
+		middles    map[string]alice.Constructor
+		expected   map[string]int
+	}{
+		{
+			desc: "API with auth, ping without auth",
+			staticConf: static.Configuration{
+				API: &static.API{
+					EntryPoint:  "traefik",
+					Middlewares: []string{"dumb"},
+				},
+				Ping: &ping.Handler{
+					EntryPoint: "traefik",
+				},
+				EntryPoints: &static.EntryPoints{
+					EntryPointList: map[string]static.EntryPoint{
+						"traefik": {},
+					},
+				},
+			},
+			middles: map[string]alice.Constructor{
+				"dumb": func(_ http.Handler) (http.Handler, error) {
+					return http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
+						w.WriteHeader(http.StatusUnauthorized)
+					}), nil
+				},
+			},
+			expected: map[string]int{
+				"/wrong": http.StatusBadGateway,
+				"/ping":  http.StatusOK,
+				//"/.well-known/acme-challenge/token": http.StatusNotFound, // FIXME
+				"/api/providers": http.StatusUnauthorized,
+			},
+		},
+		{
+			desc: "Wrong entrypoint name",
+			staticConf: static.Configuration{
+				API: &static.API{
+					EntryPoint: "no",
+				},
+				EntryPoints: &static.EntryPoints{
+					EntryPointList: map[string]static.EntryPoint{
+						"traefik": {},
+					},
+				},
+			},
+			expected: map[string]int{
+				"/api/providers": http.StatusBadGateway,
+			},
+		},
+	}
+
+	for _, test := range testCases {
+		test := test
+		t.Run(test.desc, func(t *testing.T) {
+			t.Parallel()
+
+			chainBuilder := &ChainBuilderMock{middles: test.middles}
+
+			ctx := context.Background()
+
+			router := NewRouteAppenderAggregator(ctx, chainBuilder, test.staticConf, "traefik", nil)
+
+			internalMuxRouter := mux.NewRouter()
+			router.Append(internalMuxRouter)
+
+			internalMuxRouter.NotFoundHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				w.WriteHeader(http.StatusBadGateway)
+			})
+
+			actual := make(map[string]int)
+			for calledURL := range test.expected {
+				recorder := httptest.NewRecorder()
+				request := httptest.NewRequest(http.MethodGet, calledURL, nil)
+				internalMuxRouter.ServeHTTP(recorder, request)
+				actual[calledURL] = recorder.Code
+			}
+
+			assert.Equal(t, test.expected, actual)
+		})
+	}
+}