server.js

'use strict';

const express = require('express');
const path = require('path');
const connectDB = require('./config/db');
const dotenv = require('dotenv');
dotenv.config();
const app = express();
const { getNotifications } = require('./services/notifications');

const setup = async () => {
  if (process.env.NODE_ENV !== 'test') {
    // Connect database.
    await connectDB();
    console.log('After connecting database...');
  }

  // Init middleware.
  app.use(express.json({ extended: false }));

  // Define routes.
  const API = 'api/v1';
  app.use(`/${API}/auth`, require(`./${API}/routes/auth`));
  app.use(`/${API}/users`, require(`./${API}/routes/user`));
  app.use(`/${API}/notifications`, require(`./${API}/routes/notification`));
  app.use(`/${API}/restaurants`, require(`./${API}/routes/restaurant`));

  // Serve static assets in production
  if (process.env.NODE_ENV === 'production') {
    // Set static folder
    app.use(express.static('client/build'));

    app.get('*', (_req, res) =>
      res.sendFile(path.resolve(__dirname, 'client', 'build', 'index.html'))
    );
  }

  // Listening to app.
  const PORT = process.env.PORT || 5000;
  app.listen(PORT, () => console.log(`Server started on port ${PORT}`));

  if (process.env.NODE_ENV !== 'test') {
    getNotifications();
  }
};

setup();

// Export app for testing purposes.
module.exports = app;