From d019cbb7079e4fc1dee46e701edd3041f62e0662 Mon Sep 17 00:00:00 2001
From: Jacob Emerick <jpemeric@gmail.com>
Date: Fri, 18 Dec 2015 12:11:24 -0700
Subject: [PATCH] Adds LoggingPdo helper, adds test for pulling internal logger

---
 src/PhpQuickProfiler.php                   |  2 +-
 tests/bootstrap.php                        |  3 ++-
 tests/helpers/LoggingPdo.php               | 10 ++++++++++
 tests/{ => helpers}/function-overrides.php |  0
 tests/unit/PhpQuickProfilerTest.php        | 14 ++++++++++++--
 5 files changed, 25 insertions(+), 4 deletions(-)
 create mode 100644 tests/helpers/LoggingPdo.php
 rename tests/{ => helpers}/function-overrides.php (100%)

diff --git a/src/PhpQuickProfiler.php b/src/PhpQuickProfiler.php
index 188e24f..16cf51a 100644
--- 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
index d4ef22c..8634fc5 100644
--- 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
new file mode 100644
index 0000000..ac9277d
--- /dev/null
+++ 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 b/tests/helpers/function-overrides.php
similarity index 100%
rename from tests/function-overrides.php
rename to tests/helpers/function-overrides.php
diff --git a/tests/unit/PhpQuickProfilerTest.php b/tests/unit/PhpQuickProfilerTest.php
index e11ec35..a4c15d3 100644
--- 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
      */