Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-12 04:18:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add workspace feature tests (#5501)

Browse Source
This commit is contained in:
DarkSky
2024-01-05 04:13:47 +00:00
parent 97f8927c21
commit 04ca554525
6 changed files with 217 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

137
packages/backend/server/tests/feature.spec.ts
View File

@@ -1,5 +1,6 @@
/// <reference types="../src/global.d.ts" />
import { Injectable } from '@nestjs/common';
import { Test, TestingModule } from '@nestjs/testing';
import { PrismaClient } from '@prisma/client';
import ava, { type TestFn } from 'ava';
@@ -14,14 +15,43 @@ import {
FeatureService,
FeatureType,
} from '../src/modules/features';
import { PrismaModule } from '../src/prisma';
import { UserType } from '../src/modules/users/types';
import { WorkspaceResolver } from '../src/modules/workspaces/resolvers';
import { Permission } from '../src/modules/workspaces/types';
import { PrismaModule, PrismaService } from '../src/prisma';
import { RateLimiterModule } from '../src/throttler';
import { initFeatureConfigs } from './utils';
@Injectable()
class WorkspaceResolverMock {
constructor(private readonly prisma: PrismaService) {}
async createWorkspace(user: UserType, _init: null) {
const workspace = await this.prisma.workspace.create({
data: {
public: false,
permissions: {
create: {
type: Permission.Owner,
user: {
connect: {
id: user.id,
},
},
accepted: true,
},
},
},
});
return workspace;
}
}
const test = ava as TestFn<{
auth: AuthService;
feature: FeatureService;
early_access: FeatureManagementService;
workspace: WorkspaceResolver;
management: FeatureManagementService;
app: TestingModule;
}>;
@@ -30,6 +60,7 @@ test.beforeEach(async () => {
const client = new PrismaClient();
await client.$connect();
await client.user.deleteMany({});
await client.workspace.deleteMany({});
await client.$disconnect();
});
@@ -55,12 +86,17 @@ test.beforeEach(async t => {
RevertCommand,
RunCommand,
],
}).compile();
providers: [WorkspaceResolver],
})
.overrideProvider(WorkspaceResolver)
.useClass(WorkspaceResolverMock)
.compile();
t.context.app = module;
t.context.auth = module.get(AuthService);
t.context.feature = module.get(FeatureService);
t.context.early_access = module.get(FeatureManagementService);
t.context.workspace = module.get(WorkspaceResolver);
t.context.management = module.get(FeatureManagementService);
// init features
await initFeatureConfigs(module);
@@ -70,7 +106,7 @@ test.afterEach.always(async t => {
await t.context.app.close();
});
test('should be able to set feature', async t => {
test('should be able to set user feature', async t => {
const { auth, feature } = t.context;
const u1 = await auth.signUp('DarkSky', 'darksky@example.org', '123456');
@@ -86,14 +122,14 @@ test('should be able to set feature', async t => {
});
test('should be able to check early access', async t => {
const { auth, feature, early_access } = t.context;
const { auth, feature, management } = t.context;
const u1 = await auth.signUp('DarkSky', 'darksky@example.org', '123456');
const f1 = await early_access.canEarlyAccess(u1.email);
const f1 = await management.canEarlyAccess(u1.email);
t.false(f1, 'should not have early access');
await early_access.addEarlyAccess(u1.id);
const f2 = await early_access.canEarlyAccess(u1.email);
await management.addEarlyAccess(u1.id);
const f2 = await management.canEarlyAccess(u1.email);
t.true(f2, 'should have early access');
const f3 = await feature.listFeatureUsers(FeatureType.EarlyAccess);
@@ -101,24 +137,24 @@ test('should be able to check early access', async t => {
t.is(f3[0].id, u1.id, 'should be the same user');
});
test('should be able revert quota', async t => {
const { auth, feature, early_access } = t.context;
test('should be able revert user feature', async t => {
const { auth, feature, management } = t.context;
const u1 = await auth.signUp('DarkSky', 'darksky@example.org', '123456');
const f1 = await early_access.canEarlyAccess(u1.email);
const f1 = await management.canEarlyAccess(u1.email);
t.false(f1, 'should not have early access');
await early_access.addEarlyAccess(u1.id);
const f2 = await early_access.canEarlyAccess(u1.email);
await management.addEarlyAccess(u1.id);
const f2 = await management.canEarlyAccess(u1.email);
t.true(f2, 'should have early access');
const q1 = await early_access.listEarlyAccess();
const q1 = await management.listEarlyAccess();
t.is(q1.length, 1, 'should have 1 user');
t.is(q1[0].id, u1.id, 'should be the same user');
await early_access.removeEarlyAccess(u1.id);
const f3 = await early_access.canEarlyAccess(u1.email);
await management.removeEarlyAccess(u1.id);
const f3 = await management.canEarlyAccess(u1.email);
t.false(f3, 'should not have early access');
const q2 = await early_access.listEarlyAccess();
const q2 = await management.listEarlyAccess();
t.is(q2.length, 0, 'should have no user');
const q3 = await feature.getUserFeatures(u1.id);
@@ -127,17 +163,72 @@ test('should be able revert quota', async t => {
t.is(q3[0].activated, false, 'should be deactivated');
});
test('should be same instance after reset the feature', async t => {
const { auth, feature, early_access } = t.context;
test('should be same instance after reset the user feature', async t => {
const { auth, feature, management } = t.context;
const u1 = await auth.signUp('DarkSky', 'darksky@example.org', '123456');
await early_access.addEarlyAccess(u1.id);
await management.addEarlyAccess(u1.id);
const f1 = (await feature.getUserFeatures(u1.id))[0];
await early_access.removeEarlyAccess(u1.id);
await management.removeEarlyAccess(u1.id);
await early_access.addEarlyAccess(u1.id);
await management.addEarlyAccess(u1.id);
const f2 = (await feature.getUserFeatures(u1.id))[1];
t.is(f1.feature, f2.feature, 'should be same instance');
});
test('should be able to set workspace feature', async t => {
const { auth, feature, workspace } = t.context;
const u1 = await auth.signUp('DarkSky', 'darksky@example.org', '123456');
const w1 = await workspace.createWorkspace(u1, null);
const f1 = await feature.getWorkspaceFeatures(w1.id);
t.is(f1.length, 0, 'should be empty');
await feature.addWorkspaceFeature(w1.id, FeatureType.Copilot, 1, 'test');
const f2 = await feature.getWorkspaceFeatures(w1.id);
t.is(f2.length, 1, 'should have 1 feature');
t.is(f2[0].feature.name, FeatureType.Copilot, 'should be copilot');
});
test('should be able to check workspace feature', async t => {
const { auth, feature, workspace, management } = t.context;
const u1 = await auth.signUp('DarkSky', 'darksky@example.org', '123456');
const w1 = await workspace.createWorkspace(u1, null);
const f1 = await management.hasWorkspaceFeature(w1.id, FeatureType.Copilot);
t.false(f1, 'should not have copilot');
await management.addWorkspaceFeatures(w1.id, FeatureType.Copilot, 1, 'test');
const f2 = await management.hasWorkspaceFeature(w1.id, FeatureType.Copilot);
t.true(f2, 'should have copilot');
const f3 = await feature.listFeatureWorkspaces(FeatureType.Copilot);
t.is(f3.length, 1, 'should have 1 workspace');
t.is(f3[0].id, w1.id, 'should be the same workspace');
});
test('should be able revert workspace feature', async t => {
const { auth, feature, workspace, management } = t.context;
const u1 = await auth.signUp('DarkSky', 'darksky@example.org', '123456');
const w1 = await workspace.createWorkspace(u1, null);
const f1 = await management.hasWorkspaceFeature(w1.id, FeatureType.Copilot);
t.false(f1, 'should not have feature');
await management.addWorkspaceFeatures(w1.id, FeatureType.Copilot, 1, 'test');
const f2 = await management.hasWorkspaceFeature(w1.id, FeatureType.Copilot);
t.true(f2, 'should have feature');
await management.removeWorkspaceFeature(w1.id, FeatureType.Copilot);
const f3 = await management.hasWorkspaceFeature(w1.id, FeatureType.Copilot);
t.false(f3, 'should not have feature');
const q3 = await feature.getWorkspaceFeatures(w1.id);
t.is(q3.length, 1, 'should have 1 feature');
t.is(q3[0].feature.name, FeatureType.Copilot, 'should be copilot');
t.is(q3[0].activated, false, 'should be deactivated');
});