Function TestQueryLimits

integration/limits/limits_test.go:188–233

(t *testing.T)

}

func TestQueryLimits(t *testing.T) {
	util.RunIntegrationTests(t, util.TestHarnessConfig{}, func(h *util.TempoHarness) {
		h.WaitTracesWritable(t)

		batch := util.MakeThriftBatchWithSpanCount(5)
		require.NoError(t, h.WriteJaegerBatch(batch, ""))

		// confirm the trace is queryable before adjusting overrides. otherwise the override
		// update might beat the ingestion into livestore and the test will be flaky
		h.WaitTracesQueryable(t, 1)

		// retroactively make the trace too large so it will fail on querying
		err := h.UpdateOverrides(map[string]*overrides.Overrides{
			"single-tenant": {
				Global: overrides.GlobalOverrides{
					MaxBytesPerTrace: 1,
				},
			},
		})
		require.NoError(t, err)

		// calc trace id
		traceID := [16]byte{}
		binary.BigEndian.PutUint64(traceID[:8], uint64(batch.Spans[0].TraceIdHigh))
		binary.BigEndian.PutUint64(traceID[8:], uint64(batch.Spans[0].TraceIdLow))

		// now try to query it back. this should fail b/c the trace is too large
		apiClient := h.APIClientHTTP("")

		_, err = apiClient.QueryTrace(tempoUtil.TraceIDToHexString(traceID[:]))
		require.ErrorContains(t, err, trace.ErrTraceTooLarge.Error())
		require.ErrorContains(t, err, "failed with response: 422") // confirm frontend returns 422

		// wait for live store to complete the block
		liveStore := h.Services[util.ServiceLiveStoreZoneA]
		err = liveStore.WaitSumMetricsWithOptions(e2e.Greater(0),
			[]string{"tempo_live_store_blocks_completed_total"},
			e2e.WaitMissingMetrics,
		)
		require.NoError(t, err)

		_, err = apiClient.QueryTrace(tempoUtil.TraceIDToHexString(traceID[:]))
		require.ErrorContains(t, err, trace.ErrTraceTooLarge.Error())
		require.ErrorContains(t, err, "failed with response: 422") // confirm frontend returns 422
	})
}

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