From 69ef16360f6720a14cdadb277fce7ac95e36ea42 Mon Sep 17 00:00:00 2001
From: beberlei <beberlei@625475ce-881a-0410-a577-b389adb331d8>
Date: Wed, 4 Nov 2009 11:39:49 +0000
Subject: [PATCH] [2.0] DDC-112 - Enhance debugging capabilities in
 OrmFunctionalTestCase by logging all SQL queries and enrichting all test
 failures with them (PHPUnit 3.4 only feature).

---
 lib/Doctrine/DBAL/Logging/DebugStack.php      | 37 +++++++++++++
 .../Doctrine/Tests/OrmFunctionalTestCase.php  | 52 ++++++++++++++++++-
 2 files changed, 87 insertions(+), 2 deletions(-)
 create mode 100644 lib/Doctrine/DBAL/Logging/DebugStack.php

diff --git a/lib/Doctrine/DBAL/Logging/DebugStack.php b/lib/Doctrine/DBAL/Logging/DebugStack.php
new file mode 100644
index 000000000..48235e0b7
--- /dev/null
+++ b/lib/Doctrine/DBAL/Logging/DebugStack.php
@@ -0,0 +1,37 @@
+<?php
+/*
+ *  $Id$
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * This software consists of voluntary contributions made by many individuals
+ * and is licensed under the LGPL. For more information, see
+ * <http://www.doctrine-project.org>.
+ */
+
+namespace Doctrine\DBAL\Logging;
+
+class DebugStack implements SqlLogger
+{
+    public $queries = array();
+
+    public $enabled = true;
+
+    public function logSql($sql, array $params = null)
+    {
+        if($this->enabled) {
+            $this->queries[] = array('sql' => $sql, 'params' => $params);
+        }
+    }
+}
+
diff --git a/tests/Doctrine/Tests/OrmFunctionalTestCase.php b/tests/Doctrine/Tests/OrmFunctionalTestCase.php
index 6e9af81d4..814ac22ca 100644
--- a/tests/Doctrine/Tests/OrmFunctionalTestCase.php
+++ b/tests/Doctrine/Tests/OrmFunctionalTestCase.php
@@ -17,12 +17,21 @@ class OrmFunctionalTestCase extends OrmTestCase
     /* Shared connection when a TestCase is run alone (outside of it's functional suite) */
     private static $_sharedConn;
     
-    /** The EntityManager for this testcase. */
+    /**
+     * @var \Doctrine\ORM\EntityManager
+     */
     protected $_em;
 
-    /** The SchemaTool. */
+    /**
+     * @var \Doctrine\ORM\Tools\SchemaTool
+     */
     protected $_schemaTool;
 
+    /**
+     * @var \Doctrine\DBAL\Logging\DebugStack
+     */
+    protected $_sqlLoggerStack;
+
     /** The names of the model sets used in this testcase. */
     private $_usedModelSets = array();
 
@@ -73,6 +82,8 @@ class OrmFunctionalTestCase extends OrmTestCase
     protected function tearDown()
     {
         $conn = $this->sharedFixture['conn'];
+
+        $this->_sqlLoggerStack->enabled = false;
         
         if (isset($this->_usedModelSets['cms'])) {
             $conn->executeUpdate('DELETE FROM cms_users_groups');
@@ -156,6 +167,8 @@ class OrmFunctionalTestCase extends OrmTestCase
         if ($classes) {
             $this->_schemaTool->createSchema($classes);
         }
+
+        $this->_sqlLoggerStack->enabled = true;
     }
 
     /**
@@ -176,6 +189,9 @@ class OrmFunctionalTestCase extends OrmTestCase
         if (is_null(self::$_queryCacheImpl)) {
         	self::$_queryCacheImpl = new \Doctrine\Common\Cache\ArrayCache;
         }
+
+        $this->_sqlLoggerStack = new \Doctrine\DBAL\Logging\DebugStack();
+        $this->_sqlLoggerStack->enabled = false;
         
         //FIXME: two different configs! $conn and the created entity manager have
         // different configs.
@@ -184,8 +200,40 @@ class OrmFunctionalTestCase extends OrmTestCase
         $config->setQueryCacheImpl(self::$_queryCacheImpl);
         $config->setProxyDir(__DIR__ . '/Proxies');
         $config->setProxyNamespace('Doctrine\Tests\Proxies');
+        
         $conn = $this->sharedFixture['conn'];
+        $conn->getConfiguration()->setSqlLogger($this->_sqlLoggerStack);
         
         return \Doctrine\ORM\EntityManager::create($conn, $config);
     }
+
+    protected function onNotSuccessfulTest(\Exception $e)
+    {
+        if($this->_sqlLoggerStack->queries !== null && count($this->_sqlLoggerStack->queries)) {
+            $queries = "";
+            for($i = 0; $i < count($this->_sqlLoggerStack->queries); $i++) {
+                $query = $this->_sqlLoggerStack->queries[$i];
+                $params = array_map(function($p) { return "'".$p."'"; }, $query['params']);
+                $queries .= ($i+1).". SQL: '".$query['sql']."' Params: ".implode(", ", $params).PHP_EOL;
+            }
+            
+            $trace = $e->getTrace();
+            $traceMsg = "";
+            foreach($trace AS $part) {
+                if(isset($part['file'])) {
+                    if(strpos($part['file'], "PHPUnit/") !== false) {
+                        // Beginning with PHPUnit files we don't print the trace anymore.
+                        break;
+                    }
+
+                    $traceMsg .= $part['file'].":".$part['line'].PHP_EOL;
+                }
+            }
+
+            $message = "[".get_class($e)."] ".$e->getMessage().PHP_EOL.PHP_EOL."With queries:".PHP_EOL.$queries.PHP_EOL."Trace:".PHP_EOL.$traceMsg;
+
+            throw new \Exception($message, (int)$e->getCode(), $e);
+        }
+        throw $e;
+    }
 }