|
4 | 4 |
|
5 | 5 | namespace Sentry\SentryBundle\Tests\Twig;
|
6 | 6 |
|
7 |
| -use PHPUnit\Framework\MockObject\MockObject; |
8 | 7 | use PHPUnit\Framework\TestCase;
|
| 8 | +use Sentry\Client; |
| 9 | +use Sentry\ClientInterface; |
| 10 | +use Sentry\Options; |
9 | 11 | use Sentry\SentryBundle\Twig\SentryExtension;
|
10 |
| -use Sentry\State\HubInterface; |
11 |
| -use Sentry\Tracing\DynamicSamplingContext; |
12 |
| -use Sentry\Tracing\Span; |
| 12 | +use Sentry\SentrySdk; |
| 13 | +use Sentry\State\Hub; |
| 14 | +use Sentry\State\Scope; |
| 15 | +use Sentry\Tracing\PropagationContext; |
13 | 16 | use Sentry\Tracing\SpanId;
|
14 | 17 | use Sentry\Tracing\TraceId;
|
15 | 18 | use Sentry\Tracing\Transaction;
|
16 | 19 | use Sentry\Tracing\TransactionContext;
|
| 20 | +use Sentry\Transport\NullTransport; |
17 | 21 | use Symfony\Bundle\TwigBundle\TwigBundle;
|
18 | 22 | use Twig\Environment;
|
19 | 23 | use Twig\Loader\ArrayLoader;
|
20 | 24 |
|
21 | 25 | final class SentryExtensionTest extends TestCase
|
22 | 26 | {
|
23 |
| - /** |
24 |
| - * @var MockObject&HubInterface |
25 |
| - */ |
26 |
| - private $hub; |
27 |
| - |
28 | 27 | public static function setUpBeforeClass(): void
|
29 | 28 | {
|
30 | 29 | if (!self::isTwigBundlePackageInstalled()) {
|
31 | 30 | self::markTestSkipped('This test requires the "symfony/twig-bundle" Composer package to be installed.');
|
32 | 31 | }
|
33 | 32 | }
|
34 | 33 |
|
35 |
| - protected function setUp(): void |
| 34 | + public function testTraceMetaFunctionWithNoActiveSpan(): void |
36 | 35 | {
|
37 |
| - $this->hub = $this->createMock(HubInterface::class); |
38 |
| - } |
| 36 | + $environment = new Environment(new ArrayLoader(['foo.twig' => '{{ sentry_trace_meta() }}'])); |
| 37 | + $environment->addExtension(new SentryExtension()); |
39 | 38 |
|
40 |
| - /** |
41 |
| - * @dataProvider traceMetaFunctionDataProvider |
42 |
| - */ |
43 |
| - public function testTraceMetaFunction(?Span $span, string $expectedTemplate): void |
44 |
| - { |
45 |
| - $this->hub->expects($this->once()) |
46 |
| - ->method('getSpan') |
47 |
| - ->willReturn($span); |
| 39 | + $propagationContext = PropagationContext::fromDefaults(); |
| 40 | + $propagationContext->setTraceId(new TraceId('566e3688a61d4bc888951642d6f14a19')); |
| 41 | + $propagationContext->setSpanId(new SpanId('566e3688a61d4bc8')); |
48 | 42 |
|
49 |
| - $environment = new Environment(new ArrayLoader(['foo.twig' => '{{ sentry_trace_meta() }}'])); |
50 |
| - $environment->addExtension(new SentryExtension($this->hub)); |
| 43 | + $hub = new Hub(null, new Scope($propagationContext)); |
| 44 | + |
| 45 | + SentrySdk::setCurrentHub($hub); |
51 | 46 |
|
52 |
| - $this->assertSame($expectedTemplate, $environment->render('foo.twig')); |
| 47 | + $this->assertSame('<meta name="sentry-trace" content="566e3688a61d4bc888951642d6f14a19-566e3688a61d4bc8" />', $environment->render('foo.twig')); |
53 | 48 | }
|
54 | 49 |
|
55 |
| - /** |
56 |
| - * @return \Generator<mixed> |
57 |
| - */ |
58 |
| - public function traceMetaFunctionDataProvider(): \Generator |
| 50 | + public function testTraceMetaFunctionWithActiveSpan(): void |
59 | 51 | {
|
60 |
| - yield [ |
61 |
| - null, |
62 |
| - '<meta name="sentry-trace" content="" />', |
63 |
| - ]; |
| 52 | + $environment = new Environment(new ArrayLoader(['foo.twig' => '{{ sentry_trace_meta() }}'])); |
| 53 | + $environment->addExtension(new SentryExtension()); |
| 54 | + |
| 55 | + $client = $this->createMock(ClientInterface::class); |
| 56 | + $client->expects($this->atLeastOnce()) |
| 57 | + ->method('getOptions') |
| 58 | + ->willReturn(new Options([ |
| 59 | + 'traces_sample_rate' => 1.0, |
| 60 | + 'release' => '1.0.0', |
| 61 | + 'environment' => 'development', |
| 62 | + ])); |
| 63 | + |
| 64 | + $hub = new Hub($client); |
| 65 | + |
| 66 | + SentrySdk::setCurrentHub($hub); |
64 | 67 |
|
65 | 68 | $transaction = new Transaction(new TransactionContext());
|
66 | 69 | $transaction->setTraceId(new TraceId('a3c01c41d7b94b90aee23edac90f4319'));
|
67 | 70 | $transaction->setSpanId(new SpanId('e69c2aef0ec34f2a'));
|
68 | 71 |
|
69 |
| - yield [ |
70 |
| - $transaction, |
71 |
| - '<meta name="sentry-trace" content="a3c01c41d7b94b90aee23edac90f4319-e69c2aef0ec34f2a" />', |
72 |
| - ]; |
| 72 | + $hub->setSpan($transaction); |
| 73 | + |
| 74 | + $this->assertSame('<meta name="sentry-trace" content="a3c01c41d7b94b90aee23edac90f4319-e69c2aef0ec34f2a" />', $environment->render('foo.twig')); |
73 | 75 | }
|
74 | 76 |
|
75 |
| - /** |
76 |
| - * @dataProvider baggageMetaFunctionDataProvider |
77 |
| - */ |
78 |
| - public function testBaggageMetaFunction(?Span $span, string $expectedTemplate): void |
| 77 | + public function testBaggageMetaFunctionWithNoActiveSpan(): void |
79 | 78 | {
|
80 |
| - $this->hub->expects($this->once()) |
81 |
| - ->method('getSpan') |
82 |
| - ->willReturn($span); |
83 |
| - |
84 | 79 | $environment = new Environment(new ArrayLoader(['foo.twig' => '{{ sentry_baggage_meta() }}']));
|
85 |
| - $environment->addExtension(new SentryExtension($this->hub)); |
| 80 | + $environment->addExtension(new SentryExtension()); |
| 81 | + |
| 82 | + $propagationContext = PropagationContext::fromDefaults(); |
| 83 | + $propagationContext->setTraceId(new TraceId('566e3688a61d4bc888951642d6f14a19')); |
86 | 84 |
|
87 |
| - $this->assertSame($expectedTemplate, $environment->render('foo.twig')); |
| 85 | + $client = $this->createMock(ClientInterface::class); |
| 86 | + $client->expects($this->atLeastOnce()) |
| 87 | + ->method('getOptions') |
| 88 | + ->willReturn(new Options([ |
| 89 | + 'traces_sample_rate' => 1.0, |
| 90 | + 'release' => '1.0.0', |
| 91 | + 'environment' => 'development', |
| 92 | + ])); |
| 93 | + |
| 94 | + $hub = new Hub($client, new Scope($propagationContext)); |
| 95 | + |
| 96 | + SentrySdk::setCurrentHub($hub); |
| 97 | + |
| 98 | + $this->assertSame('<meta name="baggage" content="sentry-trace_id=566e3688a61d4bc888951642d6f14a19,sentry-sample_rate=1,sentry-release=1.0.0,sentry-environment=development" />', $environment->render('foo.twig')); |
88 | 99 | }
|
89 | 100 |
|
90 |
| - /** |
91 |
| - * @return \Generator<mixed> |
92 |
| - */ |
93 |
| - public function baggageMetaFunctionDataProvider(): \Generator |
| 101 | + public function testBaggageMetaFunctionWithActiveSpan(): void |
94 | 102 | {
|
95 |
| - yield [ |
96 |
| - null, |
97 |
| - '<meta name="baggage" content="" />', |
98 |
| - ]; |
99 |
| - |
100 |
| - $samplingContext = DynamicSamplingContext::fromHeader('sentry-trace_id=566e3688a61d4bc888951642d6f14a19,sentry-public_key=public,sentry-sample_rate=1'); |
101 |
| - $transactionContext = new TransactionContext(); |
102 |
| - $transactionContext->getMetadata()->setDynamicSamplingContext($samplingContext); |
103 |
| - $transaction = new Transaction($transactionContext); |
104 |
| - |
105 |
| - yield [ |
106 |
| - $transaction, |
107 |
| - '<meta name="baggage" content="sentry-trace_id=566e3688a61d4bc888951642d6f14a19,sentry-public_key=public,sentry-sample_rate=1" />', |
108 |
| - ]; |
| 103 | + $environment = new Environment(new ArrayLoader(['foo.twig' => '{{ sentry_baggage_meta() }}'])); |
| 104 | + $environment->addExtension(new SentryExtension()); |
| 105 | + |
| 106 | + $client = new Client(new Options([ |
| 107 | + 'traces_sample_rate' => 1.0, |
| 108 | + 'release' => '1.0.0', |
| 109 | + 'environment' => 'development', |
| 110 | + ]), new NullTransport()); |
| 111 | + |
| 112 | + $hub = new Hub($client); |
| 113 | + |
| 114 | + SentrySdk::setCurrentHub($hub); |
| 115 | + |
| 116 | + $transaction = new Transaction(new TransactionContext()); |
| 117 | + $transaction->setTraceId(new TraceId('a3c01c41d7b94b90aee23edac90f4319')); |
| 118 | + |
| 119 | + $hub->setSpan($transaction); |
| 120 | + |
| 121 | + $this->assertSame('<meta name="baggage" content="sentry-trace_id=a3c01c41d7b94b90aee23edac90f4319,sentry-transaction=%3Cunlabeled%20transaction%3E,sentry-release=1.0.0,sentry-environment=development" />', $environment->render('foo.twig')); |
109 | 122 | }
|
110 | 123 |
|
111 | 124 | private static function isTwigBundlePackageInstalled(): bool
|
|
0 commit comments