You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.0 KiB
77 lines
2.0 KiB
2 months ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"tera-clone/services/tera-api/internal/directstream"
|
||
|
|
"tera-clone/services/tera-api/internal/serial"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ticketStatusStub struct {
|
||
|
|
status directstream.TicketServiceSnapshot
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s ticketStatusStub) Status(context.Context) (directstream.TicketServiceSnapshot, error) {
|
||
|
|
return s.status, s.err
|
||
|
|
}
|
||
|
|
|
||
|
|
type serialStatusStub struct {
|
||
|
|
status serial.StreamSnapshot
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s serialStatusStub) Status(context.Context) (serial.StreamSnapshot, error) {
|
||
|
|
return s.status, s.err
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSessiondHealthAndReadinessExposeDependencyFailures(t *testing.T) {
|
||
|
|
broker, err := directstream.NewBroker(unavailableTransportFactory{}, directstream.BrokerOptions{})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer broker.Close()
|
||
|
|
metrics := &directstream.WebSocketMetrics{}
|
||
|
|
handler := newSessiondStatusHandler(
|
||
|
|
broker, metrics,
|
||
|
|
ticketStatusStub{status: directstream.TicketServiceSnapshot{ActiveBindings: 1}},
|
||
|
|
serialStatusStub{err: errors.New("seriald unavailable")},
|
||
|
|
)
|
||
|
|
|
||
|
|
health := httptest.NewRecorder()
|
||
|
|
handler(false).ServeHTTP(health, httptest.NewRequest(http.MethodGet, "/healthz", nil))
|
||
|
|
if health.Code != http.StatusOK {
|
||
|
|
t.Fatalf("health status = %d", health.Code)
|
||
|
|
}
|
||
|
|
if body := health.Body.String(); !containsAll(body, `"status":"degraded"`, `"serial_error":"seriald unavailable"`, `"active_bindings":1`) {
|
||
|
|
t.Fatalf("health body = %s", body)
|
||
|
|
}
|
||
|
|
|
||
|
|
ready := httptest.NewRecorder()
|
||
|
|
handler(true).ServeHTTP(ready, httptest.NewRequest(http.MethodGet, "/readyz", nil))
|
||
|
|
if ready.Code != http.StatusServiceUnavailable {
|
||
|
|
t.Fatalf("readiness status = %d, body = %s", ready.Code, ready.Body.String())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type unavailableTransportFactory struct{}
|
||
|
|
|
||
|
|
func (unavailableTransportFactory) Open(context.Context, string) (io.ReadWriteCloser, error) {
|
||
|
|
return nil, errors.New("not used")
|
||
|
|
}
|
||
|
|
|
||
|
|
func containsAll(value string, expected ...string) bool {
|
||
|
|
for _, item := range expected {
|
||
|
|
if !strings.Contains(value, item) {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|