fix: catch unhandled rejections in SDK and poll tests

Author: NianJiuZst

test/polling/poll.test.ts

@@ -32,14 +32,22 @@ function jsonRes(body: unknown): Response {
 
 const originalFetch = globalThis.fetch;
 
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function setFetch(fn: any): void {
+  globalThis.fetch = fn as typeof globalThis.fetch;
+}
+
 afterEach(() => {
   globalThis.fetch = originalFetch;
 });
 
 describe('poll', () => {
   it('completes when isComplete returns true', async () => {
     let callCount = 0;
-    globalThis.fetch = mock(() => { callCount++; return Promise.resolve(jsonRes({ status: 'Success', task_id: 'task-1' })); });
+    setFetch(mock(() => {
+      callCount++;
+      return Promise.resolve(jsonRes({ status: 'Success', task_id: 'task-1' }));
+    }));
 
     const { poll } = await import('../../src/polling/poll');
     const result = await poll(baseConfig, {
@@ -55,9 +63,9 @@ describe('poll', () => {
   });
 
   it('throws on isFailed with status message', async () => {
-    globalThis.fetch = mock(() =>
-      Promise.resolve(jsonRes({ status: 'Failed', base_resp: { status_code: 1, status_msg: 'Task error' } })),
-    );
+    setFetch(mock(() => Promise.resolve(
+      jsonRes({ status: 'Failed', base_resp: { status_code: 1, status_msg: 'Task error' } }),
+    )));
 
     const { poll } = await import('../../src/polling/poll');
     await expect(
@@ -73,7 +81,7 @@ describe('poll', () => {
   });
 
   it('throws on timeout', async () => {
-    globalThis.fetch = mock(() => Promise.resolve(jsonRes({ status: 'Processing' })));
+    setFetch(mock(() => Promise.resolve(jsonRes({ status: 'Processing' }))));
 
     const { poll } = await import('../../src/polling/poll');
     await expect(
@@ -89,7 +97,10 @@ describe('poll', () => {
 
   it('returns immediately when first request succeeds', async () => {
     let callCount = 0;
-    globalThis.fetch = mock(() => { callCount++; return Promise.resolve(jsonRes({ status: 'Success' })); });
+    setFetch(mock(() => {
+      callCount++;
+      return Promise.resolve(jsonRes({ status: 'Success' }));
+    }));
 
     const { poll } = await import('../../src/polling/poll');
     const result = await poll(baseConfig, {

test/sdk/image.test.ts

@@ -146,11 +146,8 @@ describe('ImageSDK.validateParams', () => {
   });
 
   it('accepts valid dimensions (passes validation, fails on network)', async () => {
-    try {
-      await sdk.generate({ prompt: 'test', width: 1024, height: 1024 });
-    } catch (err) {
-      expect((err as Error).message).not.toContain('width');
-      expect((err as Error).message).not.toContain('height');
-    }
+    await expect(
+      sdk.generate({ prompt: 'test', width: 1024, height: 1024 }),
+    ).rejects.not.toThrow(/width|height/);
   });
 });

test/sdk/text.test.ts

@@ -83,11 +83,8 @@ describe('TextSDK.validateParams', () => {
   });
 
   it('applies defaults for model and max_tokens', async () => {
-    // Valid input passes validation → fails on network, not validation
-    try {
-      await sdk.chat({ messages: [{ role: 'user', content: 'Hi' }] });
-    } catch (err) {
-      expect((err as Error).message).not.toContain('At least one message');
-    }
+    await expect(
+      sdk.chat({ messages: [{ role: 'user', content: 'Hi' }] }),
+    ).rejects.not.toThrow('At least one message');
   });
 });

test/sdk/video.test.ts

@@ -87,29 +87,21 @@ describe('VideoSDK.validateParams', () => {
 
   it('auto-selects SEF model when last_frame_image is provided', async () => {
     // Validation passes → tries network → fails with non-validation error
-    try {
-      await sdk.generate({
+    await expect(
+      sdk.generate({
         prompt: 'test',
         first_frame_image: 'data:image/png;base64,xxx',
         last_frame_image: 'data:image/png;base64,yyy',
-      });
-    } catch (err) {
-      const msg = (err as Error).message;
-      expect(msg).not.toContain('prompt');
-      expect(msg).not.toContain('last_frame');
-    }
+      }),
+    ).rejects.not.toThrow(/prompt|last_frame/);
   });
 
   it('auto-selects S2V model when subject_reference is provided', async () => {
-    try {
-      await sdk.generate({
+    await expect(
+      sdk.generate({
         prompt: 'test',
         subject_reference: [{ type: 'character', image: ['data:image/png;base64,zzz'] }],
-      });
-    } catch (err) {
-      const msg = (err as Error).message;
-      expect(msg).not.toContain('prompt');
-      expect(msg).not.toContain('subject');
-    }
+      }),
+    ).rejects.not.toThrow(/prompt|subject/);
   });
 });