一个数据模拟的库,更方便地测试 Egg 应用、插件及自定义 Egg 框架。egg-mock
拓展自 node_modules/mm,你可以使用所有 mm
包含的 API。
$ npm i egg-mock --save-dev
通过 mm.app
启动应用,可以使用 App 的 API 模拟数据
// test/index.test.js
const path = require('path');
const mm = require('egg-mock');
describe('some test', () => {
let app;
before(() => {
app = mm.app({
baseDir: 'apps/foo'
customEgg: path.join(__dirname, '../node_modules/egg'),
});
return app.ready();
})
after(() => app.close());
it('should request /', () => {
return app.httpRequest()
.get('/')
.expect(200);
});
});
使用 mm.app
启动后可以通过 app.agent
访问到 agent 对象。
使用 mm.cluster
启动多进程测试,API 与 mm.app
一致。
应用开发者不需要传入 baseDir,其为当前路径
before(() => {
app = mm.app({
customEgg: path.join(__dirname, '../node_modules/egg'),
});
return app.ready();
});
框架开发者需要指定 customEgg,会将当前路径指定为框架入口
before(() => {
app = mm.app({
baseDir: 'apps/demo',
customEgg: true,
});
return app.ready();
});
在插件目录下执行测试用例时,只要 package.json
中有 eggPlugin.name
字段,就会自动把当前目录加到插件列表中。
before(() => {
app = mm.app({
baseDir: 'apps/demo',
customEgg: path.join(__dirname, '../node_modules/egg'),
});
return app.ready();
});
也可以通过 customEgg 指定其他框架,比如希望在 aliyun-egg 和 framework-b 同时测试此插件。
describe('aliyun-egg', () => {
let app;
before(() => {
app = mm.app({
baseDir: 'apps/demo',
customEgg: path.join(__dirname, 'node_modules/aliyun-egg'),
});
return app.ready();
});
});
describe('framework-b', () => {
let app;
before(() => {
app = mm.app({
baseDir: 'apps/demo',
customEgg: path.join(__dirname, 'node_modules/framework-b'),
});
return app.ready();
});
});
如果当前目录确实是一个 egg 插件,但是又不想当它是一个插件来测试,可以通过 options.plugin
选项来关闭:
before(() => {
app = mm.app({
baseDir: 'apps/demo',
customEgg: path.join(__dirname, 'node_modules/egg'),
plugin: false,
});
return app.ready();
});
创建一个 mock 的应用。
创建一个多进程应用,因为是多进程应用,无法获取 worker 的属性,只能通过 supertest 请求。
const mm = require('egg-mock');
describe('test/app.js', () => {
let app, config;
before(() => {
app = mm.cluster();
return app.ready();
});
after(() => app.close());
it('some test', () => {
return app.httpRequest()
.get('/config')
.expect(200)
});
});
默认会启用覆盖率,因为覆盖率比较慢,可以设置 coverage 关闭
mm.cluster({
coverage: false,
});
设置环境变量,主要用于启动阶段,运行阶段可以使用 app.mockEnv。
// 模拟生成环境
mm.env('prod');
mm.app({
cache: false,
});
具体值见 https://github.com/eggjs/egg-core/blob/master/lib/loader/egg_loader.js#L82
mock 终端日志打印级别
// 不输出到终端
mm.consoleLevel('NONE');
可选 level 为 DEBUG
, INFO
, WARN
, ERROR
, NONE
模拟操作系统用户目录
还原所有 mock 数据,一般需要结合 afterEach(mm.restore)
使用
mm.app 和 mm.cluster 的配置参数
当前应用的目录,如果是应用本身的测试可以不填默认为 $CWD。
指定完整路径
mm.app({
baseDir: path.join(__dirname, 'fixtures/apps/demo'),
})
也支持缩写,找 test/fixtures 目录下的
mm.app({
baseDir: 'apps/demo',
})
指定框架路径
mm.app({
baseDir: 'apps/demo',
customEgg: path.join(__dirname, 'fixtures/egg'),
})
对于框架的测试用例,可以指定 true,会自动加载当前路径。
指定插件的路径,只用于插件测试。设置为 true 会将当前路径设置到插件列表。
mm.app({
baseDir: 'apps/demo',
plugin: true,
})
传入插件列表,可以自定义多个插件
是否需要缓存,默认开启。
是通过 baseDir 缓存的,如果不需要可以关闭,但速度会慢。
是否需要清理 log 目录,默认开启。
如果是通过 ava 等并行测试框架进行测试,需要手动在执行测试前进行统一的日志清理,不能通过 mm 来处理,设置 clean
为 false
。
断言指定的字符串记录在指定的日志中。
建议 app.mockLog()
和 app.expectLog()
或者 app.notExpectLog()
配对使用。
单独使用 app.expectLog()
或者 app.notExpectLog()
需要依赖日志的写入速度,在服务器磁盘高 IO 的时候,会出现不稳定的结果。
it('should work', async () => {
// 将日志记录到内存,用于下面的 expectLog
app.mockLog();
await app.httpRequest()
.get('/')
.expect('hello world')
.expect(200);
app.expectLog('foo in logger');
app.expectLog('foo in coreLogger', 'coreLogger');
app.expectLog('foo in myCustomLogger', 'myCustomLogger');
app.notExpectLog('bar in logger');
app.notExpectLog('bar in coreLogger', 'coreLogger');
app.notExpectLog('bar in myCustomLogger', 'myCustomLogger');
});
请求当前应用 http 服务的辅助工具。
it('should work', () => {
return app.httpRequest()
.get('/')
.expect('hello world')
.expect(200);
});
更多信息请查看 supertest 的 API 说明。
断言当前请求响应不包含指定 header
it('should work', () => {
return app.httpRequest()
.get('/')
.unexpectHeader('set-cookie')
.expect(200);
});
断言当前请求响应包含指定 header
it('should work', () => {
return app.httpRequest()
.get('/')
.expectHeader('set-cookie')
.expect(200);
});
模拟上下文数据
const ctx = app.mockContext({
user: {
name: 'Jason'
}
});
console.log(ctx.user.name); // Jason
安全的模拟上下文数据,同一用例用多次调用 mockContext 可能会造成 AsyncLocalStorage 污染
await app.mockContextScope(async ctx => {
console.log(ctx.user.name); // Jason
}, {
user: {
name: 'Jason'
}
});
app.mockCookies({
foo: 'bar'
});
const ctx = app.mockContext();
console.log(ctx.getCookie('foo'));
模拟请求头
app.mockSession({
foo: 'bar'
});
const ctx = app.mockContext();
console.log(ctx.session.foo);
it('should mock user name', function* () {
app.mockService('user', 'getName', function* (ctx, methodName, args) {
return 'popomore';
});
const ctx = app.mockContext();
yield ctx.service.user.getName();
});
可以模拟一个错误
app.mockServiceError('user', 'home', new Error('mock error'));
模拟 csrf,不用传递 token
app.mockCsrf();
return app.httpRequest()
.post('/login')
.expect(302);
模拟 httpclient 的请求,例如 ctx.curl
app.get('/', async ctx => {
const ret = await ctx.curl('https://eggjs.org');
this.body = ret.data.toString();
});
app.mockHttpclient('https://eggjs.org', {
// 模拟的参数,可以是 buffer / string / json / function
// 都会转换成 buffer
// 按照请求时的 options.dataType �来做对应的转换
data: 'mock egg',
});
return app.httpRequest()
.post('/')
.expect('mock egg');
我们提供了一个 bootstrap 来减少单测中的重复代码:
const { app, mock, assert } = require('egg-mock/bootstrap');
describe('test app', () => {
it('should request success', () => {
// mock data will be restored each case
mock.data(app, 'method', { foo: 'bar' });
return app.httpRequest()
.get('/foo')
.expect(res => {
assert(!res.headers.foo);
})
.expect(/bar/);
});
});
describe('test ctx', () => {
it('can use ctx', async function() {
const res = await this.ctx.service.foo();
assert(res === 'foo');
});
});
我们将会在每个 case 中自定注入 ctx, 可以通过 app.currentContext
来获取当前的 ctx。
并且第一次使用 app.mockContext
会自动复用当前 case 的上下文。
const { app, mock, assert } = require('egg-mock/bootstrap');
describe('test ctx', () => {
it('should can use ctx', () => {
const ctx = app.currentContext;
assert(ctx);
});
it('should reuse ctx', () => {
const ctx = app.currentContext;
// 第一次调用会复用上下文
const mockCtx = app.mockContext();
assert(ctx === mockCtx);
// 后续调用会新建上下文
// 极不建议多次调用 app.mockContext
// 这会导致上下文污染
// 建议使用 app.mockContextScope
const mockCtx2 = app.mockContext();
assert(ctx !== mockCtx);
});
});
EGG_BASE_DIR: the base dir of egg app EGG_FRAMEWORK: the framework of egg app
Please open an issue here.
Made with contributors-img.