Function TestLimitsPartialSuccess

integration/limits/limits_test.go:235–280

(t *testing.T)

}

func TestLimitsPartialSuccess(t *testing.T) {
	util.RunIntegrationTests(t, util.TestHarnessConfig{
		ConfigOverlay: configIngestPartialSucess,
	}, func(h *util.TempoHarness) {
		h.WaitTracesWritable(t)
		// make request
		traceIDs := make([][]byte, 6)
		for index := range traceIDs {
			traceID := make([]byte, 16)
			_, err := crand.Read(traceID)
			require.NoError(t, err)
			traceIDs[index] = traceID
		}

		// 3 traces with trace_too_large and 3 with no error
		spanCountsByTrace := []int{1, 4, 1, 5, 6, 1}
		trace := test.MakeReqWithMultipleTraceWithSpanCount(spanCountsByTrace, traceIDs)

		// send traces to tempo
		// partial success = no error
		require.NoError(t, h.WriteTempoProtoTraces(trace, ""))

		// wait for live store to ingest traces
		h.WaitTracesQueryable(t, 3) // only 3 traces are small enough to be ingested

		// query for the traces that didn't trigger an error
		apiClient := h.APIClientHTTP("")
		for i, count := range spanCountsByTrace {
			if count == 1 {
				result, err := apiClient.QueryTrace(tempoUtil.TraceIDToHexString(traceIDs[i]))
				require.NoError(t, err)
				assert.Equal(t, 1, len(result.ResourceSpans))
			}
		}

		// test metrics
		// 3 traces with trace_too_large each with 4+5+6 spans
		liveStore := h.Services[util.ServiceLiveStoreZoneA]
		err := liveStore.WaitSumMetricsWithOptions(e2e.Equals(15),
			[]string{"tempo_discarded_spans_total"},
			e2e.WithLabelMatchers(labels.MustNewMatcher(labels.MatchEqual, "reason", "trace_too_large")),
			e2e.WaitMissingMetrics,
		)
		require.NoError(t, err)
	})
}

func TestQueryRateLimits(t *testing.T) {
	util.RunIntegrationTests(t, util.TestHarnessConfig{