diff --git a/package.json b/package.json index c0404c1..aed686d 100644 --- a/package.json +++ b/package.json @@ -12,6 +12,7 @@ "index.js" ], "devDependencies": { + "buffer-equal": "^1.0.0", "istanbul": "^0.4.2", "koa": "^1.0.0", "mocha": "^2.0.0", diff --git a/test/index.js b/test/index.js index 2f17cd7..04f9a92 100644 --- a/test/index.js +++ b/test/index.js @@ -1,3 +1,4 @@ +var bufferEqual = require('buffer-equal'); var request = require('supertest'); var join = require('path').join; var assert = require('assert'); @@ -14,8 +15,8 @@ describe('favicon()', function(){ app.use(favicon(path)); app.use(function *(next){ - assert(this.body == null); - assert(this.get('Content-Type') == null); + assert(!this.body); + assert(!this.get('Content-Type')); this.body = 'hello'; }); @@ -50,9 +51,13 @@ describe('favicon()', function(){ request(app.listen()) .get('/favicon.ico') - .expect(200) .expect('Content-Type', 'image/x-icon') - .expect(body.toString(), done); + .expect(200, function(err, res){ + if (err) return done(err); + + assert(bufferEqual(body, res.body)); + done(); + }); }); it('should set cache-control headers', function(done){