cleanup

Author: Lms24

packages/sveltekit/test/client/load.test.ts

@@ -259,7 +259,7 @@ describe('wrapLoadWithSentry', () => {
         };
       };
 
-      it.only('creates a fetch span and attaches tracing headers if event.fetch was called', async () => {
+      it('creates a fetch span and attaches tracing headers if event.fetch was called', async () => {
         const wrappedLoad = wrapLoadWithSentry(load);
         await wrappedLoad(MOCK_LOAD_ARGS);
 
@@ -279,30 +279,19 @@ describe('wrapLoadWithSentry', () => {
         );
         expect(mockTrace).toHaveBeenNthCalledWith(2, spanCtx, expect.any(Function));
 
-        // TODO: refactor
-        const expectedFetchArgs =
-          originalFetchArgs.length > 1
-            ? [
-                originalFetchArgs[0],
-                {
-                  ...(originalFetchArgs[1] as RequestInit),
-                  headers: {
-                    // @ts-ignore that's fine
-                    ...(originalFetchArgs[1].headers as RequestInit['headers']),
-                    baggage: expect.any(String),
-                    'sentry-trace': expect.any(String),
-                  },
-                },
-              ]
-            : [
-                originalFetchArgs[0],
-                {
-                  headers: {
-                    baggage: expect.any(String),
-                    'sentry-trace': expect.any(String),
-                  },
-                },
-              ];
+        const hasSecondArg = originalFetchArgs.length > 1;
+        const expectedFetchArgs = [
+          originalFetchArgs[0],
+          {
+            ...(hasSecondArg && (originalFetchArgs[1] as RequestInit)),
+            headers: {
+              // @ts-ignore that's fine
+              ...(hasSecondArg && (originalFetchArgs[1].headers as RequestInit['headers'])),
+              baggage: expect.any(String),
+              'sentry-trace': expect.any(String),
+            },
+          },
+        ];
 
         expect(mockedSveltekitFetch).toHaveBeenCalledWith(...expectedFetchArgs);
       });