First commit

apps/agent_frontend_a2a/tests/client.test.js

@@ -0,0 +1,86 @@
+import test from 'node:test';
+import assert from 'node:assert/strict';
+import { A2AClient, TaskView, displayResponse, parseSSE, serviceURLs, selectVersion, normalizeResult, isStopped, needsInput, RpcError } from '../a2a-client.js';
+
+function bodyFor(text, size = 1) {
+  const bytes=new TextEncoder().encode(text);
+  return new ReadableStream({start(controller){for(let i=0;i<bytes.length;i+=size)controller.enqueue(bytes.slice(i,i+size));controller.close();}});
+}
+
+test('SSE handles fragmented UTF-8, CRLF, comments and multi-line data',async()=>{
+  const text=': ping\r\nid: 1\r\nevent: message\r\ndata: {"text":\r\ndata: "Olá 👋"}\r\n\r\ndata: second\n\n';
+  const frames=[];for await(const frame of parseSSE(bodyFor(text)))frames.push(frame);
+  assert.deepEqual(frames,['{"text":\n"Olá 👋"}','second']);
+});
+test('SSE discards an unfinished frame for recovery by GetTask',async()=>{
+  const frames=[];for await(const frame of parseSSE(bodyFor('data: complete\n\ndata: truncated')))frames.push(frame);
+  assert.deepEqual(frames,['complete']);
+});
+test('service URLs accept a base or /a2a and reject embedded credentials',()=>{
+  assert.equal(serviceURLs('http://localhost:8000/a2a/').rpc,'http://localhost:8000/a2a');
+  assert.equal(serviceURLs('https://host/prefix').card,'https://host/prefix/.well-known/agent-card.json');
+  for(const url of ['javascript:alert(1)','https://user:secret@host','https://host?token=secret'])assert.throws(()=>serviceURLs(url));
+});
+test('version selection uses JSONRPC interfaces and allows explicit legacy compatibility',()=>{
+  assert.equal(selectVersion({supportedInterfaces:[{protocolBinding:'JSONRPC',protocolVersion:'1.0'}]}),'1.0');
+  assert.equal(selectVersion({protocolVersion:'0.3.0'}),'0.3');
+  assert.equal(selectVersion({},'0.3'),'0.3');
+  assert.throws(()=>selectVersion({supportedInterfaces:[{protocolBinding:'GRPC',protocolVersion:'1.0'}]}));
+});
+test('both versions normalize task, message and streaming events',()=>{
+  assert.equal(normalizeResult({statusUpdate:{taskId:'t'}}).type,'status');
+  assert.equal(normalizeResult({kind:'artifact-update',taskId:'t'}).type,'artifact');
+  assert.equal(normalizeResult({message:{messageId:'m'}}).type,'message');
+  assert.throws(()=>normalizeResult({unexpected:true}));
+});
+test('TaskView updates status, replaces snapshots and appends artifact chunks',()=>{
+  const view=new TaskView({id:'t',artifacts:[]});
+  view.apply({type:'artifact',value:{taskId:'t',artifact:{artifactId:'a',parts:[{text:'Hel'}]}}});
+  view.apply({type:'artifact',value:{taskId:'t',append:true,artifact:{artifactId:'a',parts:[{text:'lo'}]}}});
+  assert.equal(displayResponse(view.task).text,'Hello');
+  view.apply({type:'status',value:{taskId:'t',contextId:'c',status:{state:'TASK_STATE_COMPLETED'}}});
+  assert.ok(isStopped(view.task.status.state));assert.ok(needsInput('input-required'));
+  assert.throws(()=>view.apply({type:'status',value:{taskId:'other',status:{state:'working'}}}));
+});
+test('framework text/data duplicates render once and native agent metadata survives',()=>{
+  const native={text:'answer',metadata:{agent_responses:[{agent:'billing',text:'answer'}]}};
+  const task={artifacts:[{parts:[{text:'answer'},{data:native}]}],status:{message:{parts:[{text:'answer'}]}}};
+  assert.deepEqual(displayResponse(task),native);
+  const wrapped={parts:[{data:{text:'answer',metadata:{a2a_result:{task}}}}]};
+  assert.deepEqual(displayResponse(wrapped),native);
+});
+for(const legacy of [false,true]){
+  test(`send/get/cancel use correct envelopes and credentials (${legacy?'0.3':'1.0'})`,async()=>{
+    const requests=[];
+    const fetchImpl=async function(url,options){
+      assert.equal(this,globalThis);
+      const body=JSON.parse(options.body);requests.push({url,options,body});
+      return Response.json({jsonrpc:'2.0',id:body.id,result:body.method.includes('send')||body.method==='SendMessage' ? (legacy?{kind:'task',id:'t',status:{state:'working'}}:{task:{id:'t',status:{state:'TASK_STATE_WORKING'}}}):{id:'t'}});
+    };
+    const client=new A2AClient({url:'https://service',version:legacy?'0.3':'1.0',headers:{Authorization:'Bearer test'},fetchImpl});
+    await client.sendMessage('hello',{contextId:'c',agentId:'a',backendId:'billing',tenantId:'tenant',data:{business_context:{customer_key:'customer'}}});
+    await client.getTask('t');await client.cancelTask('t');
+    assert.deepEqual(requests.map(r=>r.body.method),legacy?['message/send','tasks/get','tasks/cancel']:['SendMessage','GetTask','CancelTask']);
+    const params=requests[0].body.params;
+    assert.equal(params.message.role,legacy?'user':'ROLE_USER');assert.equal(params.message.metadata.backend_id,'billing');
+    assert.equal(params.message.metadata.agent_id,'a');assert.equal(params.message.contextId,'c');
+    assert.equal(legacy?params.configuration.blocking:params.configuration.returnImmediately,!legacy);
+    assert.equal(requests[0].options.headers.Authorization,'Bearer test');
+    assert.equal(requests[0].options.credentials,'omit');
+  });
+}
+test('JSON-RPC errors and mismatched response IDs never become successful responses',async()=>{
+  const bad=new A2AClient({url:'https://service',fetchImpl:async()=>Response.json({jsonrpc:'2.0',id:'wrong',result:{}})});
+  await assert.rejects(bad.getTask('t'),/JSON-RPC/);
+  const failure=new A2AClient({url:'https://service',fetchImpl:async(url,o)=>Response.json({jsonrpc:'2.0',id:JSON.parse(o.body).id,error:{code:-32001,message:'Task missing'}})});
+  await assert.rejects(failure.getTask('t'),RpcError);
+});
+test('POST SSE supports auth headers, validates envelopes and streams canonical events',async()=>{
+  const client=new A2AClient({url:'https://service',headers:{'x-api-key':'test'},fetchImpl:async(url,o)=>{
+    assert.equal(o.method,'POST');assert.equal(o.headers['x-api-key'],'test');
+    const body=JSON.parse(o.body);assert.equal(body.method,'SubscribeToTask');
+    return new Response(bodyFor(`data: ${JSON.stringify({jsonrpc:'2.0',id:body.id,result:{statusUpdate:{taskId:'t',status:{state:'TASK_STATE_COMPLETED'}}}})}\n\n`),{headers:{'content-type':'text/event-stream'}});
+  }});
+  const events=[];for await(const event of client.subscribe('t'))events.push(event);
+  assert.equal(events[0].type,'status');assert.equal(events[0].value.taskId,'t');
+});

