1 files changed, 6 insertions, 1 deletions

diff --git a/system/drivers/DB_postgre.php b/system/drivers/DB_postgre.php
index fe7c704f9..bd5fbac24 100644
--- a/system/drivers/DB_postgre.php
+++ b/system/drivers/DB_postgre.php
@@ -111,7 +111,7 @@ class CI_DB_postgre extends CI_DB {
 	 * @access	public
 	 * @return	bool		 
 	 */	
-	function trans_begin()
+	function trans_begin($test_mode = FALSE)
 	{
 		if ( ! $this->trans_enabled)
 		{
@@ -124,6 +124,11 @@ class CI_DB_postgre extends CI_DB {
 			return TRUE;
 		}
 
+		// Reset the transaction failure flag.
+		// If the $test_mode flag is set to TRUE transactions will be rolled back 
+		// even if the queries produce a successful result. 
+		$this->_trans_failure = ($test_mode === TRUE) ? TRUE : FALSE;
+
 		return @pg_exec($this->conn_id, "begin");
 	}