adding integration tests with supertest and testcontainers

anthonyhastings · Oct 22, 2023 · df62141 · df62141
1 parent 2dc14f8
commit df62141
Show file tree

Hide file tree

Showing 7 changed files with 2,958 additions and 107 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -0,0 +1,27 @@
+name: Tests
+
+on:
+  pull_request:
+  push:
+    branches:
+      - main
+
+jobs:
+  test:
+    name: Integration Tests
+    runs-on: ubuntu-latest
+    steps:
+      - name: Clone repository
+        uses: actions/checkout@v2
+
+      - name: Set Node.js version
+        uses: actions/setup-node@v3
+        with:
+          node-version-file: '.nvmrc'
+          cache: 'yarn'
+
+      - name: Install application dependencies
+        run: yarn install --frozen-lockfile --prefer-offline
+
+      - name: Run integration tests
+        run: yarn test
diff --git a/jest.config.mjs b/jest.config.mjs
@@ -0,0 +1,5 @@
+export default {
+  testEnvironment: 'node',
+  testTimeout: 20000,
+  verbose: true,
+};
diff --git a/package.json b/package.json
@@ -11,7 +11,8 @@
   "scripts": {
     "dev": "nodemon src/index.mjs",
     "dev:native": "node --watch src/index.mjs",
-    "start": "node src/index.mjs"
+    "start": "node src/index.mjs",
+    "test": "DEBUG=testcontainers* node --experimental-vm-modules $(yarn bin jest) --runInBand"
   },
   "dependencies": {
     "dotenv": "^16.3.1",
@@ -21,7 +22,10 @@
     "zod": "^3.22.4"
   },
   "devDependencies": {
+    "@testcontainers/mongodb": "^10.2.1",
+    "jest": "^29.7.0",
     "nodemon": "^3.0.1",
-    "prettier": "^3.0.3"
+    "prettier": "^3.0.3",
+    "supertest": "^6.3.3"
   }
 }
diff --git a/src/__tests__/integration.spec.js b/src/__tests__/integration.spec.js
@@ -0,0 +1,159 @@
+import path from 'node:path';
+import { jest } from '@jest/globals';
+import { MongoDBContainer } from '@testcontainers/mongodb';
+import mongoose from 'mongoose';
+import supertest from 'supertest';
+
+jest.unstable_mockModule('nanoid', async () => {
+  return {
+    nanoid: jest.fn(() => 'fake-id'),
+  };
+});
+
+const nanoId = (await import('nanoid')).nanoid;
+
+describe('App Tests', () => {
+  let mongoDBContainer;
+
+  beforeAll(async () => {
+    mongoDBContainer = await new MongoDBContainer('mongo:7')
+      .withCopyDirectoriesToContainer([
+        {
+          source: path.resolve(process.cwd(), './database/init/'),
+          target: '/docker-entrypoint-initdb.d/',
+        },
+      ])
+      .start();
+
+    // Mongoose is using the docker hostname. This only works if the test code is running inside of docker.
+    // ?directConnection solves this connection issue.
+    process.env.MONGODB_URL = `${mongoDBContainer.getConnectionString()}/urlShortener?directConnection=true`;
+    process.env.APP_PORT = 5000;
+  });
+
+  afterAll(async () => {
+    await mongoose.connection.close();
+    await mongoDBContainer.stop();
+  });
+
+  describe('GET /links/:linkId', () => {
+    it('returns data when record is found', async () => {
+      const serverModule = (await import('../server.mjs')).app;
+
+      await supertest(serverModule)
+        .get('/links/Zx8lP5bWMhEgnvH')
+        .expect(200)
+        .then((response) => {
+          expect(response.body).toMatchObject({
+            success: true,
+            data: {
+              _id: expect.any(String),
+              shortId: 'Zx8lP5bWMhEgnvH',
+              target: 'https://www.youtube.com/',
+            },
+          });
+        });
+    });
+
+    it('returns error when record is not found', async () => {
+      const serverModule = (await import('../server.mjs')).app;
+
+      await supertest(serverModule)
+        .get('/links/fake-record')
+        .expect(404)
+        .then((response) => {
+          expect(response.body).toMatchObject({
+            success: false,
+            error: 'Record not found',
+          });
+        });
+    });
+  });
+
+  describe('GET /:shortId', () => {
+    it('returns redirect when record is found', async () => {
+      const serverModule = (await import('../server.mjs')).app;
+
+      await supertest(serverModule)
+        .get('/Zx8lP5bWMhEgnvH')
+        .expect(301)
+        .then((response) => {
+          expect(response.headers.location).toBe('https://www.youtube.com/');
+        });
+    });
+
+    it('returns error when record is not found', async () => {
+      const serverModule = (await import('../server.mjs')).app;
+
+      await supertest(serverModule)
+        .get('/fake-record')
+        .expect(404)
+        .then((response) => {
+          expect(response.body).toMatchObject({
+            success: false,
+            error: 'Record not found',
+          });
+        });
+    });
+  });
+
+  describe('POST /links', () => {
+    it('returns 201 whenever record created', async () => {
+      nanoId.mockReturnValueOnce('bingo');
+      const serverModule = (await import('../server.mjs')).app;
+
+      await supertest(serverModule)
+        .post('/links')
+        .set('Content-Type', 'application/json')
+        .set('Accept', 'application/json')
+        .send({ target: 'https://www.duckduckgo.com/' })
+        .expect(201)
+        .then((response) => {
+          expect(response.body).toMatchObject({
+            success: true,
+            data: {
+              _id: expect.any(String),
+              shortId: expect.any(String),
+              target: 'https://www.duckduckgo.com/',
+            },
+          });
+        });
+    });
+
+    it('returns 409 whenever URL is invalid', async () => {
+      nanoId.mockReturnValueOnce('bingo');
+      const serverModule = (await import('../server.mjs')).app;
+
+      await supertest(serverModule)
+        .post('/links')
+        .set('Content-Type', 'application/json')
+        .set('Accept', 'application/json')
+        .send({ target: 'bad-url' })
+        .expect(409)
+        .then((response) => {
+          expect(response.body).toMatchObject({
+            success: false,
+            error: 'Invalid target',
+          });
+        });
+    });
+
+    it('returns 409 when Short ID already exists', async () => {
+      nanoId.mockReturnValueOnce('Zx8lP5bWMhEgnvH');
+      const serverModule = (await import('../server.mjs')).app;
+
+      await supertest(serverModule)
+        .post('/links')
+        .set('Content-Type', 'application/json')
+        .set('Accept', 'application/json')
+        .send({ target: 'https://www.duckduckgo.com/' })
+        .expect(409)
+        .then((response) => {
+          expect(response.body).toMatchObject({
+            success: false,
+            error: 'Short Id already exists',
+          });
+        });
+    });
+  });
+});
diff --git a/src/index.mjs b/src/index.mjs
@@ -1,67 +1,6 @@
 import 'dotenv/config';
