Merge pull request #18 from shuvalov-anton/fix-multi-header

Fix setting multiple headers at once.
A · Oct 7, 2015 · 36ea0c6 · 36ea0c6
2 parents 7850897 + d09f07c
commit 36ea0c6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -84,8 +84,8 @@ function mock(superagent) {
     }
     // Recursively set keys if passed an object
     if (isObject(key)) {
-      for (var key in field) {
-        this.set(key, field[key]);
+      for (var field in key) {
+        this.set(field, key[field]);
       }
       return this;
     }

diff --git a/test.js b/test.js
@@ -168,6 +168,22 @@ describe('superagent mock', function() {
       ;
     });
 
+    it('should support multiple headers', function(done) {
+      mock.get('/', function(req) {
+        return req.headers;
+      });
+      request.get('/')
+        .set({
+          'My-Header': 'my-Value',
+          'User-Agent': 'Opera Mini',
+        })
+        .end(function(_, data) {
+          data.should.have.property('my-header', 'my-Value')
+          data.should.have.property('user-agent', 'Opera Mini')
+          done();
+        })
+    })
+
     it('should throw error when header isn\'t a string', function() {
       mock.get('/topics/:id', function(req) {
         return req.headers;