respect max execution time limit for transitions plugin (#15652)

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/PHPUnit/Unit/DbHelperTest.php b/tests/PHPUnit/Unit/DbHelperTest.php
index 80db514d57..ddce87246d 100644
--- a/tests/PHPUnit/Unit/DbHelperTest.php
+++ b/tests/PHPUnit/Unit/DbHelperTest.php
@@ -15,6 +15,7 @@ use Piwik\DbHelper;
  * @package Piwik\Tests\Unit
  * @group Core
  * @group Core_Unit
+ * @group DbHelper
  */
 class DbHelperTest extends \PHPUnit_Framework_TestCase
 {
@@ -58,4 +59,23 @@ class DbHelperTest extends \PHPUnit_Framework_TestCase
             ),
         );
     }
+
+    /**
+     * @dataProvider getTestQueries
+     */
+    public function testAddMaxExecutionTimeHintToQuery($expected, $query, $timeLimit)
+    {
+        $result = DbHelper::addMaxExecutionTimeHintToQuery($query, $timeLimit);
+        $this->assertEquals($expected, $result);
+    }
+
+    public function getTestQueries()
+    {
+        return [
+            ['SELECT  /*+ MAX_EXECUTION_TIME(1500) */  * FROM table', 'SELECT * FROM table', 1.5],
+            ['SELECT  /*+ MAX_EXECUTION_TIME(20000) */  column FROM (SELECT * FROM table)', 'SELECT column FROM (SELECT * FROM table)', 20],
+            ['SELECT * FROM table', 'SELECT * FROM table', 0],
+            ['UPDATE table SET column = value', 'UPDATE table SET column = value', 150],
+        ];
+    }
 }