Files
speckle-server/packages/server/modules/gatekeeper/tests/unit/featureAuthorization.spec.ts
T
Gergő Jedlicska 6982023dca feat(gatekeeper): add per workspace feature flags (#5303)
* feat(gatekeeper): add per workspace feature flags

* feat(workspaces): add admin api for granting and removing access to
workspace features

* fix(workspaces): use the correct constant name

* fix(workspaces): more test type fixes

* fix(shared): fix tests and types

* fix(workspaces): properly use exhaustive switch statement

* fix(workspaces): add new workspace plan feature to switch

* fix(workspaces): use regular integer, its fine for now...

* fix(workspaces): feature flag retention post checkout

* fix(gatekeeper): fix upsert plan tests
2025-08-26 10:23:02 +01:00

54 lines
2.1 KiB
TypeScript

import { canWorkspaceAccessFeatureFactory } from '@/modules/gatekeeper/services/featureAuthorization'
import { PaidWorkspacePlans, WorkspacePlanFeatures } from '@speckle/shared'
import { expect } from 'chai'
import cryptoRandomString from 'crypto-random-string'
import { buildTestWorkspacePlan } from '@/modules/gatekeeper/tests/helpers/workspacePlan'
describe('featureAuthorization @gatekeeper', () => {
describe('canWorkspaceAccessFeatureFactory creates a function, that', () => {
it('throws an error if workspace is not on a workspacePlan', async () => {
const canWorkspaceAccessFeature = canWorkspaceAccessFeatureFactory({
getWorkspacePlan: async () => null
})
const canAccess = await canWorkspaceAccessFeature({
workspaceId: cryptoRandomString({ length: 10 }),
workspaceFeature: 'domainBasedSecurityPolicies'
})
expect(canAccess).to.be.false
})
;(
[
[PaidWorkspacePlans.Team, 'canceled', WorkspacePlanFeatures.SSO, false],
[PaidWorkspacePlans.Team, 'valid', WorkspacePlanFeatures.SSO, false],
[
PaidWorkspacePlans.Team,
'valid',
WorkspacePlanFeatures.CustomDataRegion,
false
],
[PaidWorkspacePlans.Pro, 'canceled', WorkspacePlanFeatures.SSO, false],
[PaidWorkspacePlans.Pro, 'valid', WorkspacePlanFeatures.SSO, true],
[PaidWorkspacePlans.Pro, 'valid', WorkspacePlanFeatures.CustomDataRegion, true]
] as const
).forEach(([plan, status, workspaceFeature, expectedResult]) => {
it(`returns ${expectedResult} for ${plan} @ ${status} for ${workspaceFeature}`, async () => {
const workspaceId = cryptoRandomString({ length: 10 })
const canWorkspaceAccessFeature = canWorkspaceAccessFeatureFactory({
getWorkspacePlan: async () =>
buildTestWorkspacePlan({
name: plan,
status,
workspaceId
})
})
const result = await canWorkspaceAccessFeature({
workspaceId,
workspaceFeature
})
expect(result).to.equal(expectedResult)
})
})
})
})