1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
<?php
namespace Jaeger\Tests\Sender;
use Jaeger\Sender\UdpSender;
use Jaeger\Span;
use Jaeger\SpanContext;
use Jaeger\Thrift\Agent\AgentClient;
use Jaeger\Thrift\Agent\Zipkin\Annotation as ZipkinAnnotation;
use Jaeger\Thrift\Agent\Zipkin\Span as ZipkinSpan;
use Jaeger\Tracer;
use PHPUnit\Framework\TestCase;
class UdpSenderTest extends TestCase
{
/**
* @var UdpSender
*/
private $sender;
/**
* @var AgentClient
*/
private $client;
public function setUp(): void
{
$this->client = $this->createMock(AgentClient::class);
$this->sender = new UdpSender($this->client, 64000);
}
public function testMaxBufferLength(): void
{
$tracer = $this->createMock(Tracer::class);
$tracer->method('getIpAddress')->willReturn('');
$tracer->method('getServiceName')->willReturn('');
$context = $this->createMock(SpanContext::class);
$span = $this->createMock(Span::class);
$span->method('getOperationName')->willReturn('dummy-operation');
$span->method('getTracer')->willReturn($tracer);
$span->method('getContext')->willReturn($context);
$sender = new UdpSender($this->client, 100);
$this->client
->expects(self::exactly(2))
->method('emitZipkinBatch')
->withConsecutive(
[self::countOf(2)],
[self::countOf(1)]
);
// one span has a length of ~25
$sender->append($span); // 30 + 25 < 100 - chunk 1
$sender->append($span); // 30 + 25 * 2 < 100 - chunk 1
$sender->append($span); // 30 + 25 * 3 > 100 - chunk 2
self::assertEquals(3, $sender->flush());
}
public function testFlush(): void
{
$this->assertEquals(0, $this->sender->flush());
$logTimeStamp = (int) (microtime(true) * 1000000);
$tracer = $this->createMock(Tracer::class);
$tracer->method('getIpAddress')->willReturn('');
$tracer->method('getServiceName')->willReturn('');
$context = $this->createMock(SpanContext::class);
$span = $this->createMock(Span::class);
$span->method('getTracer')->willReturn($tracer);
$span->method('getContext')->willReturn($context);
$span
->expects($this->atLeastOnce())
->method('getLogs')
->willReturn([
[
'timestamp' => $logTimeStamp,
'fields' => [
'foo' => 'bar',
],
],
]);
$this->client
->expects($this->once())
->method('emitZipkinBatch')
->with($this->callback(function ($spans) use ($logTimeStamp) {
$this->assertCount(1, $spans);
/* @var $annotation ZipkinSpan */
$span = $spans[0];
$this->assertInstanceOf(ZipkinSpan::class, $span);
$this->assertCount(1, $span->annotations);
/* @var $annotation ZipkinAnnotation */
$annotation = $span->annotations[0];
$this->assertInstanceOf(ZipkinAnnotation::class, $annotation);
$this->assertSame($logTimeStamp, $annotation->timestamp);
$this->assertSame(
json_encode([
'foo' => 'bar',
]),
$annotation->value
);
return true;
}));
$this->sender->append($span);
$this->assertEquals(1, $this->sender->flush());
}
}
|
