Merge pull request #276 from ytetsuro/feat/db-connection-close

fixed Too many connections error.

application/tests/_ci_phpunit_test/CIPHPUnitTestCase.php

@@ -85,6 +85,11 @@ public static function setUpBeforeClass()
 		chdir(FCPATH);
 	}
 
+	public static function tearDownAfterClass()
+	{
+		CIPHPUnitTestDbConnectionStore::destory();
+	}
+
 	/**
 	 * Reset CodeIgniter instance and assign new CodeIgniter instance as $this->CI
 	 */

application/tests/_ci_phpunit_test/CIPHPUnitTestDbConnectionStore.php

@@ -0,0 +1,27 @@
+<?php
+
+class CIPHPUnitTestDbConnectionStore
+{
+	private static $connections = [];
+
+	public static function add(CI_DB $db)
+	{
+		self::$connections[] = $db;
+	}
+
+	public static function destory()
+	{
+		foreach (self::$connections as $db) {
+			self::closeConnection($db);
+		}
+
+		self::$connections = [];
+	}
+
+	private static function closeConnection(CI_DB $db)
+	{
+		if ($db->dsn !== 'sqlite::memory:' && $db->database !== ':memory:') {
+			$db->close();
+		}
+	}
+}

application/tests/_ci_phpunit_test/replacing/core/Loader.php

@@ -400,7 +400,9 @@ public function database($params = '', $return = FALSE, $query_builder = NULL)
 
 		if ($return === TRUE)
 		{
-			return DB($params, $query_builder);
+			$result = DB($params, $query_builder);
+			CIPHPUnitTestDbConnectionStore::add($result);
+			return $result;
 		}
 
 		// Initialize the db variable. Needed to prevent