mirror of https://github.com/linkerd/linkerd2.git
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package srv
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/julienschmidt/httprouter"
|
|
"github.com/linkerd/linkerd2/controller/api/public"
|
|
pb "github.com/linkerd/linkerd2/controller/gen/public"
|
|
)
|
|
|
|
func TestHandleIndex(t *testing.T) {
|
|
mockApiClient := &public.MockApiClient{
|
|
VersionInfoToReturn: &pb.VersionInfo{
|
|
GoVersion: "the best one",
|
|
BuildDate: "never",
|
|
ReleaseVersion: "0.3.3",
|
|
},
|
|
}
|
|
|
|
server := FakeServer()
|
|
|
|
handler := &handler{
|
|
render: server.RenderTemplate,
|
|
apiClient: mockApiClient,
|
|
}
|
|
|
|
recorder := httptest.NewRecorder()
|
|
req := httptest.NewRequest("GET", "/", nil)
|
|
handler.handleIndex(recorder, req, httprouter.Params{})
|
|
|
|
if recorder.Code != http.StatusOK {
|
|
t.Errorf("Incorrect StatusCode: %+v", recorder.Code)
|
|
t.Errorf("Expected %+v", http.StatusOK)
|
|
}
|
|
|
|
header := http.Header{
|
|
"Content-Type": []string{"text/html"},
|
|
}
|
|
if !reflect.DeepEqual(recorder.Header(), header) {
|
|
t.Errorf("Incorrect headers: %+v", recorder.Header())
|
|
t.Errorf("Expected: %+v", header)
|
|
}
|
|
|
|
actualBody := recorder.Body.String()
|
|
|
|
expectedSubstrings := []string{
|
|
"<div class=\"main\" id=\"main\"",
|
|
"data-release-version=\"0.3.3\"",
|
|
"data-go-version=\"the best one\"",
|
|
"data-controller-namespace=\"\"",
|
|
"data-uuid=\"\"",
|
|
}
|
|
for _, expectedSubstring := range expectedSubstrings {
|
|
if !strings.Contains(actualBody, expectedSubstring) {
|
|
t.Fatalf("Expected string [%s] to be present in [%s]", expectedSubstring, actualBody)
|
|
}
|
|
}
|
|
}
|