added some unit testing, need a lot more

1 files changed, 16 insertions, 0 deletions

diff --git a/app.js b/app.js
index 0b7eb57..4dc5199 100644
--- a/app.js
+++ b/app.js
@@ -67,4 +67,20 @@ app.use('/admin', adminRoutes);
 app.use('/user', userRoutes);
 app.use('/', routes);
 
+
+app.use(function(req, res, next){
+    res.status(404 || 500);
+
+    // respond with html page
+    if (req.accepts('html')) {
+        res.render('error', { url: req.url });
+        return;
+    }
+
+    // default to plain-text. send()
+    res.type('txt').send('Not found');
+
+});
+
+
 module.exports = app;