Use lazy load hooks to extend ActiveRecord::Base

Extend ActiveRecord::Base functionality only after ActiveRecord has been fully loaded. Ref: thoughtbot/factory_bot_rails#426
tagliala · Sep 29, 2023 · c36d1ad · c36d1ad
1 parent e62cc30
commit c36d1ad
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/amoeba.rb b/lib/amoeba.rb
@@ -16,5 +16,7 @@
 module Amoeba
 end
 
-ActiveRecord::Base.extend Amoeba::ClassMethods
-ActiveRecord::Base.include Amoeba::InstanceMethods
+ActiveSupport.on_load :active_record do
+  extend Amoeba::ClassMethods
+  include Amoeba::InstanceMethods
+end