diff --git a/sdk/core/core-tracing/test/internal/tracingClient.spec.ts b/sdk/core/core-tracing/test/internal/tracingClient.spec.ts index 51c434fa2237..ab8cc4369b3e 100644 --- a/sdk/core/core-tracing/test/internal/tracingClient.spec.ts +++ b/sdk/core/core-tracing/test/internal/tracingClient.spec.ts @@ -200,4 +200,29 @@ describe("TracingClient", () => { }); }); }); + + describe("#parseTraceparentHeader", () => { + it("delegates to the instrumenter", () => { + const expectedContext = createTracingContext(); + const parseTraceparentHeaderSpy = vi + .spyOn(instrumenter, "parseTraceparentHeader") + .mockReturnValue(expectedContext); + const result = client.parseTraceparentHeader("00-traceid-spanid-01"); + expect(parseTraceparentHeaderSpy).toHaveBeenCalledWith("00-traceid-spanid-01"); + assert.strictEqual(result, expectedContext); + }); + }); + + describe("#createRequestHeaders", () => { + it("delegates to the instrumenter", () => { + const expectedHeaders = { traceparent: "00-traceid-spanid-01" }; + const createRequestHeadersSpy = vi + .spyOn(instrumenter, "createRequestHeaders") + .mockReturnValue(expectedHeaders); + const tracingContext = createTracingContext(); + const result = client.createRequestHeaders(tracingContext); + expect(createRequestHeadersSpy).toHaveBeenCalledWith(tracingContext); + assert.deepEqual(result, expectedHeaders); + }); + }); });