tests: Fix DBM Tests

DataDog · Dec 14, 2023 · 6710b38 · 6710b38
1 parent eff88d7
commit 6710b38
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/Integration/DatabaseMonitoringTest.php b/tests/Integration/DatabaseMonitoringTest.php
@@ -4,6 +4,7 @@
 
 use DDTrace\HookData;
 use DDTrace\Integrations\DatabaseIntegrationHelper;
+use DDTrace\Integrations\Integration;
 use DDTrace\Tests\Common\IntegrationTestCase;
 use DDTrace\Tests\Common\SpanAssertion;
 
@@ -62,7 +63,7 @@ public function testInjectionServiceMappingOnce()
         try {
             $hook = \DDTrace\install_hook(self::class . "::instrumented", function (HookData $hook) {
                 $span = $hook->span();
-                $span->service = "pdo";
+                Integration::handleInternalSpanServiceName($span, "pdo");
                 $span->name = "instrumented";
                 DatabaseIntegrationHelper::injectDatabaseIntegrationData($hook, 'mysql', 1);
             });
@@ -94,7 +95,7 @@ public function testInjectionServiceMappingTwice()
         try {
             $hook = \DDTrace\install_hook(self::class . "::instrumented", function (HookData $hook) {
                 $span = $hook->span();
-                $span->service = "pdo";
+                Integration::handleInternalSpanServiceName($span, "pdo");
                 $span->name = "instrumented";
                 DatabaseIntegrationHelper::injectDatabaseIntegrationData($hook, 'mysql', 1);
             });