migrations_spec.rb in fx-0.2.0

- old
+ new

@@ -14,11 +14,11 @@
       example.run
     end
   end
 
   it "can run migrations that create functions" do
-    migration = Class.new(ActiveRecord::Migration) do
+    migration = Class.new(migration_class) do
       def up
         create_function :test
       end
     end
 
@@ -26,11 +26,11 @@
   end
 
   it "can run migrations that drop functions" do
     connection.create_function(:test)
 
-    migration = Class.new(ActiveRecord::Migration) do
+    migration = Class.new(migration_class) do
       def up
         drop_function :test
       end
     end
 
@@ -51,10 +51,10 @@
     with_function_definition(
       name: :test,
       version: 2,
       sql_definition: sql_definition,
     ) do
-      migration = Class.new(ActiveRecord::Migration) do
+      migration = Class.new(migration_class) do
         def change
           update_function :test, version: 2, revert_to_version: 1
         end
       end
 