Fix failing tests

index.d.ts

@@ -1,5 +1,5 @@
 import { Server } from 'http';
-import { CloudEventFunction, HTTPFunction, InvokerOptions } from './lib/types';
+import { CloudEventFunction, HTTPFunction, InvokerOptions, Function } from './lib/types';
 import { LogLevel } from 'fastify';
 
 // Invokable describes the function signature for a function that can be invoked by the server.
@@ -21,7 +21,7 @@ export declare const defaults: {
   LOG_LEVEL: LogLevel,
   PORT: number,
   INCLUDE_RAW: boolean,
-}
+};
 
 // re-export
 export * from './lib/types';

lib/types.d.ts

@@ -9,12 +9,12 @@ import { Http2ServerRequest, Http2ServerResponse } from 'http2';
  */
 export interface Function {
   // The initialization function, called before the server is started
-  // This function is optional and should be synchronous.
-  init?: () => any | Promise<any>;
+  // This function is optional.
+  init?: () => (any | Promise<any>);
 
   // The shutdown function, called after the server is stopped
-  // This function is optional and should be synchronous.
-  shutdown?: () => any | Promise<any>;
+  // This function is optional.
+  shutdown?: () => (any | Promise<any>);
 
   // The liveness function, called to check if the server is alive
   // This function is optional and should return 200/OK if the server is alive.

test/test-lifecycle.js

@@ -34,10 +34,9 @@ test('Calls async init before the server has started', async t => {
   let initCalled = false;
   const server = await start(
     {
-      init: () =>
-        new Promise(() => {
-          initCalled = true;
-        }),
+      init: async() => {
+        initCalled = true;
+      },
       handle: () => {},
     },
     defaults
@@ -88,10 +87,9 @@ test('Calls async shutdown after the server has stopped', async t => {
   const server = await start(
     {
       handle: () => {},
-      shutdown: _ =>
-        new Promise(_ => {
-          shutdownCalled = true;
-        }),
+      shutdown: async() => {
+        shutdownCalled = true;
+      },
     },
     defaults
   );