Add support dump API endpoint
This commit is contained in:
parent
d953ee69b4
commit
e85d02c530
4 changed files with 272 additions and 29 deletions
144
pkg/api/handler_support_dump_test.go
Normal file
144
pkg/api/handler_support_dump_test.go
Normal file
|
@ -0,0 +1,144 @@
|
|||
package api
|
||||
|
||||
import (
|
||||
"archive/tar"
|
||||
"compress/gzip"
|
||||
"errors"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/traefik/traefik/v3/pkg/config/dynamic"
|
||||
"github.com/traefik/traefik/v3/pkg/config/runtime"
|
||||
"github.com/traefik/traefik/v3/pkg/config/static"
|
||||
)
|
||||
|
||||
func TestHandler_SupportDump(t *testing.T) {
|
||||
testCases := []struct {
|
||||
desc string
|
||||
path string
|
||||
confStatic static.Configuration
|
||||
confDyn runtime.Configuration
|
||||
validate func(t *testing.T, files map[string][]byte)
|
||||
}{
|
||||
{
|
||||
desc: "empty configurations",
|
||||
path: "/api/support-dump",
|
||||
confStatic: static.Configuration{API: &static.API{}, Global: &static.Global{}},
|
||||
confDyn: runtime.Configuration{},
|
||||
validate: func(t *testing.T, files map[string][]byte) {
|
||||
t.Helper()
|
||||
|
||||
require.Contains(t, files, "static-config.json")
|
||||
require.Contains(t, files, "runtime-config.json")
|
||||
require.Contains(t, files, "version.json")
|
||||
|
||||
// Verify version.json contains version information
|
||||
assert.Contains(t, string(files["version.json"]), `"version":"dev"`)
|
||||
|
||||
assert.JSONEq(t, `{"global":{},"api":{}}`, string(files["static-config.json"]))
|
||||
assert.Equal(t, `{}`, string(files["runtime-config.json"]))
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "with configuration data",
|
||||
path: "/api/support-dump",
|
||||
confStatic: static.Configuration{
|
||||
API: &static.API{},
|
||||
Global: &static.Global{},
|
||||
EntryPoints: map[string]*static.EntryPoint{
|
||||
"web": {Address: ":80"},
|
||||
},
|
||||
},
|
||||
confDyn: runtime.Configuration{
|
||||
Services: map[string]*runtime.ServiceInfo{
|
||||
"test-service": {
|
||||
Service: &dynamic.Service{
|
||||
LoadBalancer: &dynamic.ServersLoadBalancer{
|
||||
Servers: []dynamic.Server{{URL: "http://127.0.0.1:8080"}},
|
||||
},
|
||||
},
|
||||
Status: runtime.StatusEnabled,
|
||||
},
|
||||
},
|
||||
},
|
||||
validate: func(t *testing.T, files map[string][]byte) {
|
||||
t.Helper()
|
||||
|
||||
require.Contains(t, files, "static-config.json")
|
||||
require.Contains(t, files, "runtime-config.json")
|
||||
require.Contains(t, files, "version.json")
|
||||
|
||||
// Verify version.json contains version information
|
||||
assert.Contains(t, string(files["version.json"]), `"version":"dev"`)
|
||||
|
||||
// Verify static config contains entry points
|
||||
assert.Contains(t, string(files["static-config.json"]), `"entryPoints":{"web":{"address":"xxxx","http":{}}}`)
|
||||
|
||||
// Verify runtime config contains services
|
||||
assert.Contains(t, string(files["runtime-config.json"]), `"services":`)
|
||||
assert.Contains(t, string(files["runtime-config.json"]), `"test-service"`)
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testCases {
|
||||
t.Run(test.desc, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := New(test.confStatic, &test.confDyn)
|
||||
server := httptest.NewServer(handler.createRouter())
|
||||
|
||||
resp, err := http.DefaultClient.Get(server.URL + test.path)
|
||||
require.NoError(t, err)
|
||||
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
assert.Equal(t, "application/gzip", resp.Header.Get("Content-Type"))
|
||||
assert.Equal(t, `attachment; filename=support-dump.tar.gz`, resp.Header.Get("Content-Disposition"))
|
||||
|
||||
// Extract and validate the tar.gz contents.
|
||||
files, err := extractTarGz(resp.Body)
|
||||
require.NoError(t, err)
|
||||
|
||||
test.validate(t, files)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// extractTarGz reads a tar.gz archive and returns a map of filename to contents
|
||||
func extractTarGz(r io.Reader) (map[string][]byte, error) {
|
||||
files := make(map[string][]byte)
|
||||
|
||||
gzr, err := gzip.NewReader(r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer gzr.Close()
|
||||
|
||||
tr := tar.NewReader(gzr)
|
||||
for {
|
||||
header, err := tr.Next()
|
||||
if errors.Is(err, io.EOF) {
|
||||
break
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if header.Typeflag != tar.TypeReg {
|
||||
continue
|
||||
}
|
||||
|
||||
contents, err := io.ReadAll(tr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
files[header.Name] = contents
|
||||
}
|
||||
|
||||
return files, nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue