Include SchemaStatements under AR::Migration

Shopify · Aug 30, 2022 · 869d6c0 · 869d6c0
1 parent 276bab9
commit 869d6c0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/rbi/annotations/activerecord.rbi b/rbi/annotations/activerecord.rbi
@@ -4,3 +4,8 @@ class ActiveRecord::Schema
   sig {params(info: T::Hash[T.untyped, T.untyped], blk: T.proc.bind(ActiveRecord::Schema).void).void}
   def self.define(info = nil, &blk); end
 end
+
+class ActiveRecord::Migration
+  # @shim: Methods on migration are delegated to `SchemaStatements` using `method_missing`
+  include ActiveRecord::ConnectionAdapters::SchemaStatements
+end