Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix(node): Improve OTEL validation logic #13079

Merged
merged 2 commits into from
Jul 29, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions packages/node/src/sdk/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,12 @@ export function validateOpenTelemetrySetup(): void {

const setup = openTelemetrySetupCheck();

const required = ['SentrySpanProcessor', 'SentryContextManager', 'SentryPropagator'] as const;
const required: ReturnType<typeof openTelemetrySetupCheck> = ['SentryContextManager', 'SentryPropagator'];

if (hasTracingEnabled()) {
required.push('SentrySpanProcessor');
}

for (const k of required) {
if (!setup.includes(k)) {
logger.error(
Expand All @@ -206,7 +211,7 @@ export function validateOpenTelemetrySetup(): void {

if (!setup.includes('SentrySampler')) {
logger.warn(
'You have to set up the SentrySampler. Without this, the OpenTelemetry & Sentry integration may still work, but sample rates set for the Sentry SDK will not be respected.',
'You have to set up the SentrySampler. Without this, the OpenTelemetry & Sentry integration may still work, but sample rates set for the Sentry SDK will not be respected. If you use a custom sampler, make sure to use `wrapSamplingDecision`.',
);
}
}
Expand Down
66 changes: 65 additions & 1 deletion packages/node/test/sdk/init.test.ts
Original file line number Diff line number Diff line change
@@ -1,8 +1,10 @@
import type { Integration } from '@sentry/types';
import { logger } from '@sentry/utils';

import * as SentryOpentelemetry from '@sentry/opentelemetry';
import { getClient, getIsolationScope } from '../../src/';
import * as auto from '../../src/integrations/tracing';
import { init } from '../../src/sdk';
import { init, validateOpenTelemetrySetup } from '../../src/sdk';
import { NodeClient } from '../../src/sdk/client';
import { cleanupOtel } from '../helpers/mockSdkInit';

Expand Down Expand Up @@ -193,3 +195,65 @@ describe('init()', () => {
});
});
});

describe('validateOpenTelemetrySetup', () => {
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@andreiborza added unit tests here!

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Awesome, thanks :)

afterEach(() => {
global.__SENTRY__ = {};
cleanupOtel();
jest.clearAllMocks();
});

it('works with correct setup', () => {
const errorSpy = jest.spyOn(logger, 'error').mockImplementation(() => {});
const warnSpy = jest.spyOn(logger, 'warn').mockImplementation(() => {});

jest.spyOn(SentryOpentelemetry, 'openTelemetrySetupCheck').mockImplementation(() => {
return ['SentryContextManager', 'SentryPropagator', 'SentrySampler'];
});

validateOpenTelemetrySetup();

expect(errorSpy).toHaveBeenCalledTimes(0);
expect(warnSpy).toHaveBeenCalledTimes(0);
});

it('works with missing setup, without tracing', () => {
const errorSpy = jest.spyOn(logger, 'error').mockImplementation(() => {});
const warnSpy = jest.spyOn(logger, 'warn').mockImplementation(() => {});

jest.spyOn(SentryOpentelemetry, 'openTelemetrySetupCheck').mockImplementation(() => {
return [];
});

validateOpenTelemetrySetup();

// Without tracing, this is expected only twice
expect(errorSpy).toHaveBeenCalledTimes(2);
expect(warnSpy).toHaveBeenCalledTimes(1);

expect(errorSpy).toBeCalledWith(expect.stringContaining('You have to set up the SentryContextManager.'));
expect(errorSpy).toBeCalledWith(expect.stringContaining('You have to set up the SentryPropagator.'));
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

l: I think it would be good to also assert on the message we don't expect to show up as well, just relying on the number of times it has been called is error prone.

expect(warnSpy).toBeCalledWith(expect.stringContaining('You have to set up the SentrySampler.'));
});

it('works with missing setup, with tracing', () => {
const errorSpy = jest.spyOn(logger, 'error').mockImplementation(() => {});
const warnSpy = jest.spyOn(logger, 'warn').mockImplementation(() => {});

jest.spyOn(SentryOpentelemetry, 'openTelemetrySetupCheck').mockImplementation(() => {
return [];
});

init({ dsn: PUBLIC_DSN, skipOpenTelemetrySetup: true, tracesSampleRate: 1 });

validateOpenTelemetrySetup();

expect(errorSpy).toHaveBeenCalledTimes(3);
expect(warnSpy).toHaveBeenCalledTimes(1);

expect(errorSpy).toBeCalledWith(expect.stringContaining('You have to set up the SentryContextManager.'));
expect(errorSpy).toBeCalledWith(expect.stringContaining('You have to set up the SentryPropagator.'));
expect(errorSpy).toBeCalledWith(expect.stringContaining('You have to set up the SentrySpanProcessor.'));
expect(warnSpy).toBeCalledWith(expect.stringContaining('You have to set up the SentrySampler.'));
});
});
Loading