Adds LoggingPdo helper, adds test for pulling internal logger

jacobemerick · Dec 18, 2015 · d019cbb · d019cbb
1 parent 0d805c2
commit d019cbb
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 4 deletions.
diff --git a/src/PhpQuickProfiler.php b/src/PhpQuickProfiler.php
@@ -17,7 +17,7 @@
 class PhpQuickProfiler
 {
 
-    /** @var  integer */
+    /** @var  double */
     protected $startTime;
 
     /** @var  Console */

diff --git a/tests/bootstrap.php b/tests/bootstrap.php
@@ -1,4 +1,5 @@
 <?php
 
 require_once __DIR__ . '/../vendor/autoload.php';
-require_once __DIR__ . '/function-overrides.php';
+require_once __DIR__ . '/helpers/function-overrides.php';
+require_once __DIR__ . '/helpers/LoggingPdo.php';
diff --git a/tests/helpers/LoggingPdo.php b/tests/helpers/LoggingPdo.php
@@ -0,0 +1,10 @@
+<?php
+
+namespace Particletree\Pqp;
+
+use PDO;
+
+class LoggingPdo extends PDO
+{
+    public $queries = array();
+}
diff --git a/tests/function-overrides.php → tests/helpers/function-overrides.php b/tests/function-overrides.php → tests/helpers/function-overrides.php
diff --git a/tests/unit/PhpQuickProfilerTest.php b/tests/unit/PhpQuickProfilerTest.php
@@ -2,7 +2,6 @@
 
 namespace Particletree\Pqp;
 
-use PDO;
 use PHPUnit_Framework_Testcase;
 use ReflectionClass;
 use ReflectionMethod;
@@ -14,7 +13,7 @@ class PhpQuickProfilerTest extends PHPUnit_Framework_TestCase
 
     public static function setUpBeforeClass()
     {
-        self::$dbConnection = new PDO('sqlite::memory:');
+        self::$dbConnection = new LoggingPdo('sqlite::memory:');
         $createTable = "
             CREATE TABLE IF NOT EXISTS `testing` (
                 `id` integer PRIMARY KEY AUTOINCREMENT,
@@ -131,6 +130,17 @@ public function testGatherQueryData()
         }
     }
 
+    public function testGatherQueryDataInternalProfiler()
+    {
+        $profiledQueries = $this->dataProfiledQueries();
+        $dbConnection = self::$dbConnection;
+        $dbConnection->queries = $profiledQueries;
+        $profiler = new PhpQuickProfiler();
+        $profiler->gatherQueryData($dbConnection);
+
+        $this->assertAttributeSame($profiledQueries, 'profiledQueries', $profiler);
+    }
+
     /**
      * @dataProvider dataProfiledQueries
      */