Function seedTestData

modules/sentry/api_test.go:11–68

seedTestData populates the test DB with sample data for API tests.

(t *testing.T, mux *http.ServeMux)

// seedTestData populates the test DB with sample data for API tests.
func seedTestData(t *testing.T, mux *http.ServeMux) {
	t.Helper()
	db := setupTestDB(t)
	registerAPI(mux, db)

	// Insert error events.
	for i, evt := range []struct {
		eventID string
		level   string
		excType string
		excVal  string
		traceID string
	}{
		{"evt-1", "error", "RuntimeException", "Something broke", "trace-abc"},
		{"evt-2", "error", "RuntimeException", "Something broke", "trace-abc"},
		{"evt-3", "warning", "LogicException", "Bad logic", ""},
	} {
		payload, _ := json.Marshal(map[string]any{
			"event_id":    evt.eventID,
			"level":       evt.level,
			"environment": "production",
			"exception":   map[string]any{"values": []map[string]any{{"type": evt.excType, "value": evt.excVal}}},
		})
		var ev ErrorEvent
		json.Unmarshal(payload, &ev)
		if evt.traceID != "" {
			ev.Contexts = &Contexts{Trace: &TraceContext{TraceID: evt.traceID, SpanID: "span-" + evt.eventID}}
		}
		_, err := storeErrorEvent(db, &ev, payload, "default")
		if err != nil {
			t.Fatalf("seed error event %d: %v", i, err)
		}
	}

	// Insert a transaction with spans.
	txnPayload, _ := json.Marshal(map[string]any{
		"event_id":        "txn-1",
		"type":            "transaction",
		"transaction":     "GET /api/users",
		"start_timestamp": 1678272500.0,
		"timestamp":       1678272503.0,
		"contexts":        map[string]any{"trace": map[string]any{"trace_id": "trace-abc", "span_id": "root"}},
		"spans": []map[string]any{
			{"span_id": "s1", "trace_id": "trace-abc", "op": "db.query", "description": "SELECT *", "start_timestamp": 1678272501.0, "timestamp": 1678272502.0, "status": "ok", "data": map[string]string{"server.address": "mysql:3306"}},
			{"span_id": "s2", "trace_id": "trace-abc", "op": "http.client", "description": "stripe.com", "start_timestamp": 1678272502.0, "timestamp": 1678272502.5, "status": "internal_error", "data": map[string]string{"http.url": "https://stripe.com/charge"}},
		},
	})
	var txn Transaction
	json.Unmarshal(txnPayload, &txn)
	storeTransaction(db, &txn, txnPayload)

	// Insert logs.
	storeLogs(db, []LogRecord{
		{TraceID: "trace-abc", Level: "info", Body: "Request started", SeverityNumber: 9},
		{TraceID: "trace-abc", Level: "warning", Body: "Slow query", SeverityNumber: 13},
		{Level: "error", Body: "Connection refused", SeverityNumber: 17},
	})
}