Shun Miyazawa 1 year ago
parent
commit
8a1657d468

+ 5 - 3
apps/app/src/features/openai/server/services/normalize-data/normalize-thread-relation-expired-at/normalize-thread-relation-expired-at.integ.ts

@@ -2,10 +2,12 @@ import { faker } from '@faker-js/faker';
 import { addDays, subDays } from 'date-fns';
 import { addDays, subDays } from 'date-fns';
 import { Types } from 'mongoose';
 import { Types } from 'mongoose';
 
 
+import { ThreadType } from '../../../../interfaces/thread-relation';
 import ThreadRelation from '../../../models/thread-relation';
 import ThreadRelation from '../../../models/thread-relation';
 
 
 import { MAX_DAYS_UNTIL_EXPIRATION, normalizeExpiredAtForThreadRelations } from './normalize-thread-relation-expired-at';
 import { MAX_DAYS_UNTIL_EXPIRATION, normalizeExpiredAtForThreadRelations } from './normalize-thread-relation-expired-at';
 
 
+
 describe('normalizeExpiredAtForThreadRelations', () => {
 describe('normalizeExpiredAtForThreadRelations', () => {
 
 
   it('should update expiredAt to 3 days from now for expired thread relations', async() => {
   it('should update expiredAt to 3 days from now for expired thread relations', async() => {
@@ -17,7 +19,7 @@ describe('normalizeExpiredAtForThreadRelations', () => {
       threadId: 'test-thread',
       threadId: 'test-thread',
       aiAssistant: new Types.ObjectId(),
       aiAssistant: new Types.ObjectId(),
       expiredAt: expiredDate,
       expiredAt: expiredDate,
-      isEditorAssistant: false,
+      threadType: ThreadType.KNOWLEDGE,
     });
     });
     await threadRelation.save();
     await threadRelation.save();
 
 
@@ -40,7 +42,7 @@ describe('normalizeExpiredAtForThreadRelations', () => {
       threadId: 'test-thread-2',
       threadId: 'test-thread-2',
       aiAssistant: new Types.ObjectId(),
       aiAssistant: new Types.ObjectId(),
       expiredAt: nonExpiredDate,
       expiredAt: nonExpiredDate,
-      isEditorAssistant: false,
+      threadType: ThreadType.KNOWLEDGE,
     });
     });
     await threadRelation.save();
     await threadRelation.save();
 
 
@@ -61,7 +63,7 @@ describe('normalizeExpiredAtForThreadRelations', () => {
       threadId: 'test-thread-3',
       threadId: 'test-thread-3',
       aiAssistant: new Types.ObjectId(),
       aiAssistant: new Types.ObjectId(),
       expiredAt: nonExpiredDate,
       expiredAt: nonExpiredDate,
-      isEditorAssistant: false,
+      threadType: ThreadType.KNOWLEDGE,
     });
     });
     await threadRelation.save();
     await threadRelation.save();