update tests for ConnectionManager

jpfuentes2 · Jun 3, 2017 · 3226ef8 · 3226ef8
1 parent 652c3b2
commit 3226ef8
Showing 1 changed file with 22 additions and 3 deletions.
diff --git a/test/ConnectionManagerTest.php b/test/ConnectionManagerTest.php
@@ -18,10 +18,15 @@ public function test_get_connection()
 
 	public function test_get_connection_uses_existing_object()
 	{
-		$a = ConnectionManager::get_connection('mysql');
-		$a->harro = 'harro there';
+		$connection = ConnectionManager::get_connection('mysql');
+		$this->assert_same($connection, ConnectionManager::get_connection('mysql'));
+	}
 
-		$this->assert_same($a,ConnectionManager::get_connection('mysql'));
+	public function test_get_connection_with_default()
+	{
+		$default = ActiveRecord\Config::instance()->get_default_connection('mysql');
+		$connection = ConnectionManager::get_connection();
+		$this->assert_same(ConnectionManager::get_connection($default), $connection);
 	}
 
 	public function test_gh_91_get_connection_with_null_connection_is_always_default()
@@ -35,4 +40,18 @@ public function test_gh_91_get_connection_with_null_connection_is_always_default
 		$this->assert_same($conn_two, $conn_three);
 		$this->assert_same($conn_four, $conn_three);
 	}
+
+	public function test_drop_connection()
+	{
+		$connection = ConnectionManager::get_connection('mysql');
+		ConnectionManager::drop_connection('mysql');
+		$this->assert_not_same($connection, ConnectionManager::get_connection('mysql'));
+	}
+
+	public function test_drop_connection_with_default()
+	{
+		$connection = ConnectionManager::get_connection();
+		ConnectionManager::drop_connection();
+		$this->assert_not_same($connection, ConnectionManager::get_connection());
+	}
 }