From 2b21b068ce18a1075e06bdb1c13422b644d7a0d1 Mon Sep 17 00:00:00 2001
From: Jeffrey Guo <jeffrey@silverstripe.com>
Date: Fri, 18 Dec 2015 15:32:20 +1300
Subject: [PATCH] clean up team db after test session run

---
 code/TestSessionEnvironment.php | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/code/TestSessionEnvironment.php b/code/TestSessionEnvironment.php
index 1262961..85ba1ef 100644
--- a/code/TestSessionEnvironment.php
+++ b/code/TestSessionEnvironment.php
@@ -362,6 +362,15 @@ class TestSessionEnvironment extends Object
         $this->extend('onBeforeEndTestSession');
 
         if (SapphireTest::using_temp_db()) {
+            $state = $this->getState();
+            $dbConn = DB::getConn();
+            $dbExists = $dbConn->databaseExists($state->database);
+            if($dbExists) {
+                // Clean up temp database
+                $dbConn->dropDatabase();
+                file_put_contents('php://stdout', "Deleted temp database: $state->database" . PHP_EOL);
+            }
+            // End test session mode
             $this->resetDatabaseName();
 
             SapphireTest::set_is_running_test(false);