feat: add project-owned error catalogs and HTTP 200 failure contracts
This commit is contained in:
51
telemetry/errorCatalog_test.go
Normal file
51
telemetry/errorCatalog_test.go
Normal file
@@ -0,0 +1,51 @@
|
||||
package telemetry_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
omni "git.yuntuo.hk/k2safe/go-omni"
|
||||
apierrors "git.yuntuo.hk/k2safe/go-omni/errors"
|
||||
"git.yuntuo.hk/k2safe/go-omni/response"
|
||||
"git.yuntuo.hk/k2safe/go-omni/telemetry"
|
||||
"go.opentelemetry.io/otel/codes"
|
||||
"go.opentelemetry.io/otel/sdk/trace/tracetest"
|
||||
)
|
||||
|
||||
func TestCatalogFailureTraceWithHTTP200(t *testing.T) {
|
||||
exporter := tracetest.NewInMemoryExporter()
|
||||
runtime, err := telemetry.New(telemetry.Config{Service: "fixture", SampleRatio: 1, Exporter: exporter})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer runtime.Close()
|
||||
app := omni.New(omni.WithErrors(apierrors.Catalog{DefaultStatus: 200, Definitions: map[string]apierrors.Definition{"INVALID": {Code: 123, Message: "invalid"}}}), omni.WithResponse(response.Config{CodeType: "int32", SuccessCode: 0, ErrorCode: 9999}))
|
||||
defer app.Close()
|
||||
app.Use(runtime.Middleware())
|
||||
app.GET("/", func(c *omni.Context) error { c.WriteError(apierrors.ByCode("INVALID")); return nil })
|
||||
w := httptest.NewRecorder()
|
||||
app.ServeHTTP(w, httptest.NewRequest("GET", "/", nil))
|
||||
if w.Code != 200 {
|
||||
t.Fatal(w.Code)
|
||||
}
|
||||
if err := runtime.Flush(context.Background()); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
spans := exporter.GetSpans()
|
||||
if len(spans) != 1 || spans[0].Status.Code != codes.Error {
|
||||
t.Fatal(spans)
|
||||
}
|
||||
seenStatus, seenFailure := false, false
|
||||
for _, attr := range spans[0].Attributes {
|
||||
if attr.Key == "http.response.status_code" && attr.Value.AsInt64() == 200 {
|
||||
seenStatus = true
|
||||
}
|
||||
if attr.Key == "omni.request.failed" && attr.Value.AsBool() {
|
||||
seenFailure = true
|
||||
}
|
||||
}
|
||||
if !seenStatus || !seenFailure {
|
||||
t.Fatal(spans[0].Attributes)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user