Make it possible to provide schema version in tests

1 files changed, 15 insertions, 0 deletions

diff --git a/spec/support/migrations_helpers.rb b/spec/support/migrations_helpers.rb
index aabdad13047..3e8c6b54c5a 100644
--- a/spec/support/migrations_helpers.rb
+++ b/spec/support/migrations_helpers.rb
@@ -31,6 +31,21 @@ module MigrationsHelpers
     end
   end
 
+  def migration_schema_version
+    self.class.metadata[:schema] || previous_migration.version
+  end
+
+  def schema_migrate_down!
+    ActiveRecord::Migrator
+      .migrate(migrations_paths, migration_schema_version)
+    reset_column_in_migration_models
+  end
+
+  def schema_migrate_up!
+    ActiveRecord::Migrator.migrate(migrations_paths)
+    reset_column_in_migration_models
+  end
+
   def migrate!
     ActiveRecord::Migrator.up(migrations_paths) do |migration|
       migration.name == described_class.name