mirror of
https://github.com/hanzoai/o11y.git
synced 2026-10-15 12:58:33 +00:00
zeekay
b9f9cdf918
Re-sync the entire SigNoz-derived Go tree to a clean fork of latest SigNoz main (upstream 3e6339019), eliminating the version-skew that broke `go build ./...` (mixed-era packages, ~14 imported-but-absent packages, partial signoz->o11y / clickhouse->datastore rebrands, self-import cycle, duplicate package decls). - pkg/ + cmd/ wholesale replaced with latest SigNoz; imports rewritten github.com/SigNoz/signoz/ -> github.com/hanzoai/o11y/. - Adopt SigNoz's CURRENT structure (resourcefilter->telemetryresourcefilter, integrationtypes->cloudintegrationtypes, roletypes/grammar/times folded in upstream); old paths dropped. - Fork deps repathed to hanzoai + bumped: signoz-otel-collector v0.144.6, clickhouse-go-mock v0.14.1, govaluate v0.1.0; expr-lang/expr => hanzoai/expr v1.17.8 (replace). go.mod reconciled from latest SigNoz; go.sum re-tidied. - ee/ + cmd/enterprise stripped — o11y stays MIT-only; community edition imports no ee packages. - Preserved o11y additions: module path github.com/hanzoai/o11y, mount.go (cloud.Register/zip mount adapter, HIP-0106), NOTICE + LICENSE (SigNoz §4c attribution retained; upstream pkg files are headerless per SigNoz convention). - Reverted o11y cosmetic renames (o11y*/datastore*) and experimental originals (authz/iamauthz, billing, zapreceiver, zapmetricreceiver) to SigNoz canonical to kill the skew; snapshots retained in git history for re-layering. go build ./... = exit 0. go vet clean except benign "unreachable code" in generated ANTLR parser files (inherited from upstream).
93 lines
3.1 KiB
Go
93 lines
3.1 KiB
Go
package querier
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/o11y/pkg/types/spantypes"
|
|
)
|
|
|
|
func TestMergeSpanAttributeColumns_ParsesEventsAndLinks(t *testing.T) {
|
|
data := map[string]any{
|
|
"attributes_string": map[string]string{"http.method": "GET"},
|
|
"attributes_number": map[string]float64{"http.status_code": 200},
|
|
"attributes_bool": map[string]bool{"is_root": true},
|
|
"resources_string": map[string]string{"service.name": "api"},
|
|
"events": []string{
|
|
`{"name":"request_received","timeUnixNano":1778489782759245000,"attributeMap":{"http.method":"GET","http.route":"/api/chat"}}`,
|
|
`{"name":"cache_lookup","timeUnixNano":1778489782811697000,"attributeMap":{"cache.hit":"true","cache.key":"user:123:prompt"}}`,
|
|
},
|
|
"links": `[{"traceId":"abc","spanId":"123","refType":"CHILD_OF"},{"traceId":"def","spanId":"456","refType":"FOLLOWS_FROM"}]`,
|
|
}
|
|
|
|
mergeSpanAttributeColumns(data)
|
|
|
|
attrs, ok := data["attributes"].(map[string]any)
|
|
if !ok {
|
|
t.Fatalf("expected attributes to be map[string]any, got %T", data["attributes"])
|
|
}
|
|
if attrs["http.method"] != "GET" || attrs["http.status_code"] != float64(200) || attrs["is_root"] != true {
|
|
t.Fatalf("attributes not merged correctly: %#v", attrs)
|
|
}
|
|
|
|
res, ok := data["resource"].(map[string]string)
|
|
if !ok || res["service.name"] != "api" {
|
|
t.Fatalf("resource not set correctly: %#v", data["resource"])
|
|
}
|
|
|
|
for _, removed := range []string{"attributes_string", "attributes_number", "attributes_bool", "resources_string"} {
|
|
if _, present := data[removed]; present {
|
|
t.Fatalf("expected %s to be removed", removed)
|
|
}
|
|
}
|
|
|
|
events, ok := data["events"].([]spantypes.EventV2)
|
|
if !ok {
|
|
t.Fatalf("expected events to be []spantypes.EventV2, got %T", data["events"])
|
|
}
|
|
wantEvents := []spantypes.EventV2{
|
|
{
|
|
Name: "request_received",
|
|
TimeUnixNano: 1778489782759245000,
|
|
Attributes: map[string]any{"http.method": "GET", "http.route": "/api/chat"},
|
|
IsError: false,
|
|
},
|
|
{
|
|
Name: "cache_lookup",
|
|
TimeUnixNano: 1778489782811697000,
|
|
Attributes: map[string]any{"cache.hit": "true", "cache.key": "user:123:prompt"},
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(events, wantEvents) {
|
|
t.Fatalf("events parsed incorrectly:\n got: %#v\nwant: %#v", events, wantEvents)
|
|
}
|
|
|
|
links, ok := data["links"].([]spantypes.Link)
|
|
if !ok {
|
|
t.Fatalf("expected links to be []spantypes.Link, got %T", data["links"])
|
|
}
|
|
wantLinks := []spantypes.Link{
|
|
{TraceID: "abc", SpanID: "123"},
|
|
{TraceID: "def", SpanID: "456"},
|
|
}
|
|
if !reflect.DeepEqual(links, wantLinks) {
|
|
t.Fatalf("links parsed incorrectly:\n got: %#v\nwant: %#v", links, wantLinks)
|
|
}
|
|
}
|
|
|
|
func TestMergeSpanAttributeColumns_EmptyEventsAndLinks(t *testing.T) {
|
|
data := map[string]any{
|
|
"events": []string{},
|
|
"links": "[]",
|
|
}
|
|
|
|
mergeSpanAttributeColumns(data)
|
|
|
|
if events, ok := data["events"].([]spantypes.EventV2); !ok || len(events) != 0 {
|
|
t.Fatalf("expected empty []spantypes.EventV2, got %#v", data["events"])
|
|
}
|
|
if links, ok := data["links"].([]spantypes.Link); !ok || len(links) != 0 {
|
|
t.Fatalf("expected empty []spantypes.Link, got %#v", data["links"])
|
|
}
|
|
}
|