Function TestTraceBuilder_BuildFromEvents

pkg/dashboard/trace_builder_test.go:10–54

(t *testing.T)

)

func TestTraceBuilder_BuildFromEvents(t *testing.T) {
	tb := NewTraceBuilder()

	now := time.Now()
	events := []types.AgentEventEnvelope{
		{
			Cursor: 1,
			Bookmark: types.Bookmark{
				Cursor:    1,
				Timestamp: now.UnixMilli(),
			},
			Event: types.MonitorStepCompleteEvent{
				Step:       1,
				DurationMs: 100,
			},
		},
		{
			Cursor: 2,
			Bookmark: types.Bookmark{
				Cursor:    2,
				Timestamp: now.Add(100 * time.Millisecond).UnixMilli(),
			},
			Event: types.MonitorTokenUsageEvent{
				InputTokens:  1000,
				OutputTokens: 500,
				TotalTokens:  1500,
			},
		},
	}

	traces := tb.BuildFromEvents(events)

	if len(traces) == 0 {
		t.Fatal("BuildFromEvents() returned no traces")
	}

	trace := traces[0]
	if trace.SpanCount == 0 {
		t.Error("BuildFromEvents() trace has no spans")
	}

	if trace.TokenUsage.Total == 0 {
		t.Error("BuildFromEvents() trace has no token usage")
	}
}

func TestTraceBuilder_BuildSpanTree(t *testing.T) {
	tb := NewTraceBuilder()