75 lines
2.2 KiB
TypeScript
75 lines
2.2 KiB
TypeScript
|
import qs from 'qs';
|
||
|
|
import supertest from 'supertest';
|
||
|
|
import Application, { ApplicationOptions } from '@nocobase/server';
|
||
|
|
import { ActionParams } from '@nocobase/resourcer';
|
||
|
|
import { getConfig } from './mockDatabase';
|
||
|
|
|
||
|
|
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>;
|
||
|
|
[name: string]: (params?: ActionParams) => Promise<supertest.Response>;
|
||
|
|
}
|
||
|
|
|
||
|
|
export class MockServer extends Application {
|
||
|
|
agent() {
|
||
|
|
return supertest.agent(this.callback());
|
||
|
|
}
|
||
|
|
|
||
|
|
resource(name: string) {
|
||
|
|
const agent = this.agent();
|
||
|
|
const keys = name.split('.');
|
||
|
|
const prefix = this.resourcer.options.prefix;
|
||
|
|
const proxy = new Proxy({}, {
|
||
|
|
get(target, method: string, receiver) {
|
||
|
|
return (params: ActionParams = {}) => {
|
||
|
|
const {
|
||
|
|
associatedKey,
|
||
|
|
resourceKey,
|
||
|
|
values = {},
|
||
|
|
file,
|
||
|
|
...restParams
|
||
|
|
} = params;
|
||
|
|
let url = prefix;
|
||
|
|
if (keys.length > 1) {
|
||
|
|
url = `/${keys[0]}/${associatedKey}/${keys[1]}}`
|
||
|
|
} else {
|
||
|
|
url = `/${name}`;
|
||
|
|
}
|
||
|
|
url += `:${method as string}`;
|
||
|
|
if (resourceKey) {
|
||
|
|
url += `/${resourceKey}`;
|
||
|
|
}
|
||
|
|
console.log('request url: ' + url);
|
||
|
|
switch (method) {
|
||
|
|
case 'upload':
|
||
|
|
return agent
|
||
|
|
.post(`${url}?${qs.stringify(restParams)}`)
|
||
|
|
.attach('file', file)
|
||
|
|
.field(values);
|
||
|
|
case 'list':
|
||
|
|
case 'get':
|
||
|
|
return agent.get(`${url}?${qs.stringify(restParams)}`);
|
||
|
|
default:
|
||
|
|
return agent
|
||
|
|
.post(`${url}?${qs.stringify(restParams)}`)
|
||
|
|
.send(values);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
},
|
||
|
|
});
|
||
|
|
return proxy as Resource;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export function mockServer(options?: ApplicationOptions) {
|
||
|
|
return new MockServer({
|
||
|
|
...options,
|
||
|
|
database: getConfig(options?.database),
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
export default mockServer;
|