Transactional using closure.

System/Database/Connection.php

@@ -95,6 +95,15 @@ public function commit() {
 	public function rollback() {
 		$this->connection->rollback();
 	}
+
+	/**
+	 * Transactional run of a code block.
+	 * 
+	 * \param $closure Closure or callback
+	 */
+	public function transactional($closure) {
+		$this->connection->transactional($closure);
+	}
 
 	/**
 	 * Create a new query.

System/Database/Tests/DatabaseTest.php

@@ -226,7 +226,6 @@ public function testTransactionCommit() {
 		$this->assertEquals(1, $stmt->rowCount());
 	}
 
-	// TODO not working
 	public function testTransactionRollback() {
 		$this->installTestTable();
 
@@ -240,4 +239,35 @@ public function testTransactionRollback() {
 		$stmt = $query->select("*")->from("test", "t")->execute();
 		$this->assertEquals(0, $stmt->rowCount());
 	}
+
+	public function testTransactionalCommit() {
+		$this->installTestTable();
+
+		$this->db->transactional(function() {
+			$this->db->insert("test", array(
+				"foo" => "bar"
+			));
+		});
+
+		$query = $this->db->newQuery();		
+		$stmt = $query->select("*")->from("test", "t")->execute();
+		$this->assertEquals(1, $stmt->rowCount());
+	}
+
+	public function testTransactionalRollback() {
+		$this->installTestTable();
+
+		try {
+			$this->db->transactional(function() {
+				$this->db->insert("test", array(
+					"foo" => "bar"
+				));
+				throw new \Exception("foo");
+			});
+		} catch (\Exception $e) {}
+
+		$query = $this->db->newQuery();		
+		$stmt = $query->select("*")->from("test", "t")->execute();
+		$this->assertEquals(0, $stmt->rowCount());
+	}
 }