Add rollBack() method to PDO class

src/PDO/PDO.php

@@ -19,7 +19,7 @@
  *   - Implement the `inTransaction` method
  *   - Finish the `lastInsertId` method
  *   - Implement the `query` method
- *   - Implement the `rollBack` method
+ *   - Finish the `rollBack` method
  *   - Implement the `setAttribute` method
  */
 class PDO extends BasePDO
@@ -124,8 +124,18 @@ public function quote($string, $paramtype = null)
         return $this->wye()->quote($string);
     }
 
+    /**
+     * Mimic for PDO::rollBack(). Rolls back the currently active transaction,
+     * or throws a PDOException if no transaction exists.
+     *
+     * @todo Allow false to be returned.
+     *
+     * @return true
+     */
     public function rollBack()
     {
+        $this->getWye()->rollBackTransaction();
 
+        return true;
     }
 }

tests/PDO/PDO/RollBackTest.php

@@ -0,0 +1,55 @@
+<?php
+
+namespace Tests\PDO\PDO;
+
+use PDOException;
+use Stratedge\Wye\Wye;
+
+class RollBackTest extends \Tests\TestCase
+{
+    public function testNoTransactionThrowsException()
+    {
+        $this->setExpectedException(PDOException::class);
+
+        Wye::makePDO()->rollBack();
+    }
+
+    public function testMarksLastTransactionRolledBack()
+    {
+        $pdo = Wye::makePDO();
+
+        $pdo->beginTransaction();
+
+        $transaction = Wye::currentTransaction();
+
+        $this->assertFalse($transaction->getRolledBack());
+
+        $pdo->rollBack();
+
+        $this->assertTrue($transaction->getRolledBack());
+    }
+
+    public function testReturnsTrue()
+    {
+        $pdo = Wye::makePDO();
+
+        $pdo->beginTransaction();
+
+        $this->assertTrue($pdo->rollBack());
+    }
+
+    public function testMarksWyeNoLongerInTransaction()
+    {
+        $pdo = Wye::makePDO();
+
+        $this->assertFalse(Wye::getInTransaction());
+
+        $pdo->beginTransaction();
+
+        $this->assertTrue(Wye::getInTransaction());
+
+        $pdo->rollBack();
+
+        $this->assertFalse(Wye::getInTransaction());
+    }
+}