Generating commit message...

backend-node/test/agnesVideoPoll.test.js

@@ -1,6 +1,6 @@
 const { describe, it } = require('node:test');
 const assert = require('node:assert/strict');
-const { pickProxyVideoUrl } = require('../src/services/videoClient');
+const { pickProxyVideoUrl, buildAgnesPollRequest } = require('../src/services/videoClient');
 
 describe('pickProxyVideoUrl Agnes completed task', () => {
   it('reads MP4 from remixed_from_video_id when video_url is absent', () => {
@@ -17,3 +17,28 @@ describe('pickProxyVideoUrl Agnes completed task', () => {
     );
   });
 });
+
+describe('buildAgnesPollRequest', () => {
+  it('uses recommended video_id query API when provider poll id is video-prefixed', () => {
+    const req = buildAgnesPollRequest(
+      { api_key: 'k', base_url: 'https://apihub.agnes-ai.com/v1' },
+      'video:video_abc123',
+      'agnes-video-v2.0'
+    );
+    assert.equal(
+      req.url,
+      'https://apihub.agnes-ai.com/agnesapi?video_id=video_abc123&model_name=agnes-video-v2.0'
+    );
+    assert.deepEqual(req.headers, { Authorization: 'Bearer k' });
+  });
+
+  it('falls back to legacy task query API for task ids', () => {
+    const req = buildAgnesPollRequest(
+      { api_key: 'k', base_url: 'https://apihub.agnes-ai.com/v1', query_endpoint: '/videos/{taskId}' },
+      'task_abc123',
+      'agnes-video-v2.0'
+    );
+    assert.equal(req.url, 'https://apihub.agnes-ai.com/v1/videos/task_abc123');
+    assert.deepEqual(req.headers, { Authorization: 'Bearer k' });
+  });
+});