apps/agent_frontend_a2a/tests/integration.test.js

@@ -0,0 +1,32 @@
+// Run against examples/a2a_frontend_demo.py with A2A_DEMO_URL=http://127.0.0.1:8765.
+import test from 'node:test';
+import assert from 'node:assert/strict';
+import { A2AClient, TaskView, displayResponse, isStopped, stateName } from '../a2a-client.js';
+const url=process.env.A2A_DEMO_URL;
+const delay=ms=>new Promise(resolve=>setTimeout(resolve,ms));
+async function complete(client,task){
+  const deadline=Date.now()+6000;
+  while(!isStopped(task.status?.state)&&Date.now()<deadline){await delay(50);task=await client.getTask(task.id);}
+  assert.ok(isStopped(task.status?.state));return task;
+}
+for(const version of ['1.0','0.3']){
+  test(`SDK integration: submit, subscribe, poll and resume (${version})`,{skip:!url},async()=>{
+    const client=new A2AClient({url,version});const card=await client.discover();assert.ok(card.capabilities.streaming);
+    const first=await client.sendMessage('hello',{contextId:crypto.randomUUID()});assert.equal(first.type,'task');assert.ok(!isStopped(first.value.status.state));
+    const view=new TaskView(first.value);let count=0;
+    for await(const event of client.subscribe(first.value.id)){view.apply(event);count++;}
+    assert.ok(count>0);const final=await client.getTask(first.value.id);
+    assert.equal(stateName(final.status.state),'completed');assert.match(displayResponse(final).text,/hello/);
+    const prompt=await client.sendMessage('/confirmar',{contextId:crypto.randomUUID()});
+    const waiting=await complete(client,prompt.value);assert.equal(stateName(waiting.status.state),'input-required');
+    const resumed=await client.sendMessage('sim',{contextId:waiting.contextId,taskId:waiting.id});
+    const finished=await complete(client,resumed.value);assert.equal(finished.id,waiting.id);assert.equal(stateName(finished.status.state),'completed');
+  });
+}
+test('SDK integration: cancellation and failed tasks',{skip:!url},async()=>{
+  const client=new A2AClient({url});
+  const slow=await client.sendMessage('/lento',{contextId:crypto.randomUUID()});
+  await delay(50);const canceled=await client.cancelTask(slow.value.id);assert.equal(stateName(canceled.status.state),'canceled');
+  const failed=await client.sendMessage('/erro',{contextId:crypto.randomUUID()});
+  assert.equal(stateName((await complete(client,failed.value)).status.state),'failed');
+});