open-telemetry
/
opentelemetry-php
mirror of https://github.com/open-telemetry/opentelemetry-php.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
opentelemetry-php/tests/SDK/Unit/Trace/SpanBuilderTest.php

543 lines
16 KiB
PHP
Raw Blame History

<?php
declare(strict_types=1);
namespace OpenTelemetry\Tests\SDK\Unit\Trace;
use Mockery;
use Mockery\Adapter\Phpunit\MockeryTestCase;
use Mockery\MockInterface;
use OpenTelemetry\API\Trace as API;
use OpenTelemetry\API\Trace\SpanContext;
use OpenTelemetry\Context\Context;
use OpenTelemetry\SDK\Trace\Attributes;
use OpenTelemetry\SDK\Trace\Link;
use OpenTelemetry\SDK\Trace\Sampler\AlwaysOffSampler;
use OpenTelemetry\SDK\Trace\SamplerInterface;
use OpenTelemetry\SDK\Trace\SamplingResult;
use OpenTelemetry\SDK\Trace\Span;
use OpenTelemetry\SDK\Trace\SpanLimitsBuilder;
use OpenTelemetry\SDK\Trace\SpanProcessorInterface;
use OpenTelemetry\SDK\Trace\TracerProvider;
use function range;
use function str_repeat;
class SpanBuilderTest extends MockeryTestCase
{
private const SPAN_NAME = 'span_name';
private API\TracerInterface $tracer;
private API\SpanContextInterface $sampledSpanContext;
/** @var MockInterface&SpanProcessorInterface */
private $spanProcessor;
protected function setUp(): void
{
$this->spanProcessor = Mockery::spy(SpanProcessorInterface::class);
$this->tracer = (new TracerProvider($this->spanProcessor))->getTracer('SpanBuilderTest');
$this->sampledSpanContext = SpanContext::create(
'12345678876543211234567887654321',
'8765432112345678',
API\SpanContextInterface::TRACE_FLAG_SAMPLED,
);
}
public function test_addLink(): void
{
/** @var Span $span */
$span = $this
->tracer
->spanBuilder(self::SPAN_NAME)
->addLink($this->sampledSpanContext)
->addLink($this->sampledSpanContext, new Attributes())
->startSpan();
$this->assertCount(2, $span->toSpanData()->getLinks());
$span->end();
}
public function test_addLink_invalid(): void
{
/** @var Span $span */
$span = $this
->tracer
->spanBuilder(self::SPAN_NAME)
->addLink(Span::getInvalid()->getContext())
->addLink(Span::getInvalid()->getContext(), new Attributes())
->startSpan();
$this->assertEmpty($span->toSpanData()->getLinks());
$span->end();
}
public function test_addLink_droppingLinks(): void
{
$maxNumberOfLinks = 8;
$spanBuilder = (new TracerProvider([], null, null, (new SpanLimitsBuilder())->setLinkCountLimit($maxNumberOfLinks)->build()))
->getTracer('test')
->spanBuilder(self::SPAN_NAME);
for ($idx = 0; $idx < $maxNumberOfLinks * 2; $idx++) {
$spanBuilder->addLink($this->sampledSpanContext);
}
/** @var Span $span */
$span = $spanBuilder->startSpan();
$spanData = $span->toSpanData();
$links = $spanData->getLinks();
$this->assertCount($maxNumberOfLinks, $links);
$this->assertSame(8, $spanData->getTotalDroppedLinks());
for ($idx = 0; $idx < $maxNumberOfLinks; $idx++) {
$this->assertEquals(new Link($this->sampledSpanContext), $links[$idx]);
}
$span->end();
}
public function test_addLink_truncateLinkAttributes(): void
{
/** @var Span $span */
$span = (new TracerProvider([], null, null, (new SpanLimitsBuilder())->setAttributePerLinkCountLimit(1)->build()))
->getTracer('test')
->spanBuilder(self::SPAN_NAME)
->addLink(
$this->sampledSpanContext,
new Attributes([
'key0' => 0,
'key1' => 1,
'key2' => 2,
])
)
->startSpan();
$this->assertCount(1, $span->toSpanData()->getLinks());
$this->assertCount(1, $span->toSpanData()->getLinks()[0]->getAttributes());
$span->end();
}
public function test_addLink_truncateLinkAttributeValue(): void
{
$maxLength = 25;
$strVal = str_repeat('a', $maxLength);
$tooLongStrVal = "${strVal}${strVal}";
/** @var Span $span */
$span = (new TracerProvider([], null, null, (new SpanLimitsBuilder())->setAttributeValueLengthLimit($maxLength)->build()))
->getTracer('test')
->spanBuilder(self::SPAN_NAME)
->addLink(
$this->sampledSpanContext,
new Attributes([
'string' => $tooLongStrVal,
'bool' => true,
'string_array' => [$strVal, $tooLongStrVal],
'int_array' => [1, 2],
])
)
->startSpan();
$attrs = $span->toSpanData()->getLinks()[0]->getAttributes();
$this->assertSame($strVal, $attrs->get('string'));
$this->assertTrue($attrs->get('bool'));
$this->assertSame(
[$strVal, $strVal],
$attrs->get('string_array')
);
$this->assertSame(
[1, 2],
$attrs->get('int_array')
);
$span->end();
}
public function test_addLink_noEffectAfterStartSpan(): void
{
$spanBuilder = $this->tracer->spanBuilder(self::SPAN_NAME);
/** @var Span $span */
$span = $spanBuilder
->addLink($this->sampledSpanContext)
->startSpan();
$this->assertCount(1, $span->toSpanData()->getLinks());
$spanBuilder
->addLink(
SpanContext::create(
'00000000000004d20000000000001a85',
'0000000000002694',
API\SpanContextInterface::TRACE_FLAG_SAMPLED
)
);
$this->assertCount(1, $span->toSpanData()->getLinks());
$span->end();
}
public function test_setAttribute(): void
{
/** @var Span $span */
$span = $this
->tracer
->spanBuilder(self::SPAN_NAME)
->setAttribute('foo', 'bar')
->setAttribute('bar', 123)
->setAttribute('empty-arr', [])
->setAttribute('int-arr', [1, 2, 3])
->setAttribute('nil', null)
->startSpan();
$attributes = $span->toSpanData()->getAttributes();
$this->assertSame(4, $attributes->count());
$this->assertSame('bar', $attributes->get('foo'));
$this->assertSame(123, $attributes->get('bar'));
$this->assertSame([], $attributes->get('empty-arr'));
$this->assertSame([1, 2, 3], $attributes->get('int-arr'));
$this->assertNull($attributes->get('nil'));
$span->end();
}
public function test_setAttribute_afterEnd(): void
{
/** @var Span $span */
$span = $this
->tracer
->spanBuilder(self::SPAN_NAME)
->setAttribute('foo', 'bar')
->setAttribute('bar', 123)
->startSpan();
$attributes = $span->toSpanData()->getAttributes();
$this->assertSame(2, $attributes->count());
$this->assertSame('bar', $attributes->get('foo'));
$this->assertSame(123, $attributes->get('bar'));
$span->end();
}
public function test_setAttribute_dropping(): void
{
$maxNumberOfAttributes = 8;
$spanBuilder = (new TracerProvider(
null,
null,
null,
(new SpanLimitsBuilder())->setAttributeCountLimit($maxNumberOfAttributes)->build()
))->getTracer('test')->spanBuilder(self::SPAN_NAME);
foreach (range(1, $maxNumberOfAttributes * 2) as $idx) {
$spanBuilder->setAttribute("str_attribute_${idx}", $idx);
}
/** @var Span $span */
$span = $spanBuilder->startSpan();
$attributes = $span->toSpanData()->getAttributes();
$this->assertCount($maxNumberOfAttributes, $attributes);
foreach (range(1, $maxNumberOfAttributes) as $idx) {
$this->assertSame($idx, $attributes->get("str_attribute_${idx}"));
}
$span->end();
}
public function test_addAttributesViaSampler(): void
{
$sampler = new class() implements SamplerInterface {
public function shouldSample(
Context $parentContext,
string $traceId,
string $spanName,
int $spanKind,
?API\AttributesInterface $attributes = null,
array $links = []
): SamplingResult {
return new SamplingResult(SamplingResult::RECORD_AND_SAMPLE, new Attributes(['cat' => 'meow']));
}
public function getDescription(): string
{
return 'test';
}
};
/** @var Span $span */
$span = (new TracerProvider([], $sampler))->getTracer('test')->spanBuilder(self::SPAN_NAME)->startSpan();
$span->end();
$attributes = $span->toSpanData()->getAttributes();
$this->assertSame(1, $attributes->count());
$this->assertSame('meow', $attributes->get('cat'));
}
public function test_setAttributes(): void
{
$attributes = new Attributes(['id' => 1, 'foo' => 'bar']);
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->setAttributes($attributes)->startSpan();
$attributes = $span->toSpanData()->getAttributes();
$this->assertSame(2, $attributes->count());
$this->assertSame('bar', $attributes->get('foo'));
$this->assertSame(1, $attributes->get('id'));
$span->end();
}
public function test_setAttributes_overridesValues(): void
{
$attributes = new Attributes(['id' => 1, 'foo' => 'bar']);
/** @var Span $span */
$span = $this
->tracer
->spanBuilder(self::SPAN_NAME)
->setAttribute('id', 0)
->setAttribute('foo', 'baz')
->setAttributes($attributes)
->startSpan();
$attributes = $span->toSpanData()->getAttributes();
$this->assertSame(2, $attributes->count());
$this->assertSame('bar', $attributes->get('foo'));
$this->assertSame(1, $attributes->get('id'));
$span->end();
}
public function test_isRecording_default(): void
{
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->startSpan();
$this->assertTrue($span->isRecording());
$span->end();
}
public function test_isRecording_sampler(): void
{
/** @var Span $span */
$span = (new TracerProvider([], new AlwaysOffSampler()))
->getTracer('test')
->spanBuilder(self::SPAN_NAME)
->startSpan();
$this->assertFalse($span->isRecording());
$span->end();
}
public function test_getKind_default(): void
{
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->startSpan();
$this->assertSame(API\SpanKind::KIND_INTERNAL, $span->getKind());
$span->end();
}
public function test_getKind(): void
{
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->setSpanKind(API\SpanKind::KIND_CONSUMER)->startSpan();
$this->assertSame(API\SpanKind::KIND_CONSUMER, $span->getKind());
$span->end();
}
public function test_startTimestamp(): void
{
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->setStartTimestamp(123)->startSpan();
$span->end();
$this->assertSame(123, $span->toSpanData()->getStartEpochNanos());
}
public function test_setNoParent(): void
{
$parentSpan = $this->tracer->spanBuilder(self::SPAN_NAME)->startSpan();
$parentScope = $parentSpan->activate();
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->setNoParent()->startSpan();
$this->assertNotSame(
$span->getContext()->getTraceId(),
$parentSpan->getContext()->getTraceId()
);
$this
->spanProcessor
->shouldHaveReceived('onStart')
->with($span, Context::getRoot())
->once();
/** @var Span $spanNoParent */
$spanNoParent = $this
->tracer
->spanBuilder(self::SPAN_NAME)
->setNoParent()
->setParent(Context::getCurrent())
->setNoParent()
->startSpan();
$this->assertNotSame($span->getContext()->getTraceId(), $spanNoParent->getContext()->getTraceId());
$this
->spanProcessor
->shouldHaveReceived('onStart')
->with($spanNoParent, Context::getRoot())
->once();
$spanNoParent->end();
$span->end();
$parentScope->detach();
$parentSpan->end();
}
public function test_setNoParent_override(): void
{
$parentSpan = $this->tracer->spanBuilder(self::SPAN_NAME)->startSpan();
$parentContext = Context::getCurrent()->withContextValue($parentSpan);
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->setNoParent()->setParent($parentContext)->startSpan();
$this
->spanProcessor
->shouldHaveReceived('onStart')
->with($span, $parentContext)
->once();
$this->assertSame(
$span->getContext()->getTraceId(),
$parentSpan->getContext()->getTraceId()
);
$this->assertSame(
$span->toSpanData()->getParentSpanId(),
$parentSpan->getContext()->getSpanId()
);
$parentContext2 = Context::getCurrent()->withContextValue($parentSpan);
/** @var Span $span2 */
$span2 = $this
->tracer
->spanBuilder(self::SPAN_NAME)
->setNoParent()
->setParent($parentContext2)
->startSpan();
$this
->spanProcessor
->shouldHaveReceived('onStart')
->with($span2, $parentContext2)
->once();
$this->assertSame(
$span2->getContext()->getTraceId(),
$parentSpan->getContext()->getTraceId()
);
$span2->end();
$span->end();
$parentSpan->end();
}
public function test_setParent_emptyContext(): void
{
$emptyContext = Context::getCurrent();
$parentSpan = $this->tracer->spanBuilder(self::SPAN_NAME)->startSpan();
$parentScope = $parentSpan->activate();
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->setParent($emptyContext)->startSpan();
$this
->spanProcessor
->shouldHaveReceived('onStart')
->with($span, $emptyContext)
->once();
$this->assertNotSame(
$span->getContext()->getTraceId(),
$parentSpan->getContext()->getTraceId()
);
$this->assertNotSame(
$span->toSpanData()->getParentSpanId(),
$parentSpan->getContext()->getSpanId()
);
$span->end();
$parentScope->detach();
$parentSpan->end();
}
public function test_setParent_currentSpan(): void
{
$parentSpan = $this->tracer->spanBuilder(self::SPAN_NAME)->startSpan();
$parentScope = $parentSpan->activate();
$implicitContext = Context::getCurrent();
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->startSpan();
$this
->spanProcessor
->shouldHaveReceived('onStart')
->with($span, $implicitContext)
->once();
$this->assertSame(
$span->getContext()->getTraceId(),
$parentSpan->getContext()->getTraceId()
);
$this->assertSame(
$span->toSpanData()->getParentSpanId(),
$parentSpan->getContext()->getSpanId()
);
$span->end();
$parentScope->detach();
$parentSpan->end();
}
public function test_setParent_invalidContext(): void
{
$parentSpan = Span::getInvalid();
$parentContext = Context::getCurrent()->withContextValue($parentSpan);
/** @var Span $span */
$span = $this->tracer->spanBuilder(self::SPAN_NAME)->setParent($parentContext)->startSpan();
$this
->spanProcessor
->shouldHaveReceived('onStart')
->with($span, $parentContext)
->once();
$this->assertNotSame(
$span->getContext()->getTraceId(),
$parentSpan->getContext()->getTraceId()
);
$this->assertFalse(SpanContext::isValidSpanId($span->toSpanData()->getParentSpanId()));
$span->end();
$parentSpan->end();
}
}
Reference in New Issue View Git Blame Copy Permalink