Fix return type of wrapper

Author: baileytincher

src/index.ts

@@ -45,20 +45,14 @@ export type TShallotSocketEvent<
   body?: TBody;
 };
 
-type TShallotSocketHandler<TEvent extends TShallotSocketEvent = TShallotSocketEvent> = (
+const ShallotSocketWrapper = <TEvent extends TShallotSocketEvent = TShallotSocketEvent>(
   handler: ShallotRawHandler<TEvent>,
-  successStatusCode?: number,
-  middlewareOpts?: {
+  successStatusCode = 200,
+  middlewareOpts: {
     HttpErrorHandlerOpts?: TShallotErrorHandlerOptions;
     HttpJsonBodyParserOpts?: TShallotJSONBodyParserOptions;
-  }
-) => ShallotAWSHandler<TEvent>;
-
-const ShallotSocketWrapper: TShallotSocketHandler = (
-  handler,
-  successStatusCode = 200,
-  middlewareOpts = {}
-) => {
+  } = {}
+): ShallotAWSHandler<TEvent> => {
   const wrappedResponseHandler: Handler = async (...args) => {
     const res = await handler(...args);
     return {

tests/index.test.ts

@@ -21,13 +21,13 @@ describe('WebSocket Wrapper', () => {
     succeed: () => undefined,
   };
 
-  const mockHandler: ShallotRawHandler<TShallotSocketEvent<{ test: string }>> =
-    async () => undefined;
+  type MockEvent = TShallotSocketEvent<{ test: string }>;
+  const mockHandler: ShallotRawHandler<MockEvent> = async () => undefined;
 
   test('Smoke test CORS default usage', async () => {
     const wrappedHandler = ShallotAWSSocketWrapper(mockHandler);
 
-    const mockEvent: TShallotSocketEvent = {} as unknown as TShallotSocketEvent;
+    const mockEvent: MockEvent = {} as unknown as MockEvent;
     await wrappedHandler(mockEvent, mockContext, jest.fn());
   });
 });