ignore invalid headers in mockResponseHeaders

APIDevTools · Mar 4, 2016 · dc67f0b · dc67f0b
1 parent 2fe55a5
commit dc67f0b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/mock/index.js b/lib/mock/index.js
@@ -145,6 +145,10 @@ function mockResponseHeaders(req, res, next) {
 
     if (res.swagger.headers) {
       _.forEach(res.swagger.headers, function(header, name) {
+        if (!header) {
+          return;
+        }
+
         // Set all HTTP headers that are defined in the Swagger API.
         // If a default value is specified in the Swagger API, then use it; otherwise generate a value.
         if (res.get(name) !== undefined) {
@@ -155,6 +159,10 @@ function mockResponseHeaders(req, res, next) {
           util.debug('    %s: %s (default)', name, header.default);
         }
         else {
+          if (!name || !_.isFunction(name.toLowerCase)) {
+            return;
+          }
+
           switch (name.toLowerCase()) {
             case 'location':
               res.location(req.baseUrl + (res.swagger.location || req.path));