Cleanup step after all tests are run to clear the database.

louissullivan4 · Jan 2, 2024 · 3a278a0 · 3a278a0
1 parent b5f9d65
commit 3a278a0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/test/integration/apiEndpointTests.js b/test/integration/apiEndpointTests.js
@@ -1,10 +1,10 @@
 const request = require('supertest');
 const createApp = require('../../src/server');
+const clearDatabase = require('../../src/utils/clearDatabase');
 
+const app = createApp(true); 
 describe('API Endpoints', () => {
   it('should create a new user', (done) => {
-    const app = createApp(true); 
-
     request(app)
       .post('/auth/signup')
       .send({ username: 'testuser', password: 'password123' })
@@ -16,8 +16,6 @@ describe('API Endpoints', () => {
   });
 
   it('should not create a new user because they already exist', (done) => {
-    const app = createApp(true); 
-
     request(app)
       .post('/auth/signup')
       .send({ username: 'testuser', password: 'password123' })
@@ -29,8 +27,6 @@ describe('API Endpoints', () => {
   });
 
   it('should not create a new user because they already exist', (done) => {
-    const app = createApp(true); 
-
     request(app)
       .post('/auth/signup')
       .send({ username: 'testuser', password: 'password123' })
@@ -43,8 +39,6 @@ describe('API Endpoints', () => {
 
 
   it ('should return 404 for unknown endpoints', (done) => {
-    const app = createApp(true); 
-
     request(app)
       .get('/unknown')
       .expect(404)
@@ -54,4 +48,7 @@ describe('API Endpoints', () => {
       });
   });
 
+  afterAll(async () => {
+    await clearDatabase();
+  });
 });
diff --git a/test/unit/modelTests.js b/test/unit/modelTests.js
@@ -11,5 +11,7 @@ describe('User Model', () => {
     expect(user.username).toBe('testUser');
     expect(user.password).toBe('password123');
   });
-  clearDatabase("User");
+  afterAll(async () => {
+    await clearDatabase();
+  });
 });