-import express from 'express';
-import mongoose from 'mongoose';
-import { nanoid } from 'nanoid';
 import { config } from './config.mjs';
-import { LinkModel } from './models/link.mjs';
-import { isValidHTTPURL } from './utils.mjs';
-
-mongoose.connection.once('open', () => {
-  console.log('Connected to MongoDB');
-});
-
-await mongoose.connect(config.MONGODB_URL);
-
-const app = express();
-
-app.use(express.json());
-
-app.get('/health', (_req, res) => {
-  res.status(200).json({ status: 'ok' });
-});
-
-app.get('/links/:shortId', async (req, res) => {
-  const link = await LinkModel.findOne({ shortId: req.params.shortId });
-
-  if (link === null) {
-    return res.status(404).json({ success: false, error: 'Record not found' });
-  }
-
-  res.status(200).json({ success: true, data: link.toJSON() });
-});
-
-app.post('/links', async (req, res) => {
-  const isValidTarget = isValidHTTPURL(req.body.target);
-  if (!isValidTarget) {
-    return res.status(409).json({ success: false, error: 'Invalid target' });
-  }
-
-  const shortId = nanoid(25);
-
-  const existingLink = await LinkModel.findOne({ shortId });
-  if (existingLink) {
-    return res
-      .status(409)
-      .json({ success: false, error: 'Short Id already exists' });
-  }
-
-  const link = await new LinkModel({
-    shortId,
-    target: req.body.target,
-  }).save();
-
-  res.status(201).json({ success: true, data: link.toJSON() });
-});
-
-app.get('/:shortId', async (req, res) => {
-  const link = await LinkModel.findOne({ shortId: req.params.shortId });
-
-  if (link === null) {
-    return res.status(404).json({ success: false, error: 'Record not found' });
-  }
-
-  res.redirect(301, link.target);
-});
+import { app } from './server.mjs';
 
 app.listen(config.APP_PORT, '0.0.0.0', () => {
   console.info(`Listening on port 0.0.0.0:${config.APP_PORT}`);

diff --git a/src/server.mjs b/src/server.mjs
@@ -0,0 +1,76 @@
+import 'dotenv/config';
+import express from 'express';
+import mongoose from 'mongoose';
+import { nanoid } from 'nanoid';
+import { config } from './config.mjs';
+import { LinkModel } from './models/link.mjs';
+import { isValidHTTPURL } from './utils.mjs';
+
+mongoose.connection.once('open', () => {
+  console.log('Connected to MongoDB');
+});
+
+mongoose.connection.once('disconnected', () => {
+  console.log('Disconnected from MongoDB');
+});
+
+try {
+  await mongoose.connect(config.MONGODB_URL, {
+    useNewUrlParser: true,
+    useUnifiedTopology: true,
+  });
+} catch (error) {
+  console.log('Mongoose connection failed:', error);
+  process.exit(1);
+}
+
+export const app = express();
+
+app.use(express.json());
+
+app.get('/health', (_req, res) => {
+  res.status(200).json({ status: 'ok' });
+});
+
+app.get('/links/:shortId', async (req, res) => {
+  const link = await LinkModel.findOne({ shortId: req.params.shortId });
+
+  if (link === null) {
+    return res.status(404).json({ success: false, error: 'Record not found' });
+  }
+
+  res.status(200).json({ success: true, data: link.toJSON() });
+});
+
+app.post('/links', async (req, res) => {
+  const isValidTarget = isValidHTTPURL(req.body.target);
+  if (!isValidTarget) {
+    return res.status(409).json({ success: false, error: 'Invalid target' });
+  }
+
+  const shortId = nanoid(25);
+
+  const existingLink = await LinkModel.findOne({ shortId });
+  if (existingLink) {
+    return res
+      .status(409)
+      .json({ success: false, error: 'Short Id already exists' });
+  }
+
+  const link = await new LinkModel({
+    shortId,
+    target: req.body.target,
+  }).save();
+
+  res.status(201).json({ success: true, data: link.toJSON() });
+});
+
+app.get('/:shortId', async (req, res) => {
+  const link = await LinkModel.findOne({ shortId: req.params.shortId });
+
+  if (link === null) {
+    return res.status(404).json({ success: false, error: 'Record not found' });
+  }
+
+  res.redirect(301, link.target);
+});