diff --git a/phpBB/install/install_install.php b/phpBB/install/install_install.php
index 9867c56a3f..103262b516 100644
--- a/phpBB/install/install_install.php
+++ b/phpBB/install/install_install.php
@@ -1175,8 +1175,31 @@ class install_install extends module
 		}
 
 		// Ok we have the db info go ahead and work on building the table
-		$db_table_schema = @file_get_contents('schemas/schema.json');
-		$db_table_schema = json_decode($db_table_schema, true);
+		if (file_exists('schemas/schema.json'))
+		{
+			$db_table_schema = @file_get_contents('schemas/schema.json');
+			$db_table_schema = json_decode($db_table_schema, true);
+		}
+		else
+		{
+			global $phpbb_root_path, $phpEx, $table_prefix;
+			$table_prefix = 'phpbb_';
+
+			if (!defined('CONFIG_TABLE'))
+			{
+				// We need to include the constants file for the table constants
+				// when we generate the schema from the migration files.
+				include($phpbb_root_path . 'includes/constants.' . $phpEx);
+			}
+
+			$finder = new \phpbb\finder(new \phpbb\filesystem(), $phpbb_root_path, null, $phpEx);
+			$classes = $finder->core_path('phpbb/db/migration/data/')
+				->get_classes();
+
+			$sqlite_db = new \phpbb\db\driver\sqlite();
+			$schema_generator = new \phpbb\db\migration\schema_generator($classes, new \phpbb\config\config(array()), $sqlite_db, new \phpbb\db\tools($sqlite_db, true), $phpbb_root_path, $phpEx, $table_prefix);
+			$db_table_schema = $schema_generator->get_schema();
+		}
 
 		if (!defined('CONFIG_TABLE'))
 		{
diff --git a/tests/test_framework/phpbb_database_test_case.php b/tests/test_framework/phpbb_database_test_case.php
index 9dbb7150f1..26989fa345 100644
--- a/tests/test_framework/phpbb_database_test_case.php
+++ b/tests/test_framework/phpbb_database_test_case.php
@@ -55,47 +55,46 @@ abstract class phpbb_database_test_case extends PHPUnit_Extensions_Database_Test
 	static public function setUpBeforeClass()
 	{
 		$setup_extensions = static::setup_extensions();
-		self::$schema_file = '';
-		if (!empty($setup_extensions))
+
+		$schema_md5 = md5(serialize($setup_extensions));
+		self::$schema_file = __DIR__ . '/../tmp/' . $schema_md5 . '.json';
+		self::$install_schema_file = __DIR__ . '/../../phpBB/install/schemas/schema.json';
+
+		if (!file_exists(self::$schema_file))
 		{
-			$schema_md5 = md5(serialize($setup_extensions));
+			global $phpbb_root_path, $phpEx, $table_prefix;
+			$finder = new \phpbb\finder(new \phpbb\filesystem(), $phpbb_root_path, null, $phpEx);
 
-			self::$schema_file = __DIR__ . '/../tmp/' . $schema_md5 . '.json';
-			self::$phpbb_schema_copy = __DIR__ . '/../tmp/schema_phpbb_copy.json';
-			self::$install_schema_file = __DIR__ . '/../../phpBB/install/schemas/schema.json';
-
-			if (!file_exists(self::$schema_file))
+			if (!empty($setup_extensions))
 			{
-				global $phpbb_root_path, $phpEx, $table_prefix;
-
-				$finder = new \phpbb\finder(new \phpbb\filesystem(), $phpbb_root_path, null, $phpEx);
 				$classes = $finder->core_path('phpbb/db/migration/data/')
 					->set_extensions($setup_extensions)
 					->extension_directory('/migrations')
 					->get_classes();
 
-				$db = new \phpbb\db\driver\sqlite();
-				$schema_generator = new \phpbb\db\migration\schema_generator($classes, new \phpbb\config\config(array()), $db, new \phpbb\db\tools($db, true), $phpbb_root_path, $phpEx, $table_prefix);
-				$schema_data = $schema_generator->get_schema();
+			}
+			else
+			{
 
-				file_put_contents(self::$schema_file, json_encode($schema_data));
+				$classes = $finder->core_path('phpbb/db/migration/data/')
+					->get_classes();
 			}
 
-			copy(self::$install_schema_file, self::$phpbb_schema_copy);
-			copy(self::$schema_file, self::$install_schema_file);
+			$db = new \phpbb\db\driver\sqlite();
+			$schema_generator = new \phpbb\db\migration\schema_generator($classes, new \phpbb\config\config(array()), $db, new \phpbb\db\tools($db, true), $phpbb_root_path, $phpEx, $table_prefix);
+			$schema_data = $schema_generator->get_schema();
+
+			file_put_contents(self::$schema_file, json_encode($schema_data));
 		}
 
+		copy(self::$schema_file, self::$install_schema_file);
+
 		parent::setUpBeforeClass();
 	}
 
 	static public function tearDownAfterClass()
 	{
-		if (self::$schema_file !== '')
-		{
-			copy(self::$phpbb_schema_copy, self::$install_schema_file);
-			unlink(self::$schema_file);
-		}
-
+		unlink(self::$install_schema_file);
 		parent::tearDownAfterClass();
 	}
 
diff --git a/tests/test_framework/phpbb_database_test_connection_manager.php b/tests/test_framework/phpbb_database_test_connection_manager.php
index 92e2080dba..5d643e43e2 100644
--- a/tests/test_framework/phpbb_database_test_connection_manager.php
+++ b/tests/test_framework/phpbb_database_test_connection_manager.php
@@ -356,8 +356,23 @@ class phpbb_database_test_connection_manager
 		}
 
 		// Ok we have the db info go ahead and work on building the table
-		$db_table_schema = file_get_contents($directory . 'schema.json');
-		$db_table_schema = json_decode($db_table_schema, true);
+		if (file_exists($directory . 'schema.json'))
+		{
+			$db_table_schema = file_get_contents($directory . 'schema.json');
+			$db_table_schema = json_decode($db_table_schema, true);
+		}
+		else
+		{
+			global $phpbb_root_path, $phpEx, $table_prefix;
+
+			$finder = new \phpbb\finder(new \phpbb\filesystem(), $phpbb_root_path, null, $phpEx);
+			$classes = $finder->core_path('phpbb/db/migration/data/')
+				->get_classes();
+
+			$db = new \phpbb\db\driver\sqlite();
+			$schema_generator = new \phpbb\db\migration\schema_generator($classes, new \phpbb\config\config(array()), $db, new \phpbb\db\tools($db, true), $phpbb_root_path, $phpEx, $table_prefix);
+			$db_table_schema = $schema_generator->get_schema();
+		}
 
 		$db_tools = new \phpbb\db\tools($db, true);
 		foreach ($db_table_schema as $table_name => $table_data)