package api import ( "encoding/json" "fmt" "io/ioutil" "net/http" "net/http/httptest" "strconv" "testing" "github.com/containous/traefik/v2/pkg/config/runtime" "github.com/containous/traefik/v2/pkg/config/static" "github.com/gorilla/mux" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestHandler_EntryPoints(t *testing.T) { type expected struct { statusCode int nextPage string jsonFile string } testCases := []struct { desc string path string conf static.Configuration expected expected }{ { desc: "all entry points, but no config", path: "/api/entrypoints", conf: static.Configuration{API: &static.API{}, Global: &static.Global{}}, expected: expected{ statusCode: http.StatusOK, nextPage: "1", jsonFile: "testdata/entrypoints-empty.json", }, }, { desc: "all entry points", path: "/api/entrypoints", conf: static.Configuration{ Global: &static.Global{}, API: &static.API{}, EntryPoints: map[string]*static.EntryPoint{ "web": { Address: ":80", Transport: &static.EntryPointsTransport{ LifeCycle: &static.LifeCycle{ RequestAcceptGraceTimeout: 1, GraceTimeOut: 2, }, RespondingTimeouts: &static.RespondingTimeouts{ ReadTimeout: 3, WriteTimeout: 4, IdleTimeout: 5, }, }, ProxyProtocol: &static.ProxyProtocol{ Insecure: true, TrustedIPs: []string{"192.168.1.1", "192.168.1.2"}, }, ForwardedHeaders: &static.ForwardedHeaders{ Insecure: true, TrustedIPs: []string{"192.168.1.3", "192.168.1.4"}, }, }, "web-secure": { Address: ":443", Transport: &static.EntryPointsTransport{ LifeCycle: &static.LifeCycle{ RequestAcceptGraceTimeout: 10, GraceTimeOut: 20, }, RespondingTimeouts: &static.RespondingTimeouts{ ReadTimeout: 30, WriteTimeout: 40, IdleTimeout: 50, }, }, ProxyProtocol: &static.ProxyProtocol{ Insecure: true, TrustedIPs: []string{"192.168.1.10", "192.168.1.20"}, }, ForwardedHeaders: &static.ForwardedHeaders{ Insecure: true, TrustedIPs: []string{"192.168.1.30", "192.168.1.40"}, }, }, }, }, expected: expected{ statusCode: http.StatusOK, nextPage: "1", jsonFile: "testdata/entrypoints.json", }, }, { desc: "all entry points, pagination, 1 res per page, want page 2", path: "/api/entrypoints?page=2&per_page=1", conf: static.Configuration{ Global: &static.Global{}, API: &static.API{}, EntryPoints: map[string]*static.EntryPoint{ "web1": {Address: ":81"}, "web2": {Address: ":82"}, "web3": {Address: ":83"}, }, }, expected: expected{ statusCode: http.StatusOK, nextPage: "3", jsonFile: "testdata/entrypoints-page2.json", }, }, { desc: "all entry points, pagination, 19 results overall, 7 res per page, want page 3", path: "/api/entrypoints?page=3&per_page=7", conf: static.Configuration{ Global: &static.Global{}, API: &static.API{}, EntryPoints: generateEntryPoints(19), }, expected: expected{ statusCode: http.StatusOK, nextPage: "1", jsonFile: "testdata/entrypoints-many-lastpage.json", }, }, { desc: "all entry points, pagination, 5 results overall, 10 res per page, want page 2", path: "/api/entrypoints?page=2&per_page=10", conf: static.Configuration{ Global: &static.Global{}, API: &static.API{}, EntryPoints: generateEntryPoints(5), }, expected: expected{ statusCode: http.StatusBadRequest, }, }, { desc: "all entry points, pagination, 10 results overall, 10 res per page, want page 2", path: "/api/entrypoints?page=2&per_page=10", conf: static.Configuration{ Global: &static.Global{}, API: &static.API{}, EntryPoints: generateEntryPoints(10), }, expected: expected{ statusCode: http.StatusBadRequest, }, }, { desc: "one entry point by id", path: "/api/entrypoints/bar", conf: static.Configuration{ Global: &static.Global{}, API: &static.API{}, EntryPoints: map[string]*static.EntryPoint{ "bar": {Address: ":81"}, }, }, expected: expected{ statusCode: http.StatusOK, jsonFile: "testdata/entrypoint-bar.json", }, }, { desc: "one entry point by id, that does not exist", path: "/api/entrypoints/foo", conf: static.Configuration{ Global: &static.Global{}, API: &static.API{}, EntryPoints: map[string]*static.EntryPoint{ "bar": {Address: ":81"}, }, }, expected: expected{ statusCode: http.StatusNotFound, }, }, { desc: "one entry point by id, but no config", path: "/api/entrypoints/foo", conf: static.Configuration{API: &static.API{}, Global: &static.Global{}}, expected: expected{ statusCode: http.StatusNotFound, }, }, } for _, test := range testCases { test := test t.Run(test.desc, func(t *testing.T) { t.Parallel() handler := New(test.conf, &runtime.Configuration{}) router := mux.NewRouter() handler.Append(router) server := httptest.NewServer(router) resp, err := http.DefaultClient.Get(server.URL + test.path) require.NoError(t, err) require.Equal(t, test.expected.statusCode, resp.StatusCode) assert.Equal(t, test.expected.nextPage, resp.Header.Get(nextPageHeader)) if test.expected.jsonFile == "" { return } assert.Equal(t, resp.Header.Get("Content-Type"), "application/json") contents, err := ioutil.ReadAll(resp.Body) require.NoError(t, err) err = resp.Body.Close() require.NoError(t, err) if *updateExpected { var results interface{} err := json.Unmarshal(contents, &results) require.NoError(t, err) newJSON, err := json.MarshalIndent(results, "", "\t") require.NoError(t, err) err = ioutil.WriteFile(test.expected.jsonFile, newJSON, 0644) require.NoError(t, err) } data, err := ioutil.ReadFile(test.expected.jsonFile) require.NoError(t, err) assert.JSONEq(t, string(data), string(contents)) }) } } func generateEntryPoints(nb int) map[string]*static.EntryPoint { eps := make(map[string]*static.EntryPoint, nb) for i := 0; i < nb; i++ { eps[fmt.Sprintf("ep%2d", i)] = &static.EntryPoint{ Address: ":" + strconv.Itoa(i), } } return eps }