如何在 Jest 测试 function 中模拟变量？

Question

我正在用 Jest 测试 Express 应用程序并遇到一个小问题 - 模块使用在测试执行之前初始化的变量，这是我的app.js文件：

const app = express();
const isDev = process.env.NODE_ENV === 'development';

app.get('*', (req, res, next) => {
  if (isDev) {
    res.status(404).json({ error: 'Wrong URL' });
  } else {
    res.sendFile(path.join(__dirname, '../index.html'));
  }
});

app.use(errorHandler);

module.exports = app;

当我运行 Jest 测试时，我的process.env.NODE_ENV等于test ，这就是为什么我不能涵盖第一个if条件，其中isDev是true 。 我试图在测试请求之前重新分配process.env.NODE_ENV - 它可以工作，但是由于isDev变量初始化已经在测试执行之前完成，它没有工作。

这是我的测试：

const request = require('supertest');
const app = require('../app');

describe('GET /*', () => {
  const OLD_ENV = process.env;

  beforeEach(() => {
    // Clear JEST cache
    jest.resetModules();
    process.env = { ...OLD_ENV };
    Reflect.deleteProperty(process.env, 'NODE_ENV');
  });

  test('Not existing path (development env) - 404 status', async () => {
    process.env.NODE_ENV = 'development';

    const response = await request(app).
      get('/wrongUrl');
    expect(response.status).toBe(404);

  });

});

如何在我的测试中模拟isDev变量？

Answer 1

你可以使用jest.isolateModules(fn)来隔离应用，如下所示：

describe("GET /*", () => {
  describe("on development", () => {
    let app;
    beforeAll(() => {
      process.env.NODE_ENV = "development";
      jest.isolateModules(() => {
        app = require("../app");
      });
    });

    it("should to this", () => {
      expect(app).....
    });
  });

  describe("on production", () => {
    let app;
    beforeAll(() => {
      process.env.NODE_ENV = "production";
      jest.isolateModules(() => {
        app = require("../app");
      });
    });

    it("should to that", () => {
      expect(app())....
    });
  });
});

Answer 2

您可以创建一个仅用于测试的.env文件。 如果您使用的是express ，您也可以使用dotenv 。 有了这个 package，您可以从不同的文件导入环境变量。

只需在文件顶部添加这一行。

require('dotenv').config({ path: process.cwd() + '/path/to/test.env' });

这样，您可以在每次测试之前随时更改要使用的环境变量。