diff --git a/packages/cli/test/integration/shared/license.ts b/packages/cli/test/integration/shared/license.ts index a343879afe..3f9e736153 100644 --- a/packages/cli/test/integration/shared/license.ts +++ b/packages/cli/test/integration/shared/license.ts @@ -3,7 +3,7 @@ import type { License } from '@/License'; export interface LicenseMockDefaults { features?: BooleanLicenseFeature[]; - quota?: Partial<{ [K in NumericLicenseFeature]: number }>; + quotas?: Partial<{ [K in NumericLicenseFeature]: number }>; } export class LicenseMocker { @@ -28,7 +28,7 @@ export class LicenseMocker { setDefaults(defaults: LicenseMockDefaults) { this._defaultFeatures = new Set(defaults.features ?? []); this._defaultQuotas = new Map( - Object.entries(defaults.quota ?? {}) as Array<[NumericLicenseFeature, number]>, + Object.entries(defaults.quotas ?? {}) as Array<[NumericLicenseFeature, number]>, ); } diff --git a/packages/cli/test/integration/shared/types.ts b/packages/cli/test/integration/shared/types.ts index a92991ffa2..ff93f11632 100644 --- a/packages/cli/test/integration/shared/types.ts +++ b/packages/cli/test/integration/shared/types.ts @@ -5,7 +5,7 @@ import type { Server } from 'http'; import type { CredentialsEntity } from '@db/entities/CredentialsEntity'; import type { User } from '@db/entities/User'; -import type { BooleanLicenseFeature, ICredentialsDb } from '@/Interfaces'; +import type { BooleanLicenseFeature, ICredentialsDb, NumericLicenseFeature } from '@/Interfaces'; import type { LicenseMocker } from './license'; type EndpointGroup = @@ -38,6 +38,7 @@ export interface SetupProps { applyAuth?: boolean; endpointGroups?: EndpointGroup[]; enabledFeatures?: BooleanLicenseFeature[]; + quotas?: Partial<{ [K in NumericLicenseFeature]: number }>; } export interface TestServer { diff --git a/packages/cli/test/integration/shared/utils/testServer.ts b/packages/cli/test/integration/shared/utils/testServer.ts index 3f250af23f..0de66cedab 100644 --- a/packages/cli/test/integration/shared/utils/testServer.ts +++ b/packages/cli/test/integration/shared/utils/testServer.ts @@ -68,6 +68,7 @@ export const setupTestServer = ({ endpointGroups, applyAuth = true, enabledFeatures, + quotas, }: SetupProps): TestServer => { const app = express(); app.use(rawBodyReader); @@ -97,6 +98,7 @@ export const setupTestServer = ({ if (enabledFeatures) { testServer.license.setDefaults({ features: enabledFeatures, + quotas, }); } diff --git a/packages/cli/test/integration/users.api.test.ts b/packages/cli/test/integration/users.api.test.ts index a6a05481e0..4e18f13d85 100644 --- a/packages/cli/test/integration/users.api.test.ts +++ b/packages/cli/test/integration/users.api.test.ts @@ -18,14 +18,10 @@ import * as testDb from './shared/testDb'; import type { SuperAgentTest } from 'supertest'; import type { Role } from '@db/entities/Role'; import type { User } from '@db/entities/User'; -import { License } from '@/License'; -import { mockInstance } from '../shared/mocking'; -const testServer = utils.setupTestServer({ endpointGroups: ['users'] }); - -const license = mockInstance(License, { - isAdvancedPermissionsLicensed: jest.fn().mockReturnValue(true), - isWithinUsersLimit: jest.fn().mockReturnValue(true), +const testServer = utils.setupTestServer({ + endpointGroups: ['users'], + enabledFeatures: ['feat:advancedPermissions'], }); describe('GET /users', () => { @@ -528,7 +524,7 @@ describe('PATCH /users/:id/role', () => { }); test('should fail to promote member to admin if not licensed', async () => { - license.isAdvancedPermissionsLicensed.mockReturnValueOnce(false); + testServer.license.disable('feat:advancedPermissions'); const response = await adminAgent.patch(`/users/${member.id}/role`).send({ newRole: { scope: 'global', name: 'admin' }, @@ -634,7 +630,7 @@ describe('PATCH /users/:id/role', () => { }); test('should fail to promote member to admin if not licensed', async () => { - license.isAdvancedPermissionsLicensed.mockReturnValueOnce(false); + testServer.license.disable('feat:advancedPermissions'); const response = await ownerAgent.patch(`/users/${member.id}/role`).send({ newRole: { scope: 'global', name: 'admin' },