mirror of
https://github.com/hanzoai/status.git
synced 2026-10-30 18:21:48 +00:00
zeekay
833094dc48
The import path a reader types is the one the org owns, rather than the forge that happens to host the bytes. hanzo.ai serves the go-import meta and points it at the same repository, so where the code lives does not change. Rewritten by role. A bare repository URL — a clone target, a badge, a build context — still names where the code is and stays on github.com.
150 lines
4.3 KiB
Go
150 lines
4.3 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"hanzo.ai/status/config"
|
|
"hanzo.ai/status/config/endpoint"
|
|
"hanzo.ai/status/storage/store"
|
|
"hanzo.ai/status/watchdog"
|
|
)
|
|
|
|
func TestResponseTimeChart(t *testing.T) {
|
|
defer store.Get().Clear()
|
|
defer cache.Clear()
|
|
cfg := &config.Config{
|
|
Metrics: true,
|
|
Endpoints: []*endpoint.Endpoint{
|
|
{
|
|
Name: "frontend",
|
|
Group: "core",
|
|
},
|
|
{
|
|
Name: "backend",
|
|
Group: "core",
|
|
},
|
|
},
|
|
}
|
|
watchdog.UpdateEndpointStatus(cfg.Endpoints[0], &endpoint.Result{Success: true, Duration: time.Millisecond, Timestamp: time.Now()})
|
|
watchdog.UpdateEndpointStatus(cfg.Endpoints[1], &endpoint.Result{Success: false, Duration: time.Second, Timestamp: time.Now()})
|
|
api := New(cfg)
|
|
router := api.Router()
|
|
type Scenario struct {
|
|
Name string
|
|
Path string
|
|
ExpectedCode int
|
|
Gzip bool
|
|
}
|
|
scenarios := []Scenario{
|
|
{
|
|
Name: "chart-response-time-24h",
|
|
Path: "/v1/status/endpoints/core_backend/response-times/24h/chart.svg",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "chart-response-time-7d",
|
|
Path: "/v1/status/endpoints/core_frontend/response-times/7d/chart.svg",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "chart-response-time-30d",
|
|
Path: "/v1/status/endpoints/core_frontend/response-times/30d/chart.svg",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "chart-response-time-with-invalid-duration",
|
|
Path: "/v1/status/endpoints/core_backend/response-times/3d/chart.svg",
|
|
ExpectedCode: http.StatusBadRequest,
|
|
},
|
|
{
|
|
Name: "chart-response-time-for-invalid-key",
|
|
Path: "/v1/status/endpoints/invalid_key/response-times/7d/chart.svg",
|
|
ExpectedCode: http.StatusNotFound,
|
|
},
|
|
}
|
|
for _, scenario := range scenarios {
|
|
t.Run(scenario.Name, func(t *testing.T) {
|
|
request := httptest.NewRequest("GET", scenario.Path, http.NoBody)
|
|
if scenario.Gzip {
|
|
request.Header.Set("Accept-Encoding", "gzip")
|
|
}
|
|
response, err := router.Fiber().Test(request)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if response.StatusCode != scenario.ExpectedCode {
|
|
t.Errorf("%s %s should have returned %d, but returned %d instead", request.Method, request.URL, scenario.ExpectedCode, response.StatusCode)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestResponseTimeHistory(t *testing.T) {
|
|
defer store.Get().Clear()
|
|
defer cache.Clear()
|
|
cfg := &config.Config{
|
|
Metrics: true,
|
|
Endpoints: []*endpoint.Endpoint{
|
|
{
|
|
Name: "frontend",
|
|
Group: "core",
|
|
},
|
|
{
|
|
Name: "backend",
|
|
Group: "core",
|
|
},
|
|
},
|
|
}
|
|
watchdog.UpdateEndpointStatus(cfg.Endpoints[0], &endpoint.Result{Success: true, Duration: time.Millisecond, Timestamp: time.Now()})
|
|
watchdog.UpdateEndpointStatus(cfg.Endpoints[1], &endpoint.Result{Success: false, Duration: time.Second, Timestamp: time.Now()})
|
|
api := New(cfg)
|
|
router := api.Router()
|
|
type Scenario struct {
|
|
Name string
|
|
Path string
|
|
ExpectedCode int
|
|
}
|
|
scenarios := []Scenario{
|
|
{
|
|
Name: "history-response-time-24h",
|
|
Path: "/v1/status/endpoints/core_backend/response-times/24h/history",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "history-response-time-7d",
|
|
Path: "/v1/status/endpoints/core_frontend/response-times/7d/history",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "history-response-time-30d",
|
|
Path: "/v1/status/endpoints/core_frontend/response-times/30d/history",
|
|
ExpectedCode: http.StatusOK,
|
|
},
|
|
{
|
|
Name: "history-response-time-with-invalid-duration",
|
|
Path: "/v1/status/endpoints/core_backend/response-times/3d/history",
|
|
ExpectedCode: http.StatusBadRequest,
|
|
},
|
|
{
|
|
Name: "history-response-time-for-invalid-key",
|
|
Path: "/v1/status/endpoints/invalid_key/response-times/7d/history",
|
|
ExpectedCode: http.StatusNotFound,
|
|
},
|
|
}
|
|
for _, scenario := range scenarios {
|
|
t.Run(scenario.Name, func(t *testing.T) {
|
|
request := httptest.NewRequest("GET", scenario.Path, http.NoBody)
|
|
response, err := router.Fiber().Test(request)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if response.StatusCode != scenario.ExpectedCode {
|
|
t.Errorf("%s %s should have returned %d, but returned %d instead", request.Method, request.URL, scenario.ExpectedCode, response.StatusCode)
|
|
}
|
|
})
|
|
}
|
|
}
|