added headers method

caolan · Jun 28, 2010 · 9cf7d30 · 9cf7d30
1 parent c4e8d94
commit 9cf7d30
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/lib/quickresponse.js b/lib/quickresponse.js
@@ -48,6 +48,12 @@ module.exports = function(){
             return res;
         };
 
+        // add headers
+        res.headers = function(headers){
+            for(var k in headers) res._headers[k] = headers[k];
+            return res;
+        };
+
         // success
         res.ok = withStatus(200);
         res.created = withStatus(201);

diff --git a/test/test-response.js b/test/test-response.js
@@ -1,5 +1,4 @@
 // TODO: add support for following styles:
-// req.text().notFound('not found');
 // req.headers({some:header}).ok('done');
 
 var quickresponse = require('quickresponse');
@@ -183,3 +182,23 @@ exports['send defaults'] = function(test){
     });
     test.done();
 };
+
+exports.headers = function(test){
+    test.expect(3);
+    var res = {};
+    quickresponse()(null, res, function(){
+        test.equals(res.headers({some:'header',test:'test'}), res);
+        test.same(res._headers, {
+            'Content-Type':'text/html',
+            some:'header',
+            test:'test'
+        });
+        res.headers({'Content-Type':'test'});
+        test.same(res._headers, {
+            'Content-Type':'test',
+            some:'header',
+            test:'test'
+        });
+    });
+    test.done();
+};