.examples
.github
alerting
client
config
controller
handler
badge.go
badge_test.go
chart.go
chart_test.go
cors.go
endpoint_status.go
endpoint_status_test.go
favicon.go
favicon_test.go
gzip.go
handler.go
handler_test.go
spa.go
spa_test.go
util.go
util_test.go
controller.go
controller_test.go
core
docs
jsonpath
metric
pattern
security
storage
util
vendor
watchdog
web
.dockerignore
.gitattributes
.gitignore
Dockerfile
LICENSE.md
Makefile
README.md
config.yaml
go.mod
go.sum
main.go
81 lines
2.4 KiB
Go
81 lines
2.4 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/TwiN/gatus/v3/config"
|
|
"github.com/TwiN/gatus/v3/core"
|
|
"github.com/TwiN/gatus/v3/storage"
|
|
"github.com/TwiN/gatus/v3/watchdog"
|
|
)
|
|
|
|
func TestResponseTimeChart(t *testing.T) {
|
|
defer storage.Get().Clear()
|
|
defer cache.Clear()
|
|
cfg := &config.Config{
|
|
Metrics: true,
|
|
Endpoints: []*core.Endpoint{
|
|
{
|
|
Name: "frontend",
|
|
Group: "core",
|
|
},
|
|
{
|
|
Name: "backend",
|
|
Group: "core",
|
|
},
|
|
},
|
|
}
|
|
watchdog.UpdateEndpointStatuses(cfg.Endpoints[0], &core.Result{Success: true, Duration: time.Millisecond, Timestamp: time.Now()})
|
|
watchdog.UpdateEndpointStatuses(cfg.Endpoints[1], &core.Result{Success: false, Duration: time.Second, Timestamp: time.Now()})
|
|
router := CreateRouter("../../web/static", cfg.Security, nil, cfg.Metrics)
|
|
type Scenario struct {
|
|
Name string
|
|
Path string
|
|
ExpectedCode int
|
|
Gzip bool
|
|
}
|
|
scenarios := []Scenario{
|
|
{
|
|
Name: "chart-response-time-24h",
|
|
Path: "/api/v1/endpoints/core_backend/response-times/24h/chart.svg",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "chart-response-time-7d",
|
|
Path: "/api/v1/endpoints/core_frontend/response-times/7d/chart.svg",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "chart-response-time-with-invalid-duration",
|
|
Path: "/api/v1/endpoints/core_backend/response-times/3d/chart.svg",
|
|
ExpectedCode: http.StatusBadRequest,
|
|
},
|
|
{
|
|
Name: "chart-response-time-for-invalid-key",
|
|
Path: "/api/v1/endpoints/invalid_key/response-times/7d/chart.svg",
|
|
ExpectedCode: http.StatusNotFound,
|
|
},
|
|
{ // XXX: Remove this in v4.0.0
|
|
Name: "backward-compatible-services-chart-response-time-24h",
|
|
Path: "/api/v1/services/core_backend/response-times/24h/chart.svg",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
}
|
|
for _, scenario := range scenarios {
|
|
t.Run(scenario.Name, func(t *testing.T) {
|
|
request, _ := http.NewRequest("GET", scenario.Path, nil)
|
|
if scenario.Gzip {
|
|
request.Header.Set("Accept-Encoding", "gzip")
|
|
}
|
|
responseRecorder := httptest.NewRecorder()
|
|
router.ServeHTTP(responseRecorder, request)
|
|
if responseRecorder.Code != scenario.ExpectedCode {
|
|
t.Errorf("%s %s should have returned %d, but returned %d instead", request.Method, request.URL, scenario.ExpectedCode, responseRecorder.Code)
|
|
}
|
|
})
|
|
}
|
|
}
|