UNPKG

@jianghujs/jianghu

Version:

Progressive Enterprise Framework

99 lines (90 loc) 2.91 kB
'use strict'; const assert = require('assert'); const sinon = require('sinon'); const path = require('path'); const mock = require('egg-mock'); const utils = require('../../../../utils'); const socketPackage = require('../../../../../plugins/socket-io/app/middleware/socketPackage'); describe('test/app/middleware/socketPackage.test.js', () => { before(() => { this.app = utils.app('apps/jianghu-config'); return this.app.ready(); }); after(async () => { await utils.deleteFileAndDirByPath(path.join(process.cwd(), 'test/fixtures/apps/jianghu-config/run')); await utils.deleteFileAndDirByPath(path.join(process.cwd(), 'test/fixtures/apps/jianghu-config/logs')); this.app.close(); }); describe('Test middleware socketPackage', () => { beforeEach(() => { const jianghuKnexResult = { where: () => {}, }; const whereResult = { first: () => {}, }; this.ctx = this.app.mockContext({}); mock(this.app, 'jianghuKnex', () => { return jianghuKnexResult; }); // this.socketPackage = socketPackage(); this.nextSpy = sinon.spy(); this.whereStub = sinon.stub(jianghuKnexResult, 'where').returns(whereResult); this.firstStub = sinon.stub(whereResult, 'first'); }); afterEach(() => { this.whereStub.restore(); this.firstStub.restore(); mock.restore(); }); it('should success', async () => { const expUserId = 'test101'; const expDeviceId = `${Date.now()}`; const expResourceId = 'page.index'; const expResponseStatus = 200; const expPackageId = `package_${Date.now()}`; const expRequestBody = { packageId: expPackageId, packageType: 'socketRequest', appData: { appId: 'jianghu', pageId: 'index', actionId: 'selectItem', actionData: {}, }, }; const expResponseBody = { status: expResponseStatus, }; const expUser = { userId: expUserId, username: 'username', user: { userId: expUserId, username: 'username', deviceId: expDeviceId, userStatus: 'active', md5Salt: 'test', }, }; const expPackageResource = { resourceId: expResourceId, }; this.ctx.userInfo = { user: expUser, allowResourceList: [{ resourceId: expResourceId }], }; this.ctx.packageResource = { resourceId: expResourceId, resourceData: {}, }; this.ctx.request.body = expRequestBody; this.ctx.body = expResponseBody; this.firstStub.returns(expPackageResource); const result = await socketPackage(this.ctx); assert.deepEqual(this.whereStub.callCount, 1); assert.deepEqual(this.firstStub.callCount, 1); assert.deepEqual(this.ctx, result); }); }); });