* feat: improve code * chore: update version * feat: api service * fix: api services * feat: improve code * feat: improve code * feat: improve code * feat: pm socket * fix: test errors * feat: add built-in plugins before upgrade * feat: update docs * feat: improve code * fix: after load
84 lines
2.0 KiB
TypeScript
84 lines
2.0 KiB
TypeScript
import Database from '@nocobase/database';
|
|
import UsersPlugin from '@nocobase/plugin-users';
|
|
import { MockServer } from '@nocobase/test';
|
|
import { setCurrentRole } from '../middlewares/setCurrentRole';
|
|
import { prepareApp } from './prepare';
|
|
|
|
describe('role', () => {
|
|
let api: MockServer;
|
|
let db: Database;
|
|
|
|
let usersPlugin: UsersPlugin;
|
|
let ctx;
|
|
|
|
beforeEach(async () => {
|
|
api = await prepareApp();
|
|
|
|
db = api.db;
|
|
usersPlugin = api.getPlugin('users');
|
|
|
|
ctx = {
|
|
db,
|
|
state: {
|
|
currentRole: '',
|
|
}
|
|
}
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await api.destroy();
|
|
});
|
|
|
|
it('should set role with X-Role when exists', async () => {
|
|
ctx.state.currentUser = await db.getRepository('users').findOne({
|
|
appends: ['roles'],
|
|
});
|
|
ctx.get = function(name) {
|
|
if (name === 'X-Role') {
|
|
return 'admin';
|
|
}
|
|
};
|
|
await setCurrentRole(ctx, () => {});
|
|
expect(ctx.state.currentRole).toBe('admin');
|
|
});
|
|
|
|
it('should set role with default', async () => {
|
|
ctx.state.currentUser = await db.getRepository('users').findOne({
|
|
appends: ['roles'],
|
|
});
|
|
ctx.get = function (name) {
|
|
if (name === 'X-Role') {
|
|
return '';
|
|
}
|
|
};
|
|
await setCurrentRole(ctx, () => {});
|
|
expect(ctx.state.currentRole).toBe('root');
|
|
});
|
|
|
|
it('should set role with default when x-role does not exist', async () => {
|
|
ctx.state.currentUser = await db.getRepository('users').findOne({
|
|
appends: ['roles'],
|
|
});
|
|
ctx.get = function (name) {
|
|
if (name === 'X-Role') {
|
|
return 'abc';
|
|
}
|
|
};
|
|
await setCurrentRole(ctx, () => {});
|
|
expect(ctx.state.currentRole).toBe('root');
|
|
});
|
|
|
|
it('should set role with anonymous', async () => {
|
|
ctx.state.currentUser = await db.getRepository('users').findOne({
|
|
appends: ['roles'],
|
|
});
|
|
ctx.get = function (name) {
|
|
if (name === 'X-Role') {
|
|
return 'anonymous';
|
|
}
|
|
};
|
|
await setCurrentRole(ctx, () => {});
|
|
expect(ctx.state.currentRole).toBe('anonymous');
|
|
});
|
|
});
|