flow working with tests - TODO: finalize API

src/extras/StatusError.ts

@@ -0,0 +1,9 @@
+export class StatusError extends Error {
+  status: number
+
+  constructor(status: number = 500, message: string = 'Internal Error.') {
+    super(message)
+    this.name = 'StatusError'
+    this.status = status
+  }
+}

src/flow.spec.ts

@@ -64,9 +64,9 @@ describe('flow(router: RouterType, options: FlowOptions): RequestHandler', () =>
 
     describe('format?: Function | false', () => {
       it('should handle custom format function', async () => {
-        const customFormat = (data) => ({ status: 200, body: `num items = ${data.length}` })
+        const customFormat = (data) => ({ status: 201, body: `num items = ${data.length}` })
         let response = await flow(router, { format: customFormat })(request('/items'))
-        expect(response.status).toBe(200)
+        expect(response.status).toBe(201)
         expect(response.body).toBe('num items = 3')
       })
 
@@ -76,20 +76,6 @@ describe('flow(router: RouterType, options: FlowOptions): RequestHandler', () =>
       })
     })
 
-    describe('notFound?: RouteHandler | false', () => {
-      it('should handle custom notFound function', async () => {
-        const customNotFound = () => ({ status: 404, body: 'Custom Not Found' })
-        let response = await flow(router, { notFound: customNotFound })(request('/missing'))
-        expect(response.status).toBe(404)
-        expect(response.body).toBe('Custom Not Found')
-      })
-
-      it('should not handle notFound if set to false', async () => {
-        let response = await flow(router, { notFound: false })(request('/missing'))
-        expect(response.status).toBeUndefined()
-      })
-    })
-
     describe('cors?: CorsOptions | true', () => {
       it('will embed CORS headers if provided', async () => {
         let response = await flow(router, {
@@ -155,7 +141,7 @@ describe('flow(router: RouterType, options: FlowOptions): RequestHandler', () =>
       it('if set to false, will not add notFound handler (allow undefined passthrough)', async () => {
         let response = await flow(router, {
           notFound: false,
-          format: false,
+          // format: false,
         })(request('/missing'))
 
         expect(response).toBe(undefined)

src/flow.ts

@@ -34,8 +34,7 @@ export const flow = (router: RouterType, options: FlowOptions = {}) => {
   return async (...args: any[]) => {
     // @ts-expect-error
     let response = router.handle(...args)
-    // @ts-expect-error - add optional formatting
-    response = format ? response.then(format) : response
+    response = format ? response.then(v => v === undefined ? v : format(v)) : response
     // @ts-expect-error - add optional error handling
     response = errors ? response.catch(errors) : response