Skip to content

Commit 9aa98ee

Browse files
committed
fix unit tests
1 parent cab0799 commit 9aa98ee

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

packages/remix/test/server/serverTimingTracePropagation.test.ts

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,18 +35,22 @@ describe('serverTimingTracePropagation', () => {
3535
vi.clearAllMocks();
3636
vi.mocked(isNodeEnv).mockReturnValue(true);
3737
vi.mocked(getActiveSpan).mockReturnValue(mockSpan);
38+
vi.mocked(getTraceData).mockReturnValue({
39+
'sentry-trace': '12345678901234567890123456789012-1234567890123456-1',
40+
baggage: 'sentry-environment=production,sentry-release=1.0.0',
41+
});
3842
});
3943

4044
describe('generateSentryServerTimingHeader', () => {
4145
it('returns null in browser environments', () => {
42-
vi.mocked(isNodeEnv).mockReturnValue(false);
46+
vi.mocked(isNodeEnv).mockReturnValueOnce(false);
4347

4448
expect(generateSentryServerTimingHeader()).toBeNull();
4549
});
4650

4751
it('returns null without trace data', () => {
48-
vi.mocked(getActiveSpan).mockReturnValue(undefined);
49-
vi.mocked(getTraceData).mockReturnValue({});
52+
vi.mocked(getActiveSpan).mockReturnValueOnce(undefined);
53+
vi.mocked(getTraceData).mockReturnValueOnce({});
5054

5155
expect(generateSentryServerTimingHeader()).toBeNull();
5256
});
@@ -60,8 +64,8 @@ describe('serverTimingTracePropagation', () => {
6064
});
6165

6266
it('falls back to getTraceData without active span', () => {
63-
vi.mocked(getActiveSpan).mockReturnValue(undefined);
64-
vi.mocked(getTraceData).mockReturnValue({
67+
vi.mocked(getActiveSpan).mockReturnValueOnce(undefined);
68+
vi.mocked(getTraceData).mockReturnValueOnce({
6569
'sentry-trace': 'fallback-trace-id-1234567890123456-0',
6670
baggage: 'sentry-fallback=true',
6771
});
@@ -73,7 +77,7 @@ describe('serverTimingTracePropagation', () => {
7377
});
7478

7579
it('generates header in Cloudflare environment when isNodeEnv is false', () => {
76-
vi.mocked(isNodeEnv).mockReturnValue(false);
80+
vi.mocked(isNodeEnv).mockReturnValueOnce(false);
7781

7882
const originalNavigator = globalThis.navigator;
7983
Object.defineProperty(globalThis, 'navigator', {

0 commit comments

Comments
 (0)