From 96857dd6ef639fc0a6d272e8c7d6d831d1abaf9a Mon Sep 17 00:00:00 2001
From: Alejandro Estringana Ruiz <alejandro.estringanaruiz@datadoghq.com>
Date: Thu, 25 Jul 2024 12:42:06 +0200
Subject: [PATCH] Fix user table on wordpress 4.8

---
 .../WordPress/AutomatedLoginEventsTestSuite.php            | 7 ++++---
 .../WordPress/V4_8/AutomatedLoginEventsTest.php            | 2 ++
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/Integrations/WordPress/AutomatedLoginEventsTestSuite.php b/tests/Integrations/WordPress/AutomatedLoginEventsTestSuite.php
index 2620f2e220b..54e04c3b0d7 100644
--- a/tests/Integrations/WordPress/AutomatedLoginEventsTestSuite.php
+++ b/tests/Integrations/WordPress/AutomatedLoginEventsTestSuite.php
@@ -12,6 +12,7 @@
  */
 class AutomatedLoginEventsTestSuite extends AppsecTestCase
 {
+    protected $users_table = 'wp55_users';
     protected function ddSetUp()
     {
         parent::ddSetUp();
@@ -27,7 +28,7 @@ public function testUserLoginSuccessEvent()
         $name = 'some name';
         //Password is test
         $this->connection()->exec(
-            'INSERT INTO wp55_users VALUES ('.$id.',"test","$P$BDzpK1XXL9P2cYWggPMUbN87GQSiI80","test","'.$email.'","","2020-10-22 16:31:15","",0,"'.$name.'")'
+            'INSERT INTO '.$this->users_table.' VALUES ('.$id.',"test","$P$BDzpK1XXL9P2cYWggPMUbN87GQSiI80","test","'.$email.'","","2020-10-22 16:31:15","",0,"'.$name.'")'
         );
 
         $spec = PostSpec::create('request', '/wp-login.php', [
@@ -70,7 +71,7 @@ public function testUserLoginFailureEventWhenUserDoesExists()
         $name = 'some name';
         //Password is test
         $this->connection()->exec(
-            'INSERT INTO wp55_users VALUES ('.$id.',"test","$P$BDzpK1XXL9P2cYWggPMUbN87GQSiI80","test","'.$email.'","","2020-10-22 16:31:15","",0,"'.$name.'")'
+            'INSERT INTO '.$this->users_table.' VALUES ('.$id.',"test","$P$BDzpK1XXL9P2cYWggPMUbN87GQSiI80","test","'.$email.'","","2020-10-22 16:31:15","",0,"'.$name.'")'
         );
 
         $spec = PostSpec::create('request', '/wp-login.php', [
@@ -98,7 +99,7 @@ public function testUserSignUp()
            ], "user_login=$username&user_email=$email&wp-submit=Register&redirect_to=")
        );
 
-       $users = $this->connection()->query("SELECT * FROM wp55_users where user_email='".$email."'")->fetchAll();
+       $users = $this->connection()->query("SELECT * FROM ".$this->users_table." where user_email='".$email."'")->fetchAll();
 
        $this->assertEquals(1, count($users));
 
diff --git a/tests/Integrations/WordPress/V4_8/AutomatedLoginEventsTest.php b/tests/Integrations/WordPress/V4_8/AutomatedLoginEventsTest.php
index eec0d008446..89b60e8978e 100644
--- a/tests/Integrations/WordPress/V4_8/AutomatedLoginEventsTest.php
+++ b/tests/Integrations/WordPress/V4_8/AutomatedLoginEventsTest.php
@@ -9,6 +9,8 @@
  */
 class AutomatedLoginEventsTest extends AutomatedLoginEventsTestSuite
 {
+    protected $users_table = 'wp_users';
+
     protected static function getAppIndexScript()
     {
         return __DIR__ . '/../../../Frameworks/WordPress/Version_4_8/index.php';