Page MenuHomeWickedGov Phorge
Files F4130689

TelemetryTest.php
No OneTemporary
Actions

Size
2 KB
Referenced Files
None
Subscribers
None

TelemetryTest.php
View Options

<?php
namespace MediaWiki\Tests\Unit\Http;
use MediaWiki\Http\Telemetry;
use MediaWikiUnitTestCase;
/**
* @covers \MediaWiki\Http\Telemetry
*/
class TelemetryTest extends MediaWikiUnitTestCase {
public function testItOverridesTheRequestIdWhenAsked() {
$newId = 'new_id';
$sut = new Telemetry( [ 'UNIQUE_ID' => 'test' ], false );
$sut->overrideRequestId( $newId );
$this->assertSame( $newId, $sut->getRequestId() );
}
public function testRegenerationProvidesNewId() {
$sut = new Telemetry( [], false );
$firstId = $sut->getRequestId();
$sut->regenerateRequestId();
$secondId = $sut->getRequestId();
$this->assertNotSame( $firstId, $secondId );
}
public function testItReturnsOpenTelemetryProperties() {
$sut = new Telemetry( [
'HTTP_TRACESTATE' => 'val1',
'HTTP_TRACEPARENT' => 'val2'
], true );
$this->assertSame( 'val1', $sut->getTracestate() );
$this->assertSame( 'val2', $sut->getTraceparent() );
}
public function testOpenTelemetryPropertiesAreSkippedWhenAllowExternalReqIDIsSetToFalse() {
$sut = new Telemetry( [
'HTTP_TRACESTATE' => 'val1',
'HTTP_TRACEPARENT' => 'val2'
], false );
$this->assertNull( $sut->getTracestate() );
$this->assertNull( $sut->getTraceparent() );
}
/**
* @dataProvider provideRequestHeaders
*/
public function testItHandlesRequestIdHeaders( $allowExternalReqId, $server, $expected ) {
$sut = new Telemetry( $server, $allowExternalReqId );
$this->assertSame( $expected, $sut->getRequestId() );
}
public function testItGeneratesRequestIdWhenHeadersNotPresentAndExternalReqIdIsSet() {
$sut = new Telemetry( [], true );
$this->assertNotEmpty( $sut->getRequestId() );
}
public function testItGeneratesRequestIdWhenHeadersNotPresentAndExternalReqIdIsNotSet() {
$sut = new Telemetry( [], false );
$this->assertNotEmpty( $sut->getRequestId() );
}
public function testGetRequestHeadersReturnsAllHeaders() {
$allSet = [ 'HTTP_TRACEPARENT' => 'value1', 'HTTP_TRACESTATE' => 'value2' ];
$sut = new Telemetry( $allSet, true );
$headers = $sut->getRequestHeaders();
$this->assertCount( 3, $headers );
$this->assertArrayHasKey( 'X-Request-Id', $headers );
$this->assertSame( 'value1', $headers['traceparent'] );
$this->assertSame( 'value2', $headers['tracestate'] );
}
public function testGetRequestHeadersDoesntReturnEmptyHeaders() {
$sut = new Telemetry( [], true );
$headers = $sut->getRequestHeaders();
$this->assertCount( 1, $headers );
$this->assertArrayHasKey( 'X-Request-Id', $headers );
}
public static function provideRequestHeaders() {
yield [
false,
[ 'UNIQUE_ID' => 'unique_hash' ],
'unique_hash'
];
yield [
false,
[
'HTTP_X_REQUEST_ID' => 'request_id',
'UNIQUE_ID' => 'unique_hash'
],
'unique_hash'
];
yield [
true,
[
'HTTP_X_REQUEST_ID' => 'request_id',
'UNIQUE_ID' => 'unique_hash'
],
'request_id'
];
yield [
true,
[
'UNIQUE_ID' => 'unique_hash'
],
'unique_hash'
];
}
}

File Metadata

Mime Type
text/x-php
Expires
Wed, Aug 19, 10:31 (2 w, 6 d ago)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
73/77/0c77a1af1ef1b5c51e44f3085a24
Default Alt Text
TelemetryTest.php (2 KB)

Event Timeline