add and fix tests

Author: lobsterkatie

packages/nextjs/test/config/wrappers.test.ts

@@ -0,0 +1,70 @@
+import * as SentryCore from '@sentry/core';
+import * as SentryTracing from '@sentry/tracing';
+import { IncomingMessage, ServerResponse } from 'http';
+
+import {
+  withSentryGetServerSideProps,
+  withSentryServerSideGetInitialProps,
+  // TODO: Leaving `withSentryGetStaticProps` out for now until we figure out what to do with it
+  // withSentryGetStaticProps,
+  // TODO: Leaving these out for now until we figure out pages with no data fetchers
+  // withSentryServerSideAppGetInitialProps,
+  // withSentryServerSideDocumentGetInitialProps,
+  // withSentryServerSideErrorGetInitialProps,
+} from '../../src/config/wrappers';
+
+const startTransactionSpy = jest.spyOn(SentryCore, 'startTransaction');
+const setMetadataSpy = jest.spyOn(SentryTracing.Transaction.prototype, 'setMetadata');
+
+describe('data-fetching function wrappers', () => {
+  const route = '/tricks/[trickName]';
+  let req: IncomingMessage;
+  let res: ServerResponse;
+
+  describe('starts a transaction if tracing enabled', () => {
+    beforeEach(() => {
+      req = { headers: {}, url: 'http://dogs.are.great/tricks/kangaroo' } as IncomingMessage;
+      res = {} as ServerResponse;
+
+      jest.spyOn(SentryTracing, 'hasTracingEnabled').mockReturnValueOnce(true);
+    });
+
+    afterEach(() => {
+      jest.clearAllMocks();
+    });
+
+    test('withSentryGetServerSideProps', async () => {
+      const origFunction = jest.fn(async () => ({ props: {} }));
+
+      const wrappedOriginal = withSentryGetServerSideProps(origFunction, route);
+      await wrappedOriginal({ req, res } as any);
+
+      expect(startTransactionSpy).toHaveBeenCalledWith(
+        expect.objectContaining({
+          name: '/tricks/[trickName]',
+          op: 'nextjs.data.server',
+          metadata: expect.objectContaining({ source: 'route' }),
+        }),
+      );
+
+      expect(setMetadataSpy).toHaveBeenCalledWith({ request: req });
+    });
+
+    test('withSentryServerSideGetInitialProps', async () => {
+      const origFunction = jest.fn(async () => ({}));
+
+      const wrappedOriginal = withSentryServerSideGetInitialProps(origFunction);
+      await wrappedOriginal({ req, res, pathname: route } as any);
+
+      expect(startTransactionSpy).toHaveBeenCalledWith(
+        expect.objectContaining({
+          name: '/tricks/[trickName]',
+          op: 'nextjs.data.server',
+          metadata: expect.objectContaining({ source: 'route' }),
+        }),
+      );
+
+      expect(setMetadataSpy).toHaveBeenCalledWith({ request: req });
+    });
+  });
+});

packages/nextjs/test/index.server.test.ts

@@ -150,8 +150,10 @@ describe('Server init()', () => {
 
       const nodeInitOptions = nodeInit.mock.calls[0][0] as ModifiedInitOptions;
       const rewriteFramesIntegration = findIntegrationByName(nodeInitOptions.integrations, 'RewriteFrames');
+      const requestDataFramesIntegration = findIntegrationByName(nodeInitOptions.integrations, 'RequestData');
 
       expect(rewriteFramesIntegration).toBeDefined();
+      expect(requestDataFramesIntegration).toBeDefined();
     });
 
     it('supports passing unrelated integrations through options', () => {

packages/nextjs/test/utils/withSentry.test.ts

@@ -119,6 +119,7 @@ describe('withSentry', () => {
           metadata: {
             baggage: expect.any(Array),
             source: 'route',
+            request: expect.objectContaining({ url: 'http://dogs.are.great' }),
           },
         },
         { request: expect.objectContaining({ url: 'http://dogs.are.great' }) },