2022-03-28 22:01:10 +08:00
|
|
|
import { Database, mockDatabase } from '@nocobase/database';
|
2023-08-24 17:47:45 +08:00
|
|
|
import Application, { AppSupervisor, ApplicationOptions, Gateway, PluginManager } from '@nocobase/server';
|
2023-06-07 23:46:42 +08:00
|
|
|
import jwt from 'jsonwebtoken';
|
2021-09-09 13:09:25 +08:00
|
|
|
import qs from 'qs';
|
2021-09-16 00:38:48 +08:00
|
|
|
import supertest, { SuperAgentTest } from 'supertest';
|
2021-09-09 13:09:25 +08:00
|
|
|
|
2021-09-09 22:38:39 +08:00
|
|
|
interface ActionParams {
|
2022-01-23 15:04:50 +08:00
|
|
|
filterByTk?: any;
|
|
|
|
fields?: string[];
|
2021-09-09 22:38:39 +08:00
|
|
|
filter?: any;
|
|
|
|
sort?: string[];
|
|
|
|
page?: number;
|
2022-02-18 09:46:10 +08:00
|
|
|
pageSize?: number;
|
2021-09-09 22:38:39 +08:00
|
|
|
values?: any;
|
2022-01-23 15:04:50 +08:00
|
|
|
/**
|
|
|
|
* @deprecated
|
|
|
|
*/
|
2021-09-09 22:38:39 +08:00
|
|
|
resourceName?: string;
|
2022-01-23 15:04:50 +08:00
|
|
|
/**
|
|
|
|
* @deprecated
|
|
|
|
*/
|
2021-12-04 16:28:52 +08:00
|
|
|
resourceIndex?: string;
|
2022-01-23 15:04:50 +08:00
|
|
|
/**
|
|
|
|
* @deprecated
|
|
|
|
*/
|
2021-09-09 22:38:39 +08:00
|
|
|
associatedName?: string;
|
2022-01-23 15:04:50 +08:00
|
|
|
/**
|
|
|
|
* @deprecated
|
|
|
|
*/
|
2021-12-04 16:28:52 +08:00
|
|
|
associatedIndex?: string;
|
2023-03-10 19:16:00 +08:00
|
|
|
|
2021-09-09 22:38:39 +08:00
|
|
|
[key: string]: any;
|
|
|
|
}
|
|
|
|
|
|
|
|
interface SortActionParams {
|
|
|
|
resourceName?: string;
|
2021-12-04 16:28:52 +08:00
|
|
|
resourceIndex?: any;
|
2021-09-09 22:38:39 +08:00
|
|
|
associatedName?: string;
|
2021-12-04 16:28:52 +08:00
|
|
|
associatedIndex?: any;
|
2021-09-09 22:38:39 +08:00
|
|
|
sourceId?: any;
|
|
|
|
targetId?: any;
|
|
|
|
sortField?: string;
|
|
|
|
method?: string;
|
|
|
|
target?: any;
|
|
|
|
sticky?: boolean;
|
2023-03-10 19:16:00 +08:00
|
|
|
|
2021-09-09 22:38:39 +08:00
|
|
|
[key: string]: any;
|
|
|
|
}
|
|
|
|
|
2021-09-09 13:09:25 +08:00
|
|
|
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>;
|
2021-09-09 22:38:39 +08:00
|
|
|
sort: (params?: SortActionParams) => Promise<supertest.Response>;
|
2023-03-10 19:16:00 +08:00
|
|
|
|
2021-09-09 13:09:25 +08:00
|
|
|
[name: string]: (params?: ActionParams) => Promise<supertest.Response>;
|
|
|
|
}
|
|
|
|
|
|
|
|
export class MockServer extends Application {
|
2022-05-19 00:40:55 +08:00
|
|
|
async loadAndInstall(options: any = {}) {
|
2022-10-28 15:09:14 +08:00
|
|
|
await this.load({ method: 'install' });
|
2023-02-16 10:53:04 +08:00
|
|
|
|
2023-03-10 19:16:00 +08:00
|
|
|
if (options.afterLoad) {
|
|
|
|
await options.afterLoad(this);
|
|
|
|
}
|
|
|
|
|
2022-02-15 22:32:02 +08:00
|
|
|
await this.install({
|
2022-05-19 00:40:55 +08:00
|
|
|
...options,
|
2022-02-15 22:32:02 +08:00
|
|
|
sync: {
|
2022-10-28 15:09:14 +08:00
|
|
|
force: false,
|
2022-02-15 22:32:02 +08:00
|
|
|
alter: {
|
|
|
|
drop: false,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
});
|
2022-01-23 15:04:50 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
async cleanDb() {
|
2023-02-16 10:53:04 +08:00
|
|
|
await this.db.clean({ drop: true });
|
2022-01-23 15:04:50 +08:00
|
|
|
}
|
|
|
|
|
2023-08-24 17:47:45 +08:00
|
|
|
async destroy(options: any = {}): Promise<void> {
|
|
|
|
await super.destroy(options);
|
|
|
|
|
|
|
|
Gateway.getInstance().destroy();
|
|
|
|
await AppSupervisor.getInstance().destroy();
|
|
|
|
}
|
|
|
|
|
2023-06-07 23:46:42 +08:00
|
|
|
agent(): SuperAgentTest & {
|
|
|
|
login: (user: any) => SuperAgentTest;
|
|
|
|
loginUsingId: (userId: number) => SuperAgentTest;
|
|
|
|
resource: (name: string, resourceOf?: any) => Resource;
|
|
|
|
} {
|
2023-08-24 17:47:45 +08:00
|
|
|
const agent = supertest.agent(this.callback());
|
2021-09-09 13:09:25 +08:00
|
|
|
const prefix = this.resourcer.options.prefix;
|
2021-12-03 07:31:22 +08:00
|
|
|
const proxy = new Proxy(agent, {
|
2021-09-09 13:09:25 +08:00
|
|
|
get(target, method: string, receiver) {
|
2023-06-07 23:46:42 +08:00
|
|
|
if (['login', 'loginUsingId'].includes(method)) {
|
|
|
|
return (userOrId: any) => {
|
|
|
|
return proxy
|
|
|
|
.auth(
|
|
|
|
jwt.sign(
|
|
|
|
{
|
|
|
|
userId: typeof userOrId === 'number' ? userOrId : userOrId?.id,
|
|
|
|
},
|
|
|
|
process.env.APP_KEY,
|
2023-07-05 21:57:57 +08:00
|
|
|
{
|
|
|
|
expiresIn: '1d',
|
|
|
|
},
|
2023-06-07 23:46:42 +08:00
|
|
|
),
|
|
|
|
{ type: 'bearer' },
|
|
|
|
)
|
|
|
|
.set('X-Authenticator', 'basic');
|
|
|
|
};
|
|
|
|
}
|
2021-09-16 00:38:48 +08:00
|
|
|
if (method === 'resource') {
|
2022-01-23 15:04:50 +08:00
|
|
|
return (name: string, resourceOf?: any) => {
|
2021-09-16 00:38:48 +08:00
|
|
|
const keys = name.split('.');
|
2021-12-06 21:23:34 +08:00
|
|
|
const proxy = new Proxy(
|
|
|
|
{},
|
|
|
|
{
|
|
|
|
get(target, method: string, receiver) {
|
|
|
|
return (params: ActionParams = {}) => {
|
2023-08-24 17:47:45 +08:00
|
|
|
let { filterByTk } = params;
|
|
|
|
const { values = {}, file, ...restParams } = params;
|
2022-01-23 15:04:50 +08:00
|
|
|
if (params.associatedIndex) {
|
|
|
|
resourceOf = params.associatedIndex;
|
|
|
|
}
|
|
|
|
if (params.resourceIndex) {
|
|
|
|
filterByTk = params.resourceIndex;
|
|
|
|
}
|
2022-01-24 14:10:35 +08:00
|
|
|
let url = prefix || '';
|
2021-12-06 21:23:34 +08:00
|
|
|
if (keys.length > 1) {
|
2022-01-23 15:04:50 +08:00
|
|
|
url += `/${keys[0]}/${resourceOf}/${keys[1]}`;
|
2021-12-06 21:23:34 +08:00
|
|
|
} else {
|
2022-01-23 15:04:50 +08:00
|
|
|
url += `/${name}`;
|
2021-12-06 21:23:34 +08:00
|
|
|
}
|
|
|
|
url += `:${method as string}`;
|
2022-01-23 15:04:50 +08:00
|
|
|
if (filterByTk) {
|
|
|
|
url += `/${filterByTk}`;
|
2021-12-06 21:23:34 +08:00
|
|
|
}
|
2021-12-03 07:31:22 +08:00
|
|
|
|
2022-02-15 22:32:02 +08:00
|
|
|
const queryString = qs.stringify(restParams, { arrayFormat: 'brackets' });
|
|
|
|
|
2023-06-06 19:30:42 +08:00
|
|
|
let request;
|
|
|
|
|
2021-12-06 21:23:34 +08:00
|
|
|
switch (method) {
|
|
|
|
case 'list':
|
|
|
|
case 'get':
|
2023-06-06 19:30:42 +08:00
|
|
|
request = agent.get(`${url}?${queryString}`);
|
|
|
|
break;
|
2021-12-06 21:23:34 +08:00
|
|
|
default:
|
2023-06-06 19:30:42 +08:00
|
|
|
request = agent.post(`${url}?${queryString}`);
|
|
|
|
break;
|
2021-12-06 21:23:34 +08:00
|
|
|
}
|
2023-06-06 19:30:42 +08:00
|
|
|
|
|
|
|
return file ? request.attach('file', file).field(values) : request.send(values);
|
2021-12-06 21:23:34 +08:00
|
|
|
};
|
|
|
|
},
|
2021-09-16 00:38:48 +08:00
|
|
|
},
|
2021-12-06 21:23:34 +08:00
|
|
|
);
|
2021-09-16 00:38:48 +08:00
|
|
|
return proxy;
|
2021-12-06 21:23:34 +08:00
|
|
|
};
|
2021-09-16 00:38:48 +08:00
|
|
|
}
|
2022-07-19 22:51:27 +08:00
|
|
|
return agent[method];
|
2021-12-06 21:23:34 +08:00
|
|
|
},
|
2021-09-09 13:09:25 +08:00
|
|
|
});
|
2021-09-16 00:38:48 +08:00
|
|
|
return proxy as any;
|
2021-09-09 13:09:25 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-02-15 00:20:25 +08:00
|
|
|
export function mockServer(options: ApplicationOptions = {}) {
|
2023-03-03 13:54:37 +08:00
|
|
|
if (typeof TextEncoder === 'undefined') {
|
|
|
|
global.TextEncoder = require('util').TextEncoder;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (typeof TextDecoder === 'undefined') {
|
|
|
|
global.TextDecoder = require('util').TextDecoder;
|
|
|
|
}
|
|
|
|
|
2023-08-24 17:47:45 +08:00
|
|
|
Gateway.getInstance().reset();
|
|
|
|
AppSupervisor.getInstance().reset();
|
|
|
|
|
2023-03-03 13:54:37 +08:00
|
|
|
// @ts-ignore
|
|
|
|
if (!PluginManager.findPackagePatched) {
|
|
|
|
PluginManager.getPackageJson = () => {
|
|
|
|
return {
|
|
|
|
version: '0.0.0',
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
|
|
|
// @ts-ignore
|
|
|
|
PluginManager.findPackagePatched = true;
|
|
|
|
}
|
|
|
|
|
2022-03-28 22:01:10 +08:00
|
|
|
let database;
|
|
|
|
if (options?.database instanceof Database) {
|
|
|
|
database = options.database;
|
|
|
|
} else {
|
|
|
|
database = mockDatabase(<any>options?.database || {});
|
|
|
|
}
|
|
|
|
|
2023-03-03 13:54:37 +08:00
|
|
|
const app = new MockServer({
|
2022-09-29 21:05:31 +08:00
|
|
|
acl: false,
|
2021-09-09 13:09:25 +08:00
|
|
|
...options,
|
2022-02-15 00:20:25 +08:00
|
|
|
database,
|
2021-09-09 13:09:25 +08:00
|
|
|
});
|
2023-03-03 13:54:37 +08:00
|
|
|
|
|
|
|
return app;
|
2021-09-09 13:09:25 +08:00
|
|
|
}
|
|
|
|
|
2022-01-24 14:10:35 +08:00
|
|
|
export function createMockServer() {}
|
|
|
|
|
2021-09-09 13:09:25 +08:00
|
|
|
export default mockServer;
|