Add basic tests.

onurtemizkan · web-flow · commit 1e7aed9de99e · 2021-09-03T12:14:09.000Z
diff --git a/packages/tracing/test/integrations/apollo.test.ts b/packages/tracing/test/integrations/apollo.test.ts
@@ -0,0 +1,105 @@
+/* eslint-disable @typescript-eslint/unbound-method */
+import { Hub, Scope } from '@sentry/hub';
+
+import { Apollo } from '../../src/integrations/apollo';
+import { Span } from '../../src/span';
+
+type ApolloResolverGroup = {
+  [key: string]: () => any;
+};
+
+type ApolloModelResolvers = {
+  [key: string]: ApolloResolverGroup;
+};
+
+class ApolloServerBase {
+  config: {
+    resolvers: ApolloModelResolvers[];
+  };
+
+  constructor() {
+    this.config = {
+      resolvers: [
+        {
+          Query: {
+            res_1(...args: unknown[]) {
+              return 'foo';
+            },
+          },
+          Mutation: {
+            res_2(...args: unknown[]) {
+              return 'bar';
+            },
+          },
+        },
+      ],
+    };
+
+    this.constructSchema();
+  }
+
+  public constructSchema(...args: unknown[]) {
+    return null;
+  }
+}
+
+// mock for 'graphql/execution/execution.js' package
+jest.mock('@sentry/utils', () => {
+  const actual = jest.requireActual('@sentry/utils');
+  return {
+    ...actual,
+    loadModule() {
+      return {
+        ApolloServerBase,
+      };
+    },
+  };
+});
+
+describe('setupOnce', () => {
+  let scope = new Scope();
+  let parentSpan: Span;
+  let childSpan: Span;
+  let ApolloServer: ApolloServerBase;
+
+  beforeAll(() => {
+    new Apollo().setupOnce(
+      () => undefined,
+      () => new Hub(undefined, scope),
+    );
+
+    ApolloServer = new ApolloServerBase();
+  });
+
+  beforeEach(() => {
+    scope = new Scope();
+    parentSpan = new Span();
+    childSpan = parentSpan.startChild();
+    jest.spyOn(scope, 'getSpan').mockReturnValueOnce(parentSpan);
+    jest.spyOn(scope, 'setSpan');
+    jest.spyOn(parentSpan, 'startChild').mockReturnValueOnce(childSpan);
+    jest.spyOn(childSpan, 'finish');
+  });
+
+  it(`should wrap a simple resolver`, () => {
+    ApolloServer.config.resolvers[0]?.['Query']?.['res_1']?.();
+    expect(scope.getSpan).toBeCalled();
+    expect(parentSpan.startChild).toBeCalledWith({
+      description: 'Query.res_1',
+      op: 'apollo',
+    });
+    expect(childSpan.finish).toBeCalled();
+    expect(scope.setSpan).toHaveBeenCalledTimes(2);
+  });
+
+  it(`should wrap another simple resolver`, () => {
+    ApolloServer.config.resolvers[0]?.['Mutation']?.['res_2']?.();
+    expect(scope.getSpan).toBeCalled();
+    expect(parentSpan.startChild).toBeCalledWith({
+      description: 'Mutation.res_2',
+      op: 'apollo',
+    });
+    expect(childSpan.finish).toBeCalled();
+    expect(scope.setSpan).toHaveBeenCalledTimes(2);
+  });
+});
diff --git a/packages/tracing/test/integrations/graphql.test.ts b/packages/tracing/test/integrations/graphql.test.ts
@@ -0,0 +1,56 @@
+/* eslint-disable @typescript-eslint/unbound-method */
+import { Hub, Scope } from '@sentry/hub';
+
+import { GraphQL } from '../../src/integrations/graphql';
+import { Span } from '../../src/span';
+
+const GQLExecute = {
+  execute() {
+    return Promise.resolve();
+  },
+};
+
+// mock for 'graphql/execution/execution.js' package
+jest.mock('@sentry/utils', () => {
+  const actual = jest.requireActual('@sentry/utils');
+  return {
+    ...actual,
+    loadModule() {
+      return GQLExecute;
+    },
+  };
+});
+
+describe('setupOnce', () => {
+  let scope = new Scope();
+  let parentSpan: Span;
+  let childSpan: Span;
+
+  beforeAll(() => {
+    new GraphQL().setupOnce(
+      () => undefined,
+      () => new Hub(undefined, scope),
+    );
+  });
+
+  beforeEach(() => {
+    scope = new Scope();
+    parentSpan = new Span();
+    childSpan = parentSpan.startChild();
+    jest.spyOn(scope, 'getSpan').mockReturnValueOnce(parentSpan);
+    jest.spyOn(scope, 'setSpan');
+    jest.spyOn(parentSpan, 'startChild').mockReturnValueOnce(childSpan);
+    jest.spyOn(childSpan, 'finish');
+  });
+
+  it(`should wrap execute method`, async () => {
+    await GQLExecute.execute();
+    expect(scope.getSpan).toBeCalled();
+    expect(parentSpan.startChild).toBeCalledWith({
+      description: 'execute',
+      op: 'graphql',
+    });
+    expect(childSpan.finish).toBeCalled();
+    expect(scope.setSpan).toHaveBeenCalledTimes(2);
+  });
+});
