Allow to check the schema in pre and post as well

Signed-off-by: Joas Schilling <coding@schilljs.com>

1 files changed, 7 insertions, 3 deletions

diff --git a/lib/private/DB/MigrationService.php b/lib/private/DB/MigrationService.php
index 9c37a4ffa71..912166988c0 100644
--- a/lib/private/DB/MigrationService.php
+++ b/lib/private/DB/MigrationService.php
@@ -390,9 +390,11 @@ class MigrationService {
 			throw new \InvalidArgumentException('Not a valid migration');
 		}
 
-		$instance->preSchemaChange($this->output);
+		$instance->preSchemaChange($this->output, function() {
+			return $this->connection->createSchema();
+		}, ['tablePrefix' => $this->connection->getPrefix()]);
 
-		$toSchema = $instance->changeSchema(function() {
+		$toSchema = $instance->changeSchema($this->output, function() {
 			return $this->connection->createSchema();
 		}, ['tablePrefix' => $this->connection->getPrefix()]);
 
@@ -400,7 +402,9 @@ class MigrationService {
 			$this->connection->migrateToSchema($toSchema);
 		}
 
-		$instance->postSchemaChange($this->output);
+		$instance->postSchemaChange($this->output, function() {
+			return $this->connection->createSchema();
+		}, ['tablePrefix' => $this->connection->getPrefix()]);
 
 		$this->markAsExecuted($version);
 	}