mirror of https://github.com/linkerd/linkerd2.git
57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/linkerd/linkerd2/controller/api/public"
|
|
"github.com/linkerd/linkerd2/pkg/k8s"
|
|
)
|
|
|
|
func TestStat(t *testing.T) {
|
|
t.Run("Returns namespace stats", func(t *testing.T) {
|
|
mockClient := &public.MockApiClient{}
|
|
|
|
counts := &public.PodCounts{
|
|
MeshedPods: 1,
|
|
RunningPods: 2,
|
|
FailedPods: 0,
|
|
}
|
|
|
|
response := public.GenStatSummaryResponse("emoji", k8s.Namespace, "emojivoto", counts)
|
|
|
|
mockClient.StatSummaryResponseToReturn = &response
|
|
|
|
expectedOutput := `NAME MESHED SUCCESS RPS LATENCY_P50 LATENCY_P95 LATENCY_P99 TLS
|
|
emoji 1/2 100.00% 2.0rps 123ms 123ms 123ms 100%
|
|
`
|
|
|
|
options := newStatOptions()
|
|
args := []string{"ns"}
|
|
req, err := buildStatSummaryRequest(args, options)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
output, err := requestStatsFromAPI(mockClient, req, options)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
if output != expectedOutput {
|
|
t.Fatalf("Wrong output:\n expected: \n%s\n, got: \n%s", expectedOutput, output)
|
|
}
|
|
})
|
|
|
|
t.Run("Returns an error for named resource queries with the --all-namespaces flag", func(t *testing.T) {
|
|
options := newStatOptions()
|
|
options.allNamespaces = true
|
|
args := []string{"po", "web"}
|
|
expectedError := "stats for a resource cannot be retrieved by name across all namespaces"
|
|
|
|
_, err := buildStatSummaryRequest(args, options)
|
|
if err == nil || err.Error() != expectedError {
|
|
t.Fatalf("Expected error [%s] instead got [%s]", expectedError, err)
|
|
}
|
|
})
|
|
}
|