Adding router

Author: geek

lib/index.js

@@ -0,0 +1,30 @@
+var Diamonator = function (server) {
+
+    var self = this;
+
+    this._routes = {};
+    server.on('request', function (req, res) {
+
+        var route = self._route(req.method, req.path);
+        if (route) {
+            return route(req, res);
+        }
+
+        res.writeHead(404);
+
+    });
+};
+
+Diamonator.prototype.register = function (route, callback) {
+
+    this._routes[route.method] = this._routes[route.method] || {};
+    this._routes[route.method][route.path] = callback;
+};
+
+
+Diamonator.prototype._route = function (method, path) {
+
+    return this._routes[method][path];
+};
+
+module.exports = Diamonator;

test/helper.js

@@ -14,13 +14,16 @@ exports.createServer = function () {
 
         var res = new Stream.PassThrough();
         res.result = '';
-        res.on('readable', function (chunk) {
+        res.on('readable', function () {
 
-            res.result += chunk.toString();
+            res.result += res.read().toString();
         });
 
         server.emit('request', req, res);
-        res.once('finish', callback);
+        res.once('end', function () {
+
+            callback(res);
+        });
     };
 
     return server;

test/index.js

@@ -20,12 +20,12 @@ describe('register routes', function () {
         var diamonator = new Diamonator(server);
         diamonator.register({ method: 'GET', path: '/' }, function (req, res) {
 
-            res.end('OK');
+            res.write('OK');
+            res.end();
         });
 
         server.inject('GET', '/', function (res) {
 
-            expect(res.statusCode).to.equal(200);
             expect(res.result).to.equal('OK');
             done();
         });