mirror of https://github.com/linkerd/linkerd2.git
69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/linkerd/linkerd2/controller/api/public"
|
|
)
|
|
|
|
type routesParamsExp struct {
|
|
options *routesOptions
|
|
routes []string
|
|
counts []uint64
|
|
file string
|
|
}
|
|
|
|
func TestRoutes(t *testing.T) {
|
|
options := newRoutesOptions()
|
|
t.Run("Returns route stats", func(t *testing.T) {
|
|
testRoutesCall(routesParamsExp{
|
|
routes: []string{"/a", "/b", "/c"},
|
|
counts: []uint64{90, 60, 0, 30},
|
|
options: options,
|
|
file: "routes_one_output.golden",
|
|
}, t)
|
|
})
|
|
|
|
options.outputFormat = jsonOutput
|
|
t.Run("Returns route stats (json)", func(t *testing.T) {
|
|
testRoutesCall(routesParamsExp{
|
|
routes: []string{"/a", "/b", "/c"},
|
|
counts: []uint64{90, 60, 0, 30},
|
|
options: options,
|
|
file: "routes_one_output_json.golden",
|
|
}, t)
|
|
})
|
|
|
|
wideOptions := newRoutesOptions()
|
|
wideOptions.toResource = "deploy/bar"
|
|
wideOptions.outputFormat = wideOutput
|
|
t.Run("Returns wider route stats", func(t *testing.T) {
|
|
testRoutesCall(routesParamsExp{
|
|
routes: []string{"/a", "/b", "/c"},
|
|
counts: []uint64{90, 60, 0, 30},
|
|
options: wideOptions,
|
|
file: "routes_one_output_wide.golden",
|
|
}, t)
|
|
})
|
|
}
|
|
|
|
func testRoutesCall(exp routesParamsExp, t *testing.T) {
|
|
mockClient := &public.MockAPIClient{}
|
|
|
|
response := public.GenTopRoutesResponse(exp.routes, exp.counts, exp.options.toResource != "", "foobar")
|
|
|
|
mockClient.TopRoutesResponseToReturn = response
|
|
|
|
req, err := buildTopRoutesRequest("deploy/foobar", exp.options)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
output, err := requestRouteStatsFromAPI(mockClient, req, exp.options)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
diffTestdata(t, exp.file, output)
|
|
}
|