Add authenticate.js tests and improve functions

package.json

@@ -51,6 +51,7 @@
     "mocha": "^3.2.0",
     "react": "^15.4.2",
     "react-dom": "^15.4.2",
+    "supertest": "^2.0.1",
     "webpack": "^1.14.0"
   }
 }

public/views/chatRoom/components/chatRoom.react.js

@@ -78,7 +78,7 @@ class ChatRoom extends React.Component {
 
     this.setState({ chats });
   }
-  
+
   scrollBottom() {
     const messageArea = document.getElementsByClassName('messages-container');
     messageArea[0].scrollTop = messageArea[0].scrollHeight;

server/server.js

@@ -19,7 +19,7 @@ app.use((req, res) => {
 server.listen(process.env.PORT);
 
 
-module.exports = { io };
+module.exports = { app, io };
 
 // run bot
 require('./bot/bot');

server/tests/web/authenticate.test.js

@@ -1,14 +1,101 @@
+/* eslint no-undef:off */
+
+const expect = require('expect');
+const request = require('supertest');
+const { app } = require('../../server');
+const { User } = require('../../db/model/user');
+
 describe('authenticate', () => {
+  describe('GET /views/chatRoom', () => {
+    before(() => {
+      token = '123';
+    });
+
+    beforeEach(() => {
+      findUserSpy = expect.spyOn(User, 'findUserByToken');
+    });
+
+    afterEach(() => {
+      findUserSpy.restore();
+    });
+
+    it('should check token cookie', (done) => {
+      request(app)
+      .get('/views')
+      .set('Cookie', `token=${token}`)
+      .end((err) => {
+        expect(findUserSpy).toHaveBeenCalled();
+        expect(findUserSpy.calls[0].arguments).toEqual(token);
+        done(err);
+      });
+    });
+
+    it('should redirect 401 to /views/login if no token cookie found', (done) => {
+      findUserSpy.reset();
+
+      request(app)
+      .get('/views')
+      .expect(401)
+      .expect('x-authorization', 'No token cookie')
+      .expect('location', '/views/login')
+      .end((err) => {
+        expect(findUserSpy).toNotHaveBeenCalled();
+        done(err);
+      });
+    });
+
+    it('should redirect 401 to /views/login if no user found', (done) => {
+      findUserSpy.andReturn(new Promise(resolve => resolve()));
+
+      request(app)
+      .get('/views')
+      .set('Cookie', `token=${token}`)
+      .expect(401)
+      .expect('x-authorization', 'No user found')
+      .expect('location', '/views/login')
+      .end((err) => {
+        done(err);
+      });
+    });
+
+    it('should redirect 401 to /views/login if token expired', (done) => {
+      findUserSpy.andReturn(new Promise((resolve) => {
+        resolve(
+          { expiration_date: new Date().getTime() - 9999999 }
+        );
+      }));
+
+      request(app)
+      .get('/views')
+      .set('Cookie', `token=${token}`)
+      .expect(401)
+      .expect('x-authorization', 'Token expired')
+      .expect('location', '/views/login')
+      .end((err) => {
+        expect(findUserSpy).toHaveBeenCalled();
+        done(err);
+      });
+    });
+
+    it('should should call next()', (done) => {
+      findUserSpy.andReturn(new Promise((resolve) => {
+        resolve(
+          { expiration_date: new Date().getTime() + 99999999 }
+        );
+      }));
 
-  describe('auth GET /views/chatRoom', () => {
-    it('should check token cookie');
-    it('should redirect 401 to /views/login if no token cookie found');
-    it('should redirect 401 to /views/login if no user found');
-    it('should redirect 401 to /views/login if token expired');
-    it('should should call next()');
+      request(app)
+      .get('/views')
+      .set('Cookie', `token=${token}`)
+      .expect(301)
+      .end((err) => {
+        done(err);
+      });
+    });
   });
 
   describe('auth POST /sendcode', () => {
+    // better to separate each reaquest on it's own function?
     it('should create user if not registered');
     it('should set token cookie');
   });

server/web/authenticate.js

@@ -16,12 +16,12 @@ auth.use(cookieParser());
 
 // Check if user can access chatRoom on request
 auth.get('/views', (req, res, next) => {
+
   // Check token cookie
   if (req.cookies && req.cookies.token) {
 
     // Check token cookie value on DB
     User.findUserByToken(req.cookies.token).then((result) => {
-
       if (result) {
         const date = new Date().getTime();
         const expiration = new Date(result.expiration_date).getTime();
@@ -30,19 +30,24 @@ auth.get('/views', (req, res, next) => {
         if (date < expiration) {
           return next();
         }
-        console.log('Token expired');
-      }
+        res.set('x-authorization', 'Token expired')
+        .redirect(401, '/views/login');
+
 
-      // No user found or token expired
-      console.log('No user found');
-      res.redirect(401, '/views/login');
+      } else {
+        // No user found or token expired
+        res.set('x-authorization', 'No user found')
+        .redirect(401, '/views/login');
+
+      }
     });
 
   } else {
     // No cookie token
-    console.log('No cookie token');
-    res.redirect(401, '/views/login');
+    res.set('x-authorization', 'No token cookie')
+    .redirect(401, '/views/login');
   }
+
 });
 
 // Route for Facebook authentication handlers