test(s3): align checks with stream lifecycle

Author: xqvvu

packages/service/test/core/workflow/dispatch/ai/toolcall/toolCall.test.ts

@@ -866,8 +866,10 @@ describe('runToolCall compression node responses', () => {
       }
     ];
     runAgentLoopMock.mockImplementation(async (options) => {
-      options.onAfterToolCall({
+      options.runtime.emitEvent({
+        type: 'tool_run_end',
         call,
+        rawResponse: '[{"fileUrl":"https://files/report","filename":"report.csv"}]',
         response: '[{"fileUrl":"https://files/report","filename":"report.csv"}]',
         seconds: 0.1,
         fileRefs

projects/app/test/api/system/file/sourceContentType.test.ts

@@ -1,4 +1,4 @@
-import { Readable } from 'node:stream';
+import { PassThrough, Readable } from 'node:stream';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 import proxyDownloadHandler from '@/pages/api/system/file/download/[token]';
 import proxyUploadHandler from '@/pages/api/system/file/upload/[token]';
@@ -29,37 +29,36 @@ vi.mock('@fastgpt/service/common/s3/sources/chat', () => ({
 }));
 
 const makeMockStream = () => {
-  const stream = {
-    pipe: vi.fn(),
-    on: vi.fn(() => stream)
-  };
+  const stream = Readable.from([Buffer.from('mock file content')]);
+  vi.spyOn(stream, 'pipe');
   return stream;
 };
 
+const makeMockReq = (overrides: Record<string, unknown>) => ({
+  aborted: false,
+  once: vi.fn(),
+  off: vi.fn(),
+  ...overrides
+});
+
 const makeMockRes = () => {
   const headers: Record<string, string | number> = {};
-  const res = {
+  const res = Object.assign(new PassThrough(), {
     headers,
     statusCode: 200,
     headersSent: false,
-    writableFinished: false,
     setHeader: vi.fn((key: string, value: string | number) => {
       headers[key] = value;
     }),
     getHeader: vi.fn((key: string) => headers[key]),
-    once: vi.fn(() => res),
     status: vi.fn((statusCode: number) => {
       res.statusCode = statusCode;
       return res;
     }),
-    end: vi.fn(() => {
-      res.writableFinished = true;
-    }),
     json: vi.fn(() => {
-      res.writableFinished = true;
-    }),
-    destroy: vi.fn()
-  };
+      res.end();
+    })
+  });
   return res;
 };
 
@@ -88,19 +87,19 @@ describe('system file response content type', () => {
       type: 'download'
     });
 
-    const req = {
+    const req = makeMockReq({
       method: 'GET',
       url: '/api/system/file/download/token',
       headers: {},
       query: { token: 'token' }
-    } as any;
+    }) as any;
     const res = makeMockRes() as any;
 
     await proxyDownloadHandler(req, res);
 
     expect(res.headers['Content-Type']).toBe('text/markdown; charset=utf-8');
     expect(res.headers['Content-Length']).toBe(32);
-    expect(stream.pipe).toHaveBeenCalledWith(res);
+    expect(stream.pipe).toHaveBeenCalledWith(res, expect.any(Object));
   });
 
   it('keeps binary content type unchanged in proxy download mode', async () => {
@@ -122,12 +121,12 @@ describe('system file response content type', () => {
       type: 'download'
     });
 
-    const req = {
+    const req = makeMockReq({
       method: 'GET',
       url: '/api/system/file/download/token',
       headers: {},
       query: { token: 'token' }
-    } as any;
+    }) as any;
     const res = makeMockRes() as any;
 
     await proxyDownloadHandler(req, res);
@@ -193,25 +192,29 @@ describe('system file response content type', () => {
   it('adds utf-8 charset for text files in legacy file entry', async () => {
     const stream = makeMockStream();
     const datasetSource = {
+      bucketName: 'fastgpt-private',
       getFileStream: vi.fn().mockResolvedValue(stream),
       getFileMetadata: vi.fn().mockResolvedValue({
         filename: 'page.html',
         contentType: 'text/html',
         contentLength: 128
       })
     };
+    (global as any).s3BucketMap = {
+      'fastgpt-private': datasetSource
+    };
     vi.mocked(getS3DatasetSource).mockReturnValue(datasetSource as any);
     vi.mocked(getS3ChatSource).mockReturnValue({} as any);
     vi.mocked(verifyToken).mockResolvedValue({
       objectKey: 'dataset/team/page.html'
     });
 
-    const req = { query: { jwt: 'jwt' } } as any;
+    const req = makeMockReq({ query: { jwt: 'jwt' } }) as any;
     const res = makeMockRes() as any;
 
     await legacyFileHandler(req, res);
 
     expect(res.headers['Content-Type']).toBe('text/html; charset=utf-8');
-    expect(stream.pipe).toHaveBeenCalledWith(res);
+    expect(stream.pipe).toHaveBeenCalledWith(res, expect.any(Object));
   });
 });

projects/app/test/tsconfig.json

@@ -25,7 +25,7 @@
       "@fastgpt/*": ["../../../packages/*"],
       "@test/*": ["../../../test/*"],
 
-      "#fastgpt/app/test/*": ["./*"],
+      "#fastgpt/app/test/*": ["./*"]
     }
   },
   "include": ["**/*.test.ts", "../../../packages/service/**/*.d.ts"],

vitest.config.mts

@@ -22,8 +22,7 @@ export default defineConfig({
         process.env.FILE_TOKEN_KEY ??
         'bfd697e7e798f75deaf2d31210bc93a2e41ad4eed9e7831071d77821b7b97cff',
       AES256_SECRET_KEY: process.env.AES256_SECRET_KEY ?? 'fastgpt_test_aes256_secret_key',
-      INVOKE_TOKEN_SECRET:
-        process.env.INVOKE_TOKEN_SECRET ?? 'fastgpt_test_invoke_token_secret_32'
+      INVOKE_TOKEN_SECRET: process.env.INVOKE_TOKEN_SECRET ?? 'fastgpt_test_invoke_token_secret_32'
     },
     coverage: {
       enabled: true,