Function TestQueryTrace

pkg/httpclient/client_test.go:22–77

(t *testing.T)

}

func TestQueryTrace(t *testing.T) {
	trace := &tempopb.Trace{}
	t.Run("returns a trace when is found", func(t *testing.T) {
		mockTransport := MockRoundTripper(func(req *http.Request) *http.Response {
			assert.Equal(t, "www.tempo.com/api/traces/100", req.URL.Path)
			assert.Equal(t, "application/protobuf", req.Header.Get("Accept"))
			response, _ := proto.Marshal(trace)
			return &http.Response{
				StatusCode: 200,
				Body:       io.NopCloser(bytes.NewReader(response)),
			}
		})

		client := New("www.tempo.com", "1000")
		client.WithTransport(mockTransport)
		response, err := client.QueryTrace("100")

		assert.NoError(t, err)
		assert.True(t, proto.Equal(trace, response))
	})

	t.Run("includes the recentDataTarget header", func(t *testing.T) {
		mockTransport := MockRoundTripper(func(req *http.Request) *http.Response {
			assert.Equal(t, liveStoreHeaderValue, req.Header.Get(recentDataTargetHeader))
			response, _ := proto.Marshal(trace)
			return &http.Response{
				StatusCode: 200,
				Body:       io.NopCloser(bytes.NewReader(response)),
			}
		})

		client := New("www.tempo.com", "1000")
		client.QueryLiveStores = true
		client.WithTransport(mockTransport)
		response, err := client.QueryTrace("100")

		assert.NoError(t, err)
		assert.True(t, proto.Equal(trace, response))
	})

	t.Run("returns a trace not found error on 404", func(t *testing.T) {
		mockTransport := MockRoundTripper(func(_ *http.Request) *http.Response {
			return &http.Response{
				StatusCode: 404,
				Body:       nil,
			}
		})

		client := New("www.tempo.com", "1000")
		client.WithTransport(mockTransport)
		response, err := client.QueryTrace("notfound")

		assert.Error(t, err)
		assert.Nil(t, response)
	})
}

func TestQueryTraceWithRange(t *testing.T) {