Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-07-17 01:56:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore(server): ignore non-exists doc embedding (#11153)

Browse Source
This commit is contained in:
darkskygit
2025-03-25 15:10:56 +00:00
parent db4406f950
commit 4bf9161e57
3 changed files with 81 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/backend/server/src/__tests__/copilot.spec.ts
+61 -41
View File
@@ -1381,19 +1381,24 @@ test('should be able to manage context', async t => {
}
// doc record
const docId = randomUUID();
await t.context.db.snapshot.create({
data: {
workspaceId: session.workspaceId,
id: docId,
blob: Buffer.from([1, 1]),
state: Buffer.from([1, 1]),
updatedAt: new Date(),
createdAt: new Date(),
},
});
const addDoc = async () => {
const docId = randomUUID();
await t.context.db.snapshot.create({
data: {
workspaceId: session.workspaceId,
id: docId,
blob: Buffer.from([1, 1]),
state: Buffer.from([1, 1]),
updatedAt: new Date(),
createdAt: new Date(),
},
});
return docId;
};
{
const docId = await addDoc();
await session.addDocRecord(docId);
const docs = session.docs.map(d => d.id);
t.deepEqual(docs, [docId], 'should list doc id');
@@ -1405,50 +1410,65 @@ test('should be able to manage context', async t => {
// tag record
{
const tagId = randomUUID();
await session.addCategoryRecord(ContextCategories.Tag, tagId, [docId]);
const tags = session.tags.map(t => t.id);
t.deepEqual(tags, [tagId], 'should list tag id');
const docId1 = await addDoc();
const docId2 = await addDoc();
{
await session.addCategoryRecord(ContextCategories.Tag, tagId, [docId1]);
const tags = session.tags.map(t => t.id);
t.deepEqual(tags, [tagId], 'should list tag id');
const docs = session.tags.flatMap(l => l.docs.map(d => d.id));
t.deepEqual(docs, [docId1], 'should list doc ids');
}
{
await session.addCategoryRecord(ContextCategories.Tag, tagId, [docId2]);
const docs = session.tags.flatMap(l => l.docs.map(d => d.id));
t.deepEqual(docs, [docId1, docId2], 'should list doc ids');
}
await session.removeCategoryRecord(ContextCategories.Tag, tagId);
t.deepEqual(session.tags, [], 'should remove tag id');
await t.throwsAsync(
session.addCategoryRecord(ContextCategories.Tag, tagId, [
'not-exists-doc',
]),
{
instanceOf: Error,
},
'should throw error if doc id not exists'
);
}
// collection record
{
const collectionId = randomUUID();
await session.addCategoryRecord(
ContextCategories.Collection,
collectionId,
[docId]
);
const collection = session.collections.map(l => l.id);
t.deepEqual(collection, [collectionId], 'should list collection id');
const docId1 = await addDoc();
const docId2 = await addDoc();
{
await session.addCategoryRecord(
ContextCategories.Collection,
collectionId,
[docId1]
);
const collection = session.collections.map(l => l.id);
t.deepEqual(collection, [collectionId], 'should list collection id');
const docs = session.collections.flatMap(l => l.docs.map(d => d.id));
t.deepEqual(docs, [docId1], 'should list doc ids');
}
{
await session.addCategoryRecord(
ContextCategories.Collection,
collectionId,
[docId2]
);
const docs = session.collections.flatMap(l => l.docs.map(d => d.id));
t.deepEqual(docs, [docId1, docId2], 'should list doc ids');
}
await session.removeCategoryRecord(
ContextCategories.Collection,
collectionId
);
t.deepEqual(session.collections, [], 'should remove collection id');
await t.throwsAsync(
session.addCategoryRecord(ContextCategories.Collection, collectionId, [
'not-exists-doc',
]),
{
instanceOf: Error,
},
'should throw error if doc id not exists'
);
}
}
});