jacobemerick / pqp

tests/unit/PhpQuickProfilerTest.php

Indentation +315 added lines, -315 removed lines

@@ -9,19 +9,19 @@ 
 class PhpQuickProfilerTest extends PHPUnit_Framework_TestCase
 {
 
-    protected static $dbConnection;
+	protected static $dbConnection;
 
-    public static function setUpBeforeClass()
-    {
-        self::$dbConnection = new LoggingPdo('sqlite::memory:');
-        $createTable = "
+	public static function setUpBeforeClass()
+	{
+		self::$dbConnection = new LoggingPdo('sqlite::memory:');
+		$createTable = "
             CREATE TABLE IF NOT EXISTS `testing` (
                 `id` integer PRIMARY KEY AUTOINCREMENT,
                 `title` varchar(60) NOT NULL
             );";
-        self::$dbConnection->exec($createTable);
+		self::$dbConnection->exec($createTable);
 
-        $hydrateTable = "
+		$hydrateTable = "
             INSERT INTO `testing`
                 (`title`)
             VALUES
@@ -29,312 +29,312 @@ 
                 ('beta'),
                 ('charlie'),
                 ('delta');";
-        self::$dbConnection->exec($hydrateTable);
-    }
-
-    public function testConstruct()
-    {
-        $startTime = microtime(true);
-
-        $profiler = new PhpQuickProfiler();
-        $this->assertAttributeEquals($startTime, 'startTime', $profiler);
-
-        $profiler = new PhpQuickProfiler($startTime);
-        $this->assertAttributeEquals($startTime, 'startTime', $profiler);
-    }
-
-    public function testSetConsole()
-    {
-        $console = new Console();
-        $profiler = new PhpQuickProfiler();
-        $profiler->setConsole($console);
-
-        $this->assertAttributeSame($console, 'console', $profiler);
-    }
-
-    public function testSetDisplay()
-    {
-        $display = new Display();
-        $profiler = new PhpQuickProfiler();
-        $profiler->setDisplay($display);
-
-        $this->assertAttributeSame($display, 'display', $profiler);
-    }
-
-    public function testGatherFileData()
-    {
-        $files = get_included_files();
-        $profiler = new PhpQuickProfiler();
-        $gatheredFileData = $profiler->gatherFileData();
-
-        $this->assertInternalType('array', $gatheredFileData);
-        $this->assertEquals(count($files), count($gatheredFileData));
-        foreach ($gatheredFileData as $fileData) {
-            $this->assertInternalType('array', $fileData);
-            $this->assertArrayHasKey('name', $fileData);
-            $this->assertContains($fileData['name'], $files);
-            $this->assertArrayHasKey('size', $fileData);
-            $this->assertEquals($fileData['size'], filesize($fileData['name']));
-        }
-    }
-
-    public function testGatherMemoryData()
-    {
-        $memoryUsage = memory_get_peak_usage();
-        $allowedLimit = ini_get('memory_limit');
-        $profiler = new PhpQuickProfiler();
-        $gatheredMemoryData = $profiler->gatherMemoryData();
-
-        $this->assertInternalType('array', $gatheredMemoryData);
-        $this->assertEquals(2, count($gatheredMemoryData));
-        $this->assertArrayHasKey('used', $gatheredMemoryData);
-        $this->assertEquals($memoryUsage, $gatheredMemoryData['used']);
-        $this->assertArrayHasKey('allowed', $gatheredMemoryData);
-        $this->assertEquals($allowedLimit, $gatheredMemoryData['allowed']);
-    }
-
-    public function testSetProfiledQueries()
-    {
-        $profiledQueries = $this->dataProfiledQueries();
-        $profiler = new PhpQuickProfiler();
-        $profiler->setProfiledQueries($profiledQueries);
-
-        $this->assertAttributeEquals($profiledQueries, 'profiledQueries', $profiler);
-    }
-
-    public function testGatherQueryData()
-    {
-        $profiledQueries = $this->dataProfiledQueries();
-        $profiledQueriesSql = array();
-        $profiledQueriesTime = array();
-        foreach ($profiledQueries as $queryData) {
-            array_push($profiledQueriesSql, $queryData['sql']);
-            array_push($profiledQueriesTime, $queryData['time']);
-        }
-
-        $profiler = new PhpQuickProfiler();
-        $profiler->setProfiledQueries($profiledQueries);
-        $gatheredQueryData = $profiler->gatherQueryData(self::$dbConnection);
-
-        $this->assertInternalType('array', $gatheredQueryData);
-        $this->assertEquals(count($profiledQueries), count($gatheredQueryData));
-        foreach ($gatheredQueryData as $queryData) {
-            $this->assertInternalType('array', $queryData);
-            $this->assertArrayHasKey('sql', $queryData);
-            $this->assertContains($queryData['sql'], $profiledQueriesSql);
-            $this->assertArrayHasKey('explain', $queryData);
-            $this->assertInternaltype('array', $queryData['explain']);
-            $this->assertGreaterThan(0, count($queryData['explain']));
-            $this->assertArrayHasKey('time', $queryData);
-            $this->assertContains($queryData['time'], $profiledQueriesTime);
-        }
-    }
-
-    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
-     */
-    public function testExplainQuery($sql, $parameters)
-    {
-        $profiler = new PhpQuickProfiler();
-        $reflectedMethod = $this->getAccessibleMethod($profiler, 'explainQuery');
-
-        $explainedQuery = $reflectedMethod->invokeArgs(
-            $profiler,
-            array(self::$dbConnection, $sql, $parameters)
-        );
-        $this->assertInternalType('array', $explainedQuery);
-        $this->assertGreaterThan(0, count($explainedQuery));
-    }
-
-    /**
-     * @expectedException Exception
-     */
-    public function testExplainQueryBadQueryException()
-    {
-        $invalidQuery = 'SELECT * FROM `fake_table`';
-        $profiler = new PhpQuickProfiler();
-        $reflectedMethod = $this->getAccessibleMethod($profiler, 'explainQuery');
-
-        $reflectedMethod->invokeArgs(
-            $profiler,
-            array(self::$dbConnection, $invalidQuery)
-        );
-    }
-
-    /**
-     * @expectedException Exception
-     */
-    public function testExplainQueryBadParametersException()
-    {
-        $query = 'SELECT * FROM `testing` WHERE `title` = :title';
-        $invalidParams = array('id' => 1);
-        $profiler = new PhpQuickProfiler();
-        $reflectedMethod = $this->getAccessibleMethod($profiler, 'explainQuery');
-
-        $reflectedMethod->invokeArgs(
-            $profiler,
-            array(self::$dbConnection, $query, $invalidParams)
-        );
-    }
-
-    /**
-     * @dataProvider dataConnectionDrivers
-     */
-    public function testGetExplainQuery($driver, $prefix)
-    {
-        $query = 'SELECT * FROM `testing`';
-        $profiler = new PhpQuickProfiler();
-        $reflectedMethod = $this->getAccessibleMethod($profiler, 'getExplainQuery');
-
-        $explainQuery = $reflectedMethod->invokeArgs(
-            $profiler,
-            array($query, $driver)
-        );
-
-        $explainPrefix = str_replace($query, '', $explainQuery);
-        $explainPrefix = trim($explainPrefix);
-        $this->assertEquals($prefix, $explainPrefix);
-    }
-
-    /**
-     * @expectedException Exception
-     */
-    public function testGetExplainQueryUnsupportedDriver()
-    {
-        $query = 'SELECT * FROM `testing`';
-        $unsupportedDriver = 'zz';
-        $profiler = new PhpQuickProfiler();
-        $reflectedMethod = $this->getAccessibleMethod($profiler, 'getExplainQuery');
-
-        $reflectedMethod->invokeArgs(
-            $profiler,
-            array($query, $unsupportedDriver)
-        );
-    }
-
-    public function testGatherSpeedData()
-    {
-        $elapsedTime = 1.234;
-        $startTime = microtime(true) - $elapsedTime;
-        $allowedTime = ini_get('max_execution_time');
-        $profiler = new PhpQuickProfiler($startTime);
-        $gatheredSpeedData = $profiler->gatherSpeedData();
-
-        $this->assertInternalType('array', $gatheredSpeedData);
-        $this->assertEquals(2, count($gatheredSpeedData));
-        $this->assertArrayHasKey('elapsed', $gatheredSpeedData);
-        $this->assertEquals($elapsedTime, $gatheredSpeedData['elapsed']);
-        $this->assertArrayHasKey('allowed', $gatheredSpeedData);
-        $this->assertEquals($allowedTime, $gatheredSpeedData['allowed']);
-    }
-
-    public function testDisplay()
-    {
-        $console = new Console();
-        $profiler = new PhpQuickProfiler();
-
-        $expectedDisplay = new Display();
-        $expectedDisplay->setConsole($console);
-        $expectedDisplay->setFileData($profiler->gatherFileData());
-        $expectedDisplay->setMemoryData($profiler->gatherMemoryData());
-        $expectedDisplay->setQueryData($profiler->gatherQueryData());
-        $expectedDisplay->setSpeedData($profiler->gatherSpeedData());
-        ob_start();
-        $expectedDisplay->__invoke();
-        ob_end_clean();
-
-        $display = new Display();
-        $profiler->setConsole($console);
-        $profiler->setDisplay($display);
-        ob_start();
-        $profiler->display();
-        ob_end_clean();
-
-        $this->assertAttributeEquals($expectedDisplay, 'display', $profiler);
-    }
-
-    /**
-     * @expectedException Exception
-     */
-    public function testDisplayNothingSetException()
-    {
-        $profiler = new PhpQuickProfiler();
-        $profiler->display();
-    }
-
-    /**
-     * @expectedException Exception
-     */
-    public function testDisplayNoConsoleException()
-    {
-        $display = new Display();
-        $profiler = new PhpQuickProfiler();
-        $profiler->setDisplay($display);
-        $profiler->display();
-    }
-
-    /**
-     * @expectedException Exception
-     */
-    public function testDisplayNoDisplayException()
-    {
-        $console = new Console();
-        $profiler = new PhpQuickProfiler();
-        $profiler->setConsole($console);
-        $profiler->display();
-    }
-
-    public function dataProfiledQueries()
-    {
-        return array(
-            array(
-              'sql' => "SELECT * FROM testing",
-              'parameters' => array(),
-              'time' => 25
-            ),
-            array(
-              'sql' => "SELECT id FROM testing WHERE title = :title",
-              'parameters' => array('title' => 'beta'),
-              'time' => 5
-            )
-        );
-    }
-
-    public function dataConnectionDrivers()
-    {
-        return array(
-            array(
-                'driver' => 'mysql',
-                'prefix' => 'EXPLAIN'
-            ),
-            array(
-                'driver' => 'sqlite',
-                'prefix' => 'EXPLAIN QUERY PLAN'
-            )
-        );
-    }
-
-    protected function getAccessibleMethod(PhpQuickProfiler $profiler, $methodName)
-    {
-        $reflectedConsole = new ReflectionClass(get_class($profiler));
-        $reflectedMethod = $reflectedConsole->getMethod($methodName);
-        $reflectedMethod->setAccessible(true);
-        return $reflectedMethod;
-    }
-
-    public static function tearDownAfterClass()
-    {
-        self::$dbConnection = null;
-    }
+		self::$dbConnection->exec($hydrateTable);
+	}
+
+	public function testConstruct()
+	{
+		$startTime = microtime(true);
+
+		$profiler = new PhpQuickProfiler();
+		$this->assertAttributeEquals($startTime, 'startTime', $profiler);
+
+		$profiler = new PhpQuickProfiler($startTime);
+		$this->assertAttributeEquals($startTime, 'startTime', $profiler);
+	}
+
+	public function testSetConsole()
+	{
+		$console = new Console();
+		$profiler = new PhpQuickProfiler();
+		$profiler->setConsole($console);
+
+		$this->assertAttributeSame($console, 'console', $profiler);
+	}
+
+	public function testSetDisplay()
+	{
+		$display = new Display();
+		$profiler = new PhpQuickProfiler();
+		$profiler->setDisplay($display);
+
+		$this->assertAttributeSame($display, 'display', $profiler);
+	}
+
+	public function testGatherFileData()
+	{
+		$files = get_included_files();
+		$profiler = new PhpQuickProfiler();
+		$gatheredFileData = $profiler->gatherFileData();
+
+		$this->assertInternalType('array', $gatheredFileData);
+		$this->assertEquals(count($files), count($gatheredFileData));
+		foreach ($gatheredFileData as $fileData) {
+			$this->assertInternalType('array', $fileData);
+			$this->assertArrayHasKey('name', $fileData);
+			$this->assertContains($fileData['name'], $files);
+			$this->assertArrayHasKey('size', $fileData);
+			$this->assertEquals($fileData['size'], filesize($fileData['name']));
+		}
+	}
+
+	public function testGatherMemoryData()
+	{
+		$memoryUsage = memory_get_peak_usage();
+		$allowedLimit = ini_get('memory_limit');
+		$profiler = new PhpQuickProfiler();
+		$gatheredMemoryData = $profiler->gatherMemoryData();
+
+		$this->assertInternalType('array', $gatheredMemoryData);
+		$this->assertEquals(2, count($gatheredMemoryData));
+		$this->assertArrayHasKey('used', $gatheredMemoryData);
+		$this->assertEquals($memoryUsage, $gatheredMemoryData['used']);
+		$this->assertArrayHasKey('allowed', $gatheredMemoryData);
+		$this->assertEquals($allowedLimit, $gatheredMemoryData['allowed']);
+	}
+
+	public function testSetProfiledQueries()
+	{
+		$profiledQueries = $this->dataProfiledQueries();
+		$profiler = new PhpQuickProfiler();
+		$profiler->setProfiledQueries($profiledQueries);
+
+		$this->assertAttributeEquals($profiledQueries, 'profiledQueries', $profiler);
+	}
+
+	public function testGatherQueryData()
+	{
+		$profiledQueries = $this->dataProfiledQueries();
+		$profiledQueriesSql = array();
+		$profiledQueriesTime = array();
+		foreach ($profiledQueries as $queryData) {
+			array_push($profiledQueriesSql, $queryData['sql']);
+			array_push($profiledQueriesTime, $queryData['time']);
+		}
+
+		$profiler = new PhpQuickProfiler();
+		$profiler->setProfiledQueries($profiledQueries);
+		$gatheredQueryData = $profiler->gatherQueryData(self::$dbConnection);
+
+		$this->assertInternalType('array', $gatheredQueryData);
+		$this->assertEquals(count($profiledQueries), count($gatheredQueryData));
+		foreach ($gatheredQueryData as $queryData) {
+			$this->assertInternalType('array', $queryData);
+			$this->assertArrayHasKey('sql', $queryData);
+			$this->assertContains($queryData['sql'], $profiledQueriesSql);
+			$this->assertArrayHasKey('explain', $queryData);
+			$this->assertInternaltype('array', $queryData['explain']);
+			$this->assertGreaterThan(0, count($queryData['explain']));
+			$this->assertArrayHasKey('time', $queryData);
+			$this->assertContains($queryData['time'], $profiledQueriesTime);
+		}
+	}
+
+	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
+	 */
+	public function testExplainQuery($sql, $parameters)
+	{
+		$profiler = new PhpQuickProfiler();
+		$reflectedMethod = $this->getAccessibleMethod($profiler, 'explainQuery');
+
+		$explainedQuery = $reflectedMethod->invokeArgs(
+			$profiler,
+			array(self::$dbConnection, $sql, $parameters)
+		);
+		$this->assertInternalType('array', $explainedQuery);
+		$this->assertGreaterThan(0, count($explainedQuery));
+	}
+
+	/**
+	 * @expectedException Exception
+	 */
+	public function testExplainQueryBadQueryException()
+	{
+		$invalidQuery = 'SELECT * FROM `fake_table`';
+		$profiler = new PhpQuickProfiler();
+		$reflectedMethod = $this->getAccessibleMethod($profiler, 'explainQuery');
+
+		$reflectedMethod->invokeArgs(
+			$profiler,
+			array(self::$dbConnection, $invalidQuery)
+		);
+	}
+
+	/**
+	 * @expectedException Exception
+	 */
+	public function testExplainQueryBadParametersException()
+	{
+		$query = 'SELECT * FROM `testing` WHERE `title` = :title';
+		$invalidParams = array('id' => 1);
+		$profiler = new PhpQuickProfiler();
+		$reflectedMethod = $this->getAccessibleMethod($profiler, 'explainQuery');
+
+		$reflectedMethod->invokeArgs(
+			$profiler,
+			array(self::$dbConnection, $query, $invalidParams)
+		);
+	}
+
+	/**
+	 * @dataProvider dataConnectionDrivers
+	 */
+	public function testGetExplainQuery($driver, $prefix)
+	{
+		$query = 'SELECT * FROM `testing`';
+		$profiler = new PhpQuickProfiler();
+		$reflectedMethod = $this->getAccessibleMethod($profiler, 'getExplainQuery');
+
+		$explainQuery = $reflectedMethod->invokeArgs(
+			$profiler,
+			array($query, $driver)
+		);
+
+		$explainPrefix = str_replace($query, '', $explainQuery);
+		$explainPrefix = trim($explainPrefix);
+		$this->assertEquals($prefix, $explainPrefix);
+	}
+
+	/**
+	 * @expectedException Exception
+	 */
+	public function testGetExplainQueryUnsupportedDriver()
+	{
+		$query = 'SELECT * FROM `testing`';
+		$unsupportedDriver = 'zz';
+		$profiler = new PhpQuickProfiler();
+		$reflectedMethod = $this->getAccessibleMethod($profiler, 'getExplainQuery');
+
+		$reflectedMethod->invokeArgs(
+			$profiler,
+			array($query, $unsupportedDriver)
+		);
+	}
+
+	public function testGatherSpeedData()
+	{
+		$elapsedTime = 1.234;
+		$startTime = microtime(true) - $elapsedTime;
+		$allowedTime = ini_get('max_execution_time');
+		$profiler = new PhpQuickProfiler($startTime);
+		$gatheredSpeedData = $profiler->gatherSpeedData();
+
+		$this->assertInternalType('array', $gatheredSpeedData);
+		$this->assertEquals(2, count($gatheredSpeedData));
+		$this->assertArrayHasKey('elapsed', $gatheredSpeedData);
+		$this->assertEquals($elapsedTime, $gatheredSpeedData['elapsed']);
+		$this->assertArrayHasKey('allowed', $gatheredSpeedData);
+		$this->assertEquals($allowedTime, $gatheredSpeedData['allowed']);
+	}
+
+	public function testDisplay()
+	{
+		$console = new Console();
+		$profiler = new PhpQuickProfiler();
+
+		$expectedDisplay = new Display();
+		$expectedDisplay->setConsole($console);
+		$expectedDisplay->setFileData($profiler->gatherFileData());
+		$expectedDisplay->setMemoryData($profiler->gatherMemoryData());
+		$expectedDisplay->setQueryData($profiler->gatherQueryData());
+		$expectedDisplay->setSpeedData($profiler->gatherSpeedData());
+		ob_start();
+		$expectedDisplay->__invoke();
+		ob_end_clean();
+
+		$display = new Display();
+		$profiler->setConsole($console);
+		$profiler->setDisplay($display);
+		ob_start();
+		$profiler->display();
+		ob_end_clean();
+
+		$this->assertAttributeEquals($expectedDisplay, 'display', $profiler);
+	}
+
+	/**
+	 * @expectedException Exception
+	 */
+	public function testDisplayNothingSetException()
+	{
+		$profiler = new PhpQuickProfiler();
+		$profiler->display();
+	}
+
+	/**
+	 * @expectedException Exception
+	 */
+	public function testDisplayNoConsoleException()
+	{
+		$display = new Display();
+		$profiler = new PhpQuickProfiler();
+		$profiler->setDisplay($display);
+		$profiler->display();
+	}
+
+	/**
+	 * @expectedException Exception
+	 */
+	public function testDisplayNoDisplayException()
+	{
+		$console = new Console();
+		$profiler = new PhpQuickProfiler();
+		$profiler->setConsole($console);
+		$profiler->display();
+	}
+
+	public function dataProfiledQueries()
+	{
+		return array(
+			array(
+			  'sql' => "SELECT * FROM testing",
+			  'parameters' => array(),
+			  'time' => 25
+			),
+			array(
+			  'sql' => "SELECT id FROM testing WHERE title = :title",
+			  'parameters' => array('title' => 'beta'),
+			  'time' => 5
+			)
+		);
+	}
+
+	public function dataConnectionDrivers()
+	{
+		return array(
+			array(
+				'driver' => 'mysql',
+				'prefix' => 'EXPLAIN'
+			),
+			array(
+				'driver' => 'sqlite',
+				'prefix' => 'EXPLAIN QUERY PLAN'
+			)
+		);
+	}
+
+	protected function getAccessibleMethod(PhpQuickProfiler $profiler, $methodName)
+	{
+		$reflectedConsole = new ReflectionClass(get_class($profiler));
+		$reflectedMethod = $reflectedConsole->getMethod($methodName);
+		$reflectedMethod->setAccessible(true);
+		return $reflectedMethod;
+	}
+
+	public static function tearDownAfterClass()
+	{
+		self::$dbConnection = null;
+	}
 }

