Fix wordpress database setup in benchmark tests

DataDog · Sep 20, 2024 · 339adfc · 339adfc
1 parent f38dddc
commit 339adfc
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/Benchmarks/Integrations/WordPressBench.php b/tests/Benchmarks/Integrations/WordPressBench.php
@@ -13,7 +13,7 @@
 class WordPressBench extends FrameworkBenchmarksCase
 {
     /**
-     * @BeforeMethods("enableDatadog")
+     * @BeforeMethods({"enableDatadog", "createDatabase"})
      * @AfterMethods("afterMethod")
      * @Revs(10)
      * @Iterations(10)
@@ -34,12 +34,11 @@ public static function getAppIndexScript()
         return __DIR__ . '/../../Frameworks/WordPress/Version_6_1/index.php';
     }
 
-    public function disableWordPressTracing()
+    public function createDatabase(): void
     {
         $pdo = new \PDO('mysql:host=mysql_integration', 'test', 'test');
         $pdo->exec('CREATE DATABASE IF NOT EXISTS wp61');
         $pdo->exec(file_get_contents(__DIR__ . '/../../Frameworks/WordPress/Version_6_1/scripts/wp_initdb.sql'));
-        $this->disableDatadog();
     }
 
     public function afterMethod()
@@ -48,7 +47,7 @@ public function afterMethod()
     }
 
     /**
-     * @BeforeMethods("disableWordPressTracing")
+     * @BeforeMethods({"disableDatadog", "createDatabase"})
      * @AfterMethods("afterMethod")
      * @Revs(10)
      * @Iterations(10)