add tests

Author: amirbey

index.js

@@ -0,0 +1,6 @@
+const app = require('./server');
+const PORT = process.env.PORT || 3002;
+
+app.listen(PORT, () => {
+  console.log(`Server running on port ${PORT}`);
+});

package.json

@@ -5,7 +5,8 @@
   "main": "server.js",
   "scripts": {
     "start": "node server.js",
-    "dev": "nodemon server.js"
+    "dev": "nodemon server.js",
+    "test": "NODE_ENV=test mocha"
   },
   "dependencies": {
     "axios": "^1.9.0",
@@ -15,7 +16,11 @@
     "redis": "^5.5.6"
   },
   "devDependencies": {
-    "nodemon": "^3.0.0"
+    "chai": "^5.2.1",
+    "mocha": "^11.7.1",
+    "nodemon": "^3.0.0",
+    "proxyquire": "^2.1.3",
+    "supertest": "^7.1.4"
   },
   "private": true,
   "license": "CC0-1.0"

server.js

@@ -3,29 +3,26 @@ require('dotenv').config();
 const express = require('express');
 const app = express();
 
-// Import the id_plus route
 const idPlusRoute = require('./routes/id_plus');
 const docvRoute = require('./routes/docv');
 const dosRoute = require('./routes/dos');
 
 app.use(express.json());
-
-// Mount the /id_plus route
 app.use('/api/3.0/EmailAuthScore', idPlusRoute);
-// Mount the /docv route
 app.use('/docv', docvRoute);
-// Mount the /dos route
 app.use('/dos', dosRoute);
 
-if (process.env.NODE_ENV === 'production') {
-  // Global error handler (add this before app.listen)
+if (process.env.NODE_ENV === 'test') {
+  app.get('/error', (req, res, next) => {
+    next(new Error('Test error'));
+  });
+}
+
+if (process.env.NODE_ENV === 'production' || process.env.NODE_ENV === 'test') {
   app.use((err, req, res, next) => {
     console.error(err.stack);
     res.status(500).json({ error: 'Internal Server Error' });
   });
 }
 
-const PORT = process.env.PORT;
-app.listen(PORT, () => {
-  console.log(`Server running on port ${PORT}`);
-});
+module.exports = app;

test/server.test.js

@@ -0,0 +1,51 @@
+const proxyquire = require('proxyquire').noCallThru();
+const request = require('supertest');
+const { expect } = require('chai');
+const express = require('express');
+
+describe('Server', function () {
+  let app;
+  let server;
+
+  // Mock route handlers
+  const idPlusRoute = express.Router().get('/', (req, res) => res.json({ route: 'id_plus' }));
+  const docvRoute = express.Router().get('/', (req, res) => res.json({ route: 'docv' }));
+  const dosRoute = express.Router().get('/', (req, res) => res.json({ route: 'dos' }));
+
+  before(function () {
+    // Proxyquire to inject mocks
+    app = proxyquire('../server', {
+      './routes/id_plus': idPlusRoute,
+      './routes/docv': docvRoute,
+      './routes/dos': dosRoute,
+      dotenv: { config: () => {} }
+    });
+
+    server = app;
+  });
+
+  it('should mount /api/3.0/EmailAuthScore', async function () {
+    const res = await request(server).get('/api/3.0/EmailAuthScore');
+    expect(res.status).to.equal(200);
+    expect(res.body.route).to.equal('id_plus');
+  });
+
+  it('should mount /docv', async function () {
+    const res = await request(server).get('/docv');
+    expect(res.status).to.equal(200);
+    expect(res.body.route).to.equal('docv');
+  });
+
+  it('should mount /dos', async function () {
+    const res = await request(server).get('/dos');
+    expect(res.status).to.equal(200);
+    expect(res.body.route).to.equal('dos');
+  });
+
+  it('should handle errors globally in production', async function () {
+    server.get('/error', (req, res, next) => { next(new Error('Test error')); });
+    const res = await request(server).get('/error');
+    expect(res.status).to.equal(500);
+    expect(res.body).to.have.property('error', 'Internal Server Error');
+  });
+});