src/PhpQuickProfiler.php

Indentation +189 added lines, -189 removed lines

@@ -17,193 +17,193 @@
 class PhpQuickProfiler
 {
 
-    /** @var  double */
-    protected $startTime;
-
-    /** @var  Console */
-    protected $console;
-
-    /** @var  Display */
-    protected $display;
-
-    /** @var  array */
-    protected $profiledQueries = array();
-
-    /**
-     * @param double $startTime
-     */
-    public function __construct($startTime = null)
-    {
-        if (is_null($startTime)) {
-            $startTime = microtime(true);
-        }
-        $this->startTime = $startTime;
-    }
-
-    /**
-     * @param Console $console
-     */
-    public function setConsole(Console $console)
-    {
-        $this->console = $console;
-    }
-
-    /**
-     * @param Display $display
-     */
-    public function setDisplay(Display $display)
-    {
-        $this->display = $display;
-    }
-
-    /**
-     * Get data about files loaded for the application to current point
-     *
-     * @returns array
-     */
-    public function gatherFileData()
-    {
-        $files = get_included_files();
-        $data = array();
-        foreach ($files as $file) {
-            array_push($data, array(
-                'name' => $file,
-                'size' => filesize($file)
-            ));
-        }
-        return $data;
-    }
-
-    /**
-     * Get data about memory usage of the application
-     *
-     * @returns array
-     */
-    public function gatherMemoryData()
-    {
-        $usedMemory = memory_get_peak_usage();
-        $allowedMemory = ini_get('memory_limit');
-        return array(
-            'used'    => $usedMemory,
-            'allowed' => $allowedMemory
-        );
-    }
-
-    /**
-     * @param array $profiled_queries
-     */
-    public function setProfiledQueries(array $profiledQueries)
-    {
-        $this->profiledQueries = $profiledQueries;
-    }
-
-    /**
-     * Get data about sql usage of the application
-     *
-     * @param object $dbConnection
-     * @returns array
-     */
-    public function gatherQueryData($dbConnection = null)
-    {
-        if (is_null($dbConnection)) {
-            return array();
-        }
-
-        if (empty($this->profiledQueries) && property_exists($dbConnection, 'queries')) {
-            $this->setProfiledQueries($dbConnection->queries);
-        }
-
-        $data = array();
-        foreach ($this->profiledQueries as $query) {
-            array_push($data, array(
-                'sql'     => $query['sql'],
-                'explain' => $this->explainQuery($dbConnection, $query['sql'], $query['parameters']),
-                'time'    => $query['time']
-            ));
-        }
-        return $data;
-    }
-
-    /**
-     * Attempts to explain a query
-     *
-     * @param object $dbConnection
-     * @param string $query
-     * @param array  $parameters
-     * @throws Exception
-     * @return array
-     */
-    protected function explainQuery($dbConnection, $query, $parameters = array())
-    {
-        $driver = $dbConnection->getAttribute(\PDO::ATTR_DRIVER_NAME);
-        $query = $this->getExplainQuery($query, $driver);
-        $statement = $dbConnection->prepare($query);
-        if ($statement === false) {
-            throw new Exception('Invalid query passed to explainQuery method');
-        }
-        $statement->execute($parameters);
-        $result = $statement->fetch(\PDO::FETCH_ASSOC);
-        if ($result === false) {
-            throw new Exception('Query could not be explained with given parameters');
-        }
-        return $result;
-    }
-
-    /**
-     * Attempts to figure out what kind of explain query format the db wants
-     *
-     * @param string $query
-     * @param string $driver
-     * @throws Exception
-     * @return string
-     */
-    protected function getExplainQuery($query, $driver)
-    {
-        if ($driver == 'mysql') {
-            return "EXPLAIN {$query}";
-        } elseif ($driver == 'sqlite') {
-            return "EXPLAIN QUERY PLAN {$query}";
-        }
-        throw new Exception('Could not process db driver');
-    }
-
-    /**
-     * Get data about speed of the application
-     *
-     * @returns array
-     */
-    public function gatherSpeedData()
-    {
-        $elapsedTime = microtime(true) - $this->startTime;
-        $elapsedTime = round($elapsedTime, 3);
-        $allowedTime = ini_get('max_execution_time');
-        return array(
-            'elapsed' => $elapsedTime,
-            'allowed' => $allowedTime
-        );
-    }
-
-    /**
-     * Triggers end display of the profiling data
-     *
-     * @param object $dbConnection
-     * @throws Exception
-     */
-    public function display($dbConnection = null)
-    {
-        if (!isset($this->display)) {
-            throw new Exception('Display object has not been injected into Profiler');
-        }
-        if (!isset($this->console)) {
-            throw new Exception('Console object has not been injected into Profiler');
-        }
-
-        $this->display->setConsole($this->console);
-        $this->display->setFileData($this->gatherFileData());
-        $this->display->setMemoryData($this->gatherMemoryData());
-        $this->display->setQueryData($this->gatherQueryData($dbConnection));
-        $this->display->setSpeedData($this->gatherSpeedData());
-
-        $this->display->__invoke();
-    }
+	/** @var  double */
+	protected $startTime;
+
+	/** @var  Console */
+	protected $console;
+
+	/** @var  Display */
+	protected $display;
+
+	/** @var  array */
+	protected $profiledQueries = array();
+
+	/**
+	 * @param double $startTime
+	 */
+	public function __construct($startTime = null)
+	{
+		if (is_null($startTime)) {
+			$startTime = microtime(true);
+		}
+		$this->startTime = $startTime;
+	}
+
+	/**
+	 * @param Console $console
+	 */
+	public function setConsole(Console $console)
+	{
+		$this->console = $console;
+	}
+
+	/**
+	 * @param Display $display
+	 */
+	public function setDisplay(Display $display)
+	{
+		$this->display = $display;
+	}
+
+	/**
+	 * Get data about files loaded for the application to current point
+	 *
+	 * @returns array
+	 */
+	public function gatherFileData()
+	{
+		$files = get_included_files();
+		$data = array();
+		foreach ($files as $file) {
+			array_push($data, array(
+				'name' => $file,
+				'size' => filesize($file)
+			));
+		}
+		return $data;
+	}
+
+	/**
+	 * Get data about memory usage of the application
+	 *
+	 * @returns array
+	 */
+	public function gatherMemoryData()
+	{
+		$usedMemory = memory_get_peak_usage();
+		$allowedMemory = ini_get('memory_limit');
+		return array(
+			'used'    => $usedMemory,
+			'allowed' => $allowedMemory
+		);
+	}
+
+	/**
+	 * @param array $profiled_queries
+	 */
+	public function setProfiledQueries(array $profiledQueries)
+	{
+		$this->profiledQueries = $profiledQueries;
+	}
+
+	/**
+	 * Get data about sql usage of the application
+	 *
+	 * @param object $dbConnection
+	 * @returns array
+	 */
+	public function gatherQueryData($dbConnection = null)
+	{
+		if (is_null($dbConnection)) {
+			return array();
+		}
+
+		if (empty($this->profiledQueries) && property_exists($dbConnection, 'queries')) {
+			$this->setProfiledQueries($dbConnection->queries);
+		}
+
+		$data = array();
+		foreach ($this->profiledQueries as $query) {
+			array_push($data, array(
+				'sql'     => $query['sql'],
+				'explain' => $this->explainQuery($dbConnection, $query['sql'], $query['parameters']),
+				'time'    => $query['time']
+			));
+		}
+		return $data;
+	}
+
+	/**
+	 * Attempts to explain a query
+	 *
+	 * @param object $dbConnection
+	 * @param string $query
+	 * @param array  $parameters
+	 * @throws Exception
+	 * @return array
+	 */
+	protected function explainQuery($dbConnection, $query, $parameters = array())
+	{
+		$driver = $dbConnection->getAttribute(\PDO::ATTR_DRIVER_NAME);
+		$query = $this->getExplainQuery($query, $driver);
+		$statement = $dbConnection->prepare($query);
+		if ($statement === false) {
+			throw new Exception('Invalid query passed to explainQuery method');
+		}
+		$statement->execute($parameters);
+		$result = $statement->fetch(\PDO::FETCH_ASSOC);
+		if ($result === false) {
+			throw new Exception('Query could not be explained with given parameters');
+		}
+		return $result;
+	}
+
+	/**
+	 * Attempts to figure out what kind of explain query format the db wants
+	 *
+	 * @param string $query
+	 * @param string $driver
+	 * @throws Exception
+	 * @return string
+	 */
+	protected function getExplainQuery($query, $driver)
+	{
+		if ($driver == 'mysql') {
+			return "EXPLAIN {$query}";
+		} elseif ($driver == 'sqlite') {
+			return "EXPLAIN QUERY PLAN {$query}";
+		}
+		throw new Exception('Could not process db driver');
+	}
+
+	/**
+	 * Get data about speed of the application
+	 *
+	 * @returns array
+	 */
+	public function gatherSpeedData()
+	{
+		$elapsedTime = microtime(true) - $this->startTime;
+		$elapsedTime = round($elapsedTime, 3);
+		$allowedTime = ini_get('max_execution_time');
+		return array(
+			'elapsed' => $elapsedTime,
+			'allowed' => $allowedTime
+		);
+	}
+
+	/**
+	 * Triggers end display of the profiling data
+	 *
+	 * @param object $dbConnection
+	 * @throws Exception
+	 */
+	public function display($dbConnection = null)
+	{
+		if (!isset($this->display)) {
+			throw new Exception('Display object has not been injected into Profiler');
+		}
+		if (!isset($this->console)) {
+			throw new Exception('Console object has not been injected into Profiler');
+		}
+
+		$this->display->setConsole($this->console);
+		$this->display->setFileData($this->gatherFileData());
+		$this->display->setMemoryData($this->gatherMemoryData());
+		$this->display->setQueryData($this->gatherQueryData($dbConnection));
+		$this->display->setSpeedData($this->gatherSpeedData());
+
+		$this->display->__invoke();
+	}
 }