Merge pull request #13 from openaddresses/auto-serialize

Automatic Object Serialization

Author: ingalls

lib/middleware.js

@@ -3,6 +3,7 @@ import Param from './param.js';
 import Ajv from 'ajv';
 import addFormats from 'ajv-formats';
 import { ValidationError } from 'express-json-validator-middleware';
+import Generic from '@openaddresses/batch-generic';
 
 const ajv = addFormats(new Ajv({
     allErrors: true,
@@ -99,6 +100,7 @@ export default class Middleware {
 
     /**
      * Express middleware to validate the response body
+     * and optionally serialize Generic object to JSON
      *
      * @param {Object} schema JSON Schema
      *
@@ -110,6 +112,10 @@ export default class Middleware {
         return function (req, res, next) {
             const json = res.json;
             res.json = function(obj) {
+                if (obj instanceof Generic && typeof obj.serialize === 'function') {
+                    obj = obj.serialize();
+                }
+
                 if (res.statusCode === null || res.statusCode === 200) {
                     validate(obj);
                 }