tachybase_todo/packages/core/test/src/mockServer.ts
Dunqing 25a3a8affa
feat: support token blacklist (#2168)
* feat: support token blacklist, Close T-799

* feat: clean

* fix: possible token does not exist

* fix: update

* feat: update

* feat: add node-cron to delete expired token

* fix: findOrCreate not work and add test case

* test: add token-blacklist tests

* feat: add test cases for blacklist in authManager

* test: update better

* fix: should hidden token field

* test: clean

* test: clean

* fix: should stop cron in afterStop

* refactor: move delete expired token in token blacklist service

* feat: remove plugin disable/enable logic

* fix: clean

* test: revert

* fix: cron typo
2023-07-05 21:57:57 +08:00

208 lines
5.4 KiB
TypeScript

import { Database, mockDatabase } from '@nocobase/database';
import Application, { ApplicationOptions, PluginManager } from '@nocobase/server';
import jwt from 'jsonwebtoken';
import qs from 'qs';
import supertest, { SuperAgentTest } from 'supertest';
interface ActionParams {
filterByTk?: any;
fields?: string[];
filter?: any;
sort?: string[];
page?: number;
pageSize?: number;
values?: any;
/**
* @deprecated
*/
resourceName?: string;
/**
* @deprecated
*/
resourceIndex?: string;
/**
* @deprecated
*/
associatedName?: string;
/**
* @deprecated
*/
associatedIndex?: string;
[key: string]: any;
}
interface SortActionParams {
resourceName?: string;
resourceIndex?: any;
associatedName?: string;
associatedIndex?: any;
sourceId?: any;
targetId?: any;
sortField?: string;
method?: string;
target?: any;
sticky?: boolean;
[key: string]: any;
}
interface Resource {
get: (params?: ActionParams) => Promise<supertest.Response>;
list: (params?: ActionParams) => Promise<supertest.Response>;
create: (params?: ActionParams) => Promise<supertest.Response>;
update: (params?: ActionParams) => Promise<supertest.Response>;
destroy: (params?: ActionParams) => Promise<supertest.Response>;
sort: (params?: SortActionParams) => Promise<supertest.Response>;
[name: string]: (params?: ActionParams) => Promise<supertest.Response>;
}
export class MockServer extends Application {
async loadAndInstall(options: any = {}) {
await this.load({ method: 'install' });
if (options.afterLoad) {
await options.afterLoad(this);
}
await this.install({
...options,
sync: {
force: false,
alter: {
drop: false,
},
},
});
}
async cleanDb() {
await this.db.clean({ drop: true });
}
agent(): SuperAgentTest & {
login: (user: any) => SuperAgentTest;
loginUsingId: (userId: number) => SuperAgentTest;
resource: (name: string, resourceOf?: any) => Resource;
} {
const agent = supertest.agent(this.appManager.callback());
const prefix = this.resourcer.options.prefix;
const proxy = new Proxy(agent, {
get(target, method: string, receiver) {
if (['login', 'loginUsingId'].includes(method)) {
return (userOrId: any) => {
return proxy
.auth(
jwt.sign(
{
userId: typeof userOrId === 'number' ? userOrId : userOrId?.id,
},
process.env.APP_KEY,
{
expiresIn: '1d',
},
),
{ type: 'bearer' },
)
.set('X-Authenticator', 'basic');
};
}
if (method === 'resource') {
return (name: string, resourceOf?: any) => {
const keys = name.split('.');
const proxy = new Proxy(
{},
{
get(target, method: string, receiver) {
return (params: ActionParams = {}) => {
let { filterByTk, values = {}, file, ...restParams } = params;
if (params.associatedIndex) {
resourceOf = params.associatedIndex;
}
if (params.resourceIndex) {
filterByTk = params.resourceIndex;
}
let url = prefix || '';
if (keys.length > 1) {
url += `/${keys[0]}/${resourceOf}/${keys[1]}`;
} else {
url += `/${name}`;
}
url += `:${method as string}`;
if (filterByTk) {
url += `/${filterByTk}`;
}
const queryString = qs.stringify(restParams, { arrayFormat: 'brackets' });
let request;
switch (method) {
case 'list':
case 'get':
request = agent.get(`${url}?${queryString}`);
break;
default:
request = agent.post(`${url}?${queryString}`);
break;
}
return file ? request.attach('file', file).field(values) : request.send(values);
};
},
},
);
return proxy;
};
}
return agent[method];
},
});
return proxy as any;
}
}
export function mockServer(options: ApplicationOptions = {}) {
if (typeof TextEncoder === 'undefined') {
global.TextEncoder = require('util').TextEncoder;
}
if (typeof TextDecoder === 'undefined') {
global.TextDecoder = require('util').TextDecoder;
}
// @ts-ignore
if (!PluginManager.findPackagePatched) {
PluginManager.getPackageJson = () => {
return {
version: '0.0.0',
};
};
// @ts-ignore
PluginManager.findPackagePatched = true;
}
let database;
if (options?.database instanceof Database) {
database = options.database;
} else {
database = mockDatabase(<any>options?.database || {});
}
const app = new MockServer({
acl: false,
...options,
database,
});
app.pm.generateClientFile = async () => {};
return app;
}
export function createMockServer() {}
export default mockServer;