Merge pull request #21 from mtkopone/test-restify

Add test for restify

package.json

@@ -6,6 +6,7 @@
   "dependencies": {},
   "devDependencies": {
     "express": "*",
+    "restify": "*",
     "keygrip": "*"
   },
   "engines": {
@@ -18,7 +19,7 @@
     "url": "git://github.com/jed/cookies.git"
   },
   "scripts": {
-    "test": "node ./test/http.js && node ./test/express.js"
+    "test": "node ./test/http.js && node ./test/express.js && node ./test/restify.js"
   },
   "optionalDependencies": {}
 }

test/restify.js

@@ -0,0 +1,69 @@
+var assert = require('assert'),
+    restify = require('restify'),
+    keys = require('keygrip')(),
+    http = require('http'),
+    Cookies = require('../')
+
+var server = restify.createServer()
+
+server.get('/set', function (req, res) {
+  setCookies(req, res)
+  res.json({ status : 'ok'})
+})
+
+server.get('/get', function (req, res) {
+  assertCookies(req, res)
+  res.send(200)
+})
+
+server.listen(8000, function() {
+  http.get({ path: '/set', host: 'localhost', port: 8000 }, function(res) {
+    assert.equal(res.statusCode, 200)
+    var header = res.headers['set-cookie']
+    assertSetCookieHeader(header)
+    http.get({ path: '/get', host: 'localhost', port: 8000, headers: { 'Cookie': header.join(';') } }, function(res) {
+      assert.equal(res.statusCode, 200)
+      server.close()
+    })
+  })
+})
+
+function setCookies(req, res) {
+  var cookies = new Cookies(req, res, keys)
+  cookies
+    .set('unsigned', 'foo', { signed:false, httpOnly: false })
+    .set('signed', 'bar', { signed: true })
+    .set('tampered', 'baz')
+    .set('tampered.sig', 'bogus')
+    .set('overwrite', 'old-value', { signed: true })
+    .set('overwrite', 'new-value', { overwrite: true, signed: true })
+}
+
+function assertCookies(req, res) {
+  var cookies = new Cookies(req, res, keys)
+  var unsigned = cookies.get('unsigned'),
+    signed = cookies.get('signed', { signed: true }),
+    tampered = cookies.get('tampered', { signed: true }),
+    overwrite = cookies.get('overwrite', { signed: true })
+
+  assert.equal(unsigned, 'foo')
+  assert.equal(cookies.get('unsigned.sig', { signed:false }), undefined)
+  assert.equal(signed, 'bar')
+  assert.equal(cookies.get('signed.sig', { signed: false }), keys.sign('signed=bar'))
+  assert.notEqual(tampered, 'baz')
+  assert.equal(tampered, undefined)
+  assert.equal(overwrite, 'new-value')
+  assert.equal(cookies.get('overwrite.sig', { signed:false }), keys.sign('overwrite=new-value'))
+  assert.equal(res.getHeader('Set-Cookie'), 'tampered.sig=; path=/; expires=Thu, 01 Jan 1970 00:00:00 GMT; httponly')
+}
+
+function assertSetCookieHeader(header) {
+  assert.equal(header.length, 7)
+  assert.equal(header[0], 'unsigned=foo; path=/')
+  assert.equal(header[1], 'signed=bar; path=/; httponly')
+  assert.ok(/^signed\.sig=.{27}; path=\/; httponly$/.test(header[2]))
+  assert.equal(header[3], 'tampered=baz; path=/; httponly')
+  assert.equal(header[4], 'tampered.sig=bogus; path=/; httponly')
+  assert.equal(header[5], 'overwrite=new-value; path=/; httponly')
+  assert.ok(/^overwrite\.sig=.{27}; path=\/; httponly$/.test(header[6]))
+}