Fix heroku error (#57)

* Fix heroku error

* fix import

package.json

@@ -64,6 +64,9 @@
     "coverageReporters": [
       "json",
       "html"
+    ],
+    "setupFilesAfterEnv": [
+      "<rootDir>/src/setupTests.js"
     ]
   },
   "babel": {

src/auth/auth.test.js

@@ -1,6 +1,6 @@
 import request from 'supertest';
 import { app } from '../app.js';
-import dbConnection from '../helpers/dbConnection.js';
+import { connect, disconnect } from '../helpers/dbConnection.js';
 
 const userBody = {
   email: `[email protected]`,
@@ -28,10 +28,10 @@ jest.mock('nodemailer', () => ({
 
 describe('auth endpoints', () => {
   beforeAll(async () => {
-    await dbConnection.connect();
+    await connect();
   });
   afterAll(async () => {
-    await dbConnection.disconnect();
+    await disconnect();
   });
 
   it('should allow a POST to /auth/register with correct data', async () => {

src/helpers/dbConnection.js

@@ -1,7 +1,16 @@
-import * as productionDbConnection from './productionDbConnection.js';
-import * as testDbConnection from './testDbConnection.js';
+import mongoose from 'mongoose';
 
-const dbConnection =
-  process.env.NODE_ENV === 'test' ? testDbConnection : productionDbConnection;
+export const connect = async () => {
+  try {
+    await mongoose.connect(process.env.MONGO_URL, {
+      useNewUrlParser: true,
+      useUnifiedTopology: true,
+    });
+  } catch (error) {
+    console.log(error.message);
+  }
+};
 
-export default dbConnection;
+export const disconnect = () => {
+  return mongoose.disconnect();
+};

src/offer/offer.test.js

@@ -1,7 +1,7 @@
 import request from 'supertest';
 import { app } from '../app.js';
 import { USER_ROLE } from '../constants.js';
-import dbConnection from '../helpers/dbConnection.js';
+import { connect, disconnect } from '../helpers/dbConnection.js';
 
 const offerBody = {
   title: 'Super oferta',
@@ -88,12 +88,12 @@ const registerUsers = async () => {
 
 describe('offer endpoints', () => {
   beforeAll(async () => {
-    await dbConnection.connect();
+    await connect();
     await registerUsers();
   });
 
   afterAll(async () => {
-    await dbConnection.disconnect();
+    await disconnect();
   });
 
   it('should allow a POST to /offer when authorized', async () => {

src/server.js

@@ -1,8 +1,7 @@
 import { app } from './app.js';
-import dbConnection from './helpers/dbConnection.js';
+import { connect } from './helpers/dbConnection.js';
 
-dbConnection
-  .connect()
+connect()
   .then(() =>
     app.listen(process.env.PORT, () => {
       console.log('Server is listing on port', process.env.PORT);

src/setupTests.js

@@ -0,0 +1 @@
+jest.mock('./helpers/dbConnection');

src/user/user.test.js

@@ -1,7 +1,7 @@
 import request from 'supertest';
 import { app } from '../app.js';
 import { USER_ROLE } from '../constants.js';
-import dbConnection from '../helpers/dbConnection.js';
+import { connect, disconnect } from '../helpers/dbConnection.js';
 
 const userBody = {
   email: `[email protected]`,
@@ -60,13 +60,13 @@ jest.mock('nodemailer', () => ({
 
 describe('user endpoints', () => {
   beforeAll(async () => {
-    await dbConnection.connect();
+    await connect();
     await registerUsers();
     await getUsersData(adminToken);
   });
 
   afterAll(async () => {
-    await dbConnection.disconnect();
+    await disconnect();
   });
 
   describe('admin user', () => {