mirror of
https://github.com/labring/FastGPT.git
synced 2026-01-16 04:01:55 +08:00
V4.14.4 features (#6036)
* feat: add query optimize and bill (#6021) * add query optimize and bill * perf: query extension * fix: embe model * remove log * remove log * fix: test --------- Co-authored-by: xxyyh <2289112474@qq> Co-authored-by: archer <545436317@qq.com> * feat: notice (#6013) * feat: record user's language * feat: notice points/dataset indexes; support count limit; update docker-compose.yml * fix: ts error * feat: send auth code i18n * chore: dataset notice limit * chore: adjust * fix: ts * fix: countLimit race condition; i18n en-prefix locale fallback to en --------- Co-authored-by: archer <545436317@qq.com> * perf: comment * perf: send inform code * fix: type error (#6029) * feat: add ip region for chat logs (#6010) * feat: add ip region for chat logs * refactor: use Geolite2.mmdb * fix: export chat logs * fix: return location directly * test: add unit test * perf: log show ip data * adjust commercial plans (#6008) * plan frontend * plan limit * coupon * discount coupon * fix * type * fix audit * type * plan name * legacy plan * track * feat: add discount coupon * fix * fix discount coupon * openapi * type * type * env * api type * fix * fix: simple agent plugin input & agent dashboard card (#6034) * refactor: remove gridfs (#6031) * fix: replace gridfs multer operations with s3 compatible ops * wip: s3 features * refactor: remove gridfs * fix * perf: mock test * doc * doc * doc * fix: test * fix: s3 * fix: mock s3 * remove invalid config * fix: init query extension * initv4144 (#6037) * chore: initv4144 * fix * version * fix: new plans (#6039) * fix: new plans * qr modal tip * fix: buffer raw text filename (#6040) * fix: initv4144 (#6041) * fix: pay refresh (#6042) * fix: migration shell * rename collection * clear timerlock * clear timerlock * perf: faq * perf: bill schema * fix: openapi * doc * fix: share var render * feat: delete dataset queue * plan usage display (#6043) * plan usage display * text * fix * fix: ts * perf: remove invalid code * perf: init shell * doc * perf: rename field * perf: avatar presign * init * custom plan text (#6045) * fix plans * fix * fixed * computed --------- Co-authored-by: archer <545436317@qq.com> * init shell * plan text & price page back button (#6046) * init * index * delete dataset * delete dataset * perf: delete dataset * init --------- Co-authored-by: YeYuheng <57035043+YYH211@users.noreply.github.com> Co-authored-by: xxyyh <2289112474@qq> Co-authored-by: Finley Ge <32237950+FinleyGe@users.noreply.github.com> Co-authored-by: Roy <whoeverimf5@gmail.com> Co-authored-by: heheer <heheer@sealos.io>
This commit is contained in:
@@ -5,57 +5,15 @@ import { initGlobalVariables } from '@/service/common/system';
|
||||
import { afterAll, beforeAll, beforeEach, inject, onTestFinished, vi } from 'vitest';
|
||||
import setupModels from './setupModels';
|
||||
import { clean } from './datas/users';
|
||||
import type { Mongoose } from '@fastgpt/service/common/mongo';
|
||||
import { connectionLogMongo, connectionMongo } from '@fastgpt/service/common/mongo';
|
||||
import { randomUUID } from 'crypto';
|
||||
import { delay } from '@fastgpt/global/common/system/utils';
|
||||
|
||||
vi.stubEnv('NODE_ENV', 'test');
|
||||
|
||||
vi.mock(import('@fastgpt/service/common/mongo/init'), async (importOriginal: any) => {
|
||||
const mod = await importOriginal();
|
||||
return {
|
||||
...mod,
|
||||
connectMongo: async (db: Mongoose, url: string) => {
|
||||
await db.connect(url, { dbName: randomUUID() });
|
||||
await db.connection.db?.dropDatabase();
|
||||
}
|
||||
};
|
||||
});
|
||||
|
||||
vi.mock(import('@/service/common/system'), async (importOriginal) => {
|
||||
const mod = await importOriginal();
|
||||
return {
|
||||
...mod,
|
||||
getSystemVersion: async () => {
|
||||
return '0.0.0';
|
||||
},
|
||||
readConfigData: async () => {
|
||||
return readFileSync('projects/app/data/config.json', 'utf-8');
|
||||
},
|
||||
initSystemConfig: async () => {
|
||||
// read env from projects/app/.env
|
||||
|
||||
const str = readFileSync('projects/app/.env.local', 'utf-8');
|
||||
const lines = str.split('\n');
|
||||
const systemEnv: Record<string, string> = {};
|
||||
for (const line of lines) {
|
||||
const [key, value] = line.split('=');
|
||||
if (key && value) {
|
||||
systemEnv[key] = value;
|
||||
}
|
||||
}
|
||||
global.systemEnv = systemEnv as any;
|
||||
|
||||
return;
|
||||
}
|
||||
};
|
||||
});
|
||||
|
||||
beforeAll(async () => {
|
||||
vi.stubEnv('MONGODB_URI', inject('MONGODB_URI'));
|
||||
await connectMongo(connectionMongo, inject('MONGODB_URI'));
|
||||
await connectMongo(connectionLogMongo, inject('MONGODB_URI'));
|
||||
await connectMongo({ db: connectionMongo, url: inject('MONGODB_URI') });
|
||||
await connectMongo({ db: connectionLogMongo, url: inject('MONGODB_URI') });
|
||||
|
||||
initGlobalVariables();
|
||||
global.systemEnv = {} as any;
|
||||
@@ -86,16 +44,27 @@ afterAll(async () => {
|
||||
});
|
||||
|
||||
beforeEach(async () => {
|
||||
await connectMongo(connectionMongo, inject('MONGODB_URI'));
|
||||
await connectMongo(connectionLogMongo, inject('MONGODB_URI'));
|
||||
await connectMongo({ db: connectionMongo, url: inject('MONGODB_URI') });
|
||||
await connectMongo({ db: connectionLogMongo, url: inject('MONGODB_URI') });
|
||||
|
||||
onTestFinished(async () => {
|
||||
clean();
|
||||
await delay(200); // wait for asynchronous operations to complete
|
||||
await Promise.all([
|
||||
connectionMongo?.connection.db?.dropDatabase(),
|
||||
connectionLogMongo?.connection.db?.dropDatabase()
|
||||
]);
|
||||
// Wait for any ongoing transactions and operations to complete
|
||||
await delay(500);
|
||||
|
||||
// Ensure all sessions are closed before dropping database
|
||||
try {
|
||||
await Promise.all([
|
||||
connectionMongo?.connection.db?.dropDatabase(),
|
||||
connectionLogMongo?.connection.db?.dropDatabase()
|
||||
]);
|
||||
} catch (error) {
|
||||
// Ignore errors during cleanup
|
||||
console.warn('Error during test cleanup:', error);
|
||||
}
|
||||
|
||||
// Additional delay to prevent lock contention between tests
|
||||
await delay(100);
|
||||
});
|
||||
});
|
||||
|
||||
|
||||
Reference in New Issue
Block a user