mirror of
https://github.com/hanzoai/status.git
synced 2026-10-30 18:21:48 +00:00
zeekayandHanzo Dev
78ab4f2f83
zip's registration is a verb method on the group, so a route that is not a typed operation says which method it answers with. The security package's own two-method router goes with it: it existed so both the app and a sub-router could satisfy one name, and restating a routing surface is what let the two disagree — the methods it named no longer exist. The app spells itself as a group, which registers at exactly the paths it did. The toolchain and the pin move together, because below zip v1.36.50 the JSON encoder was selected on a build tag: a repo whose go directive rises first writes a wire that is neither the old release's nor the new one's until the pin catches up. metrics_test.go still does not compile, before and after: it compares a metric registry against a gatherer from a different metrics library, so the two Gather signatures disagree. That is its own change. Co-authored-by: Hanzo Dev <[email protected]>
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"math/rand"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/zap-proto/zip"
|
|
"hanzo.ai/status/config"
|
|
"hanzo.ai/status/config/endpoint"
|
|
"hanzo.ai/status/config/web"
|
|
)
|
|
|
|
func TestHandle(t *testing.T) {
|
|
cfg := &config.Config{
|
|
Web: &web.Config{
|
|
Address: "0.0.0.0",
|
|
Port: rand.Intn(65534),
|
|
},
|
|
Endpoints: []*endpoint.Endpoint{
|
|
{
|
|
Name: "frontend",
|
|
Group: "core",
|
|
},
|
|
{
|
|
Name: "backend",
|
|
Group: "core",
|
|
},
|
|
},
|
|
}
|
|
_ = os.Setenv("ROUTER_TEST", "true")
|
|
_ = os.Setenv("ENVIRONMENT", "dev")
|
|
defer os.Clearenv()
|
|
Handle(cfg)
|
|
defer Shutdown()
|
|
request := httptest.NewRequest("GET", "/health", http.NoBody)
|
|
response, err := app.Fiber().Test(request)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if response.StatusCode != 200 {
|
|
t.Error("expected GET /health to return status code 200")
|
|
}
|
|
if app == nil {
|
|
t.Fatal("server should've been set (but because we set ROUTER_TEST, it shouldn't have been started)")
|
|
}
|
|
}
|
|
|
|
func TestHandleTLS(t *testing.T) {
|
|
scenarios := []struct {
|
|
name string
|
|
tls *web.TLSConfig
|
|
expectedStatusCode int
|
|
}{
|
|
{
|
|
name: "good-tls-config",
|
|
tls: &web.TLSConfig{CertificateFile: "../testdata/cert.pem", PrivateKeyFile: "../testdata/cert.key"},
|
|
expectedStatusCode: 200,
|
|
},
|
|
}
|
|
for _, scenario := range scenarios {
|
|
t.Run(scenario.name, func(t *testing.T) {
|
|
cfg := &config.Config{
|
|
Web: &web.Config{Address: "0.0.0.0", Port: rand.Intn(65534), TLS: scenario.tls},
|
|
Endpoints: []*endpoint.Endpoint{
|
|
{Name: "frontend", Group: "core"},
|
|
{Name: "backend", Group: "core"},
|
|
},
|
|
}
|
|
if err := cfg.Web.ValidateAndSetDefaults(); err != nil {
|
|
t.Error("expected no error from web (TLS) validation, got", err)
|
|
}
|
|
_ = os.Setenv("ROUTER_TEST", "true")
|
|
_ = os.Setenv("ENVIRONMENT", "dev")
|
|
defer os.Clearenv()
|
|
Handle(cfg)
|
|
defer Shutdown()
|
|
request := httptest.NewRequest("GET", "/health", http.NoBody)
|
|
response, err := app.Fiber().Test(request)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if response.StatusCode != scenario.expectedStatusCode {
|
|
t.Errorf("%s %s should have returned %d, but returned %d instead", request.Method, request.URL, scenario.expectedStatusCode, response.StatusCode)
|
|
}
|
|
if app == nil {
|
|
t.Fatal("server should've been set (but because we set ROUTER_TEST, it shouldn't have been started)")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestShutdown(t *testing.T) {
|
|
// Pretend that we called controller.Handle(), which initializes the server variable
|
|
app = zip.New(zip.Config{})
|
|
Shutdown()
|
|
if app != nil {
|
|
t.Error("server should've been shut down")
|
|
}
|
|
}
|