fix tests

Author: Lms24

packages/sveltekit/src/server/load.ts

@@ -1,5 +1,5 @@
 /* eslint-disable @sentry-internal/sdk/no-optional-chaining */
-import { getCurrentHub, startSpan, trace } from '@sentry/core';
+import { flush, getCurrentHub, startSpan, trace } from '@sentry/core';
 import { captureException } from '@sentry/node';
 import type { TransactionContext } from '@sentry/types';
 import { addExceptionMechanism, addNonEnumerableProperty, objectify } from '@sentry/utils';
@@ -153,12 +153,13 @@ export function wrapServerLoadWithSentry<T extends (...args: any) => any>(origSe
       };
 
       try {
-        const serverLoadResult = startSpan(traceLoadContext, () => wrappingTarget.apply(thisArg, args));
+        const serverLoadResult = await startSpan(traceLoadContext, () => wrappingTarget.apply(thisArg, args));
         return serverLoadResult;
       } catch (e: unknown) {
         sendErrorToSentry(e);
         throw e;
       } finally {
+        await flush(4000);
         await flushIfServerless();
       }
     },

packages/sveltekit/test/server/handle.test.ts

@@ -23,7 +23,7 @@ vi.mock('@sentry/node', async () => {
   };
 });
 
-const mockAddExceptionMechanism = vi.fn(() => console.trace());
+const mockAddExceptionMechanism = vi.fn();
 
 vi.mock('@sentry/utils', async () => {
   const original = (await vi.importActual('@sentry/utils')) as any;

packages/sveltekit/test/server/load.test.ts

@@ -134,8 +134,6 @@ afterEach(() => {
   mockAddExceptionMechanism.mockClear();
   mockStartSpan.mockClear();
   mockScope = new Scope();
-  // @ts-expect-error - this is fine (just tests here)
-  mockScope.mockClear?.call(mockScope);
 });
 
 describe.each([
@@ -195,12 +193,12 @@ describe.each([
     expect(mockCaptureException).not.toHaveBeenCalled();
   });
 
-  vi.spyOn(mockScope, 'addEventProcessor').mockImplementationOnce(callback => {
-    void callback({}, { event_id: 'fake-event-id' });
-    return mockScope;
-  });
+  it('adds an exception mechanism', async () => {
+    vi.spyOn(mockScope, 'addEventProcessor').mockImplementationOnce(callback => {
+      void callback({}, { event_id: 'fake-event-id' });
+      return mockScope;
+    });
 
-  it.only('adds an exception mechanism', async () => {
     async function load({ params }) {
       return {
         post: getById(params.id),
@@ -212,7 +210,7 @@ describe.each([
     await expect(res).rejects.toThrow();
 
     // expect(addEventProcessorSpy).toHaveBeenCalledTimes(1);
-    expect(mockAddExceptionMechanism).toHaveBeenCalledTimes(1);
+    expect(mockAddExceptionMechanism).toBeCalledTimes(1);
     expect(mockAddExceptionMechanism).toBeCalledWith(
       {},
       { handled: false, type: 'sveltekit', data: { function: 'load' } },