[Server][Client] Bridge W3C trace context between HTTP headers and _meta (SEP-414) (#473)
RequestMeta::parseTraceContext() now falls back to the native traceparent/ tracestate/baggage HTTP headers when _meta carries none, and HeaderFactory mirrors an application-set _meta trace context onto those same headers on outbound requests. _meta wins when both are present.
This commit is contained in:
parent
1e9520e731
commit
d59950f1ac
2 changed files with 134 additions and 10 deletions
|
|
@ -81,6 +81,45 @@ final class HeaderFactoryTest extends TestCase
|
|||
$this->assertSame(['MCP-Protocol-Version' => '2026-07-28'], $headers);
|
||||
}
|
||||
|
||||
#[TestDox('mirrors trace context an application put in `_meta` onto its native headers')]
|
||||
public function testTraceContextMirrorsOntoHeaders(): void
|
||||
{
|
||||
$payload = [
|
||||
'method' => 'tools/list',
|
||||
'params' => [
|
||||
'_meta' => [
|
||||
'traceparent' => '00-0af7651916cd43dd8448eb211c80319c-00f067aa0ba902b7-01',
|
||||
'tracestate' => 'acme=1',
|
||||
],
|
||||
],
|
||||
];
|
||||
|
||||
$headers = $this->headersFor($payload);
|
||||
|
||||
$this->assertSame('00-0af7651916cd43dd8448eb211c80319c-00f067aa0ba902b7-01', $headers['traceparent']);
|
||||
$this->assertSame('acme=1', $headers['tracestate']);
|
||||
}
|
||||
|
||||
#[TestDox('a response with no method still mirrors its trace context')]
|
||||
public function testTraceContextMirrorsOntoAResponse(): void
|
||||
{
|
||||
$payload = ['id' => 1, 'result' => [], 'params' => ['_meta' => ['traceparent' => 'tp-1']]];
|
||||
|
||||
$this->assertSame(
|
||||
['MCP-Protocol-Version' => '2026-07-28', 'traceparent' => 'tp-1'],
|
||||
$this->headersFor($payload),
|
||||
);
|
||||
}
|
||||
|
||||
#[TestDox('an untraced request mirrors nothing')]
|
||||
public function testNoTraceContextMirrorsNothing(): void
|
||||
{
|
||||
$headers = $this->headersFor(['method' => 'tools/list', 'params' => []]);
|
||||
|
||||
$this->assertArrayNotHasKey('traceparent', $headers);
|
||||
$this->assertArrayNotHasKey('tracestate', $headers);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param array<string, mixed> $payload
|
||||
*
|
||||
|
|
|
|||
|
|
@ -643,6 +643,53 @@ class StatelessProtocolTest extends TestCase
|
|||
);
|
||||
}
|
||||
|
||||
#[TestDox('a native traceparent/tracestate header reaches the handler when `_meta` carries none')]
|
||||
public function testHttpTraceHeadersReachTheHandler(): void
|
||||
{
|
||||
$answer = self::call(
|
||||
self::protocol(),
|
||||
'tools/call',
|
||||
['name' => 'probe_trace', 'arguments' => []],
|
||||
[
|
||||
'Mcp-Name' => 'probe_trace',
|
||||
'traceparent' => '00-0af7651916cd43dd8448eb211c80319c-00f067aa0ba902b7-01',
|
||||
'tracestate' => 'acme=1',
|
||||
],
|
||||
);
|
||||
|
||||
$this->assertSame(
|
||||
'traceparent=00-0af7651916cd43dd8448eb211c80319c-00f067aa0ba902b7-01;tracestate=acme=1',
|
||||
$answer['body']['result']['content'][0]['text'],
|
||||
);
|
||||
}
|
||||
|
||||
#[TestDox('`_meta` trace context wins over a conflicting native header')]
|
||||
public function testMetaTraceContextWinsOverHeader(): void
|
||||
{
|
||||
$answer = self::call(
|
||||
self::protocol(),
|
||||
'tools/call',
|
||||
[
|
||||
'name' => 'probe_trace',
|
||||
'arguments' => [],
|
||||
'_meta' => [
|
||||
RequestMeta::PROTOCOL_VERSION => ProtocolVersion::V2026_07_28->value,
|
||||
RequestMeta::CLIENT_CAPABILITIES => new \stdClass(),
|
||||
'traceparent' => '00-aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa-bbbbbbbbbbbbbbbb-01',
|
||||
],
|
||||
],
|
||||
[
|
||||
'Mcp-Name' => 'probe_trace',
|
||||
'traceparent' => '00-0af7651916cd43dd8448eb211c80319c-00f067aa0ba902b7-01',
|
||||
],
|
||||
);
|
||||
|
||||
$this->assertSame(
|
||||
'traceparent=00-aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa-bbbbbbbbbbbbbbbb-01',
|
||||
$answer['body']['result']['content'][0]['text'],
|
||||
);
|
||||
}
|
||||
|
||||
#[TestDox('notifications caused by a traced request carry its trace context')]
|
||||
public function testNotificationsCarryTheTraceContext(): void
|
||||
{
|
||||
|
|
|
|||
Loading…
Reference in a new issue