lib/timber/frameworks/rails.rb in timber-2.0.4 vs lib/timber/frameworks/rails.rb in timber-2.0.5

@@ -45,14 +45,14 @@
         # Set the various Rails framework loggers. We *have* to do this because Rails
         # internally sets these with an ActiveSupport.onload(:active_record) { } callback.
         # We don't have an opportunity to intercept this since the :initialize_logger
         # initializer loads these modules. Moreover, earlier version of rails don't do this
         # at all, hence the defined? checks. Yay for being implicit.
-        ::ActionCable::Server::Base.logger = logger if defined?(::ActionCable::Server::Base)
-        ::ActionController::Base.logger = logger if defined?(::ActionController::Base)
+        ::ActionCable::Server::Base.logger = logger if defined?(::ActionCable::Server::Base) && ::ActionCable::Server::Base.respond_to?(:logger=)
+        ::ActionController::Base.logger = logger if defined?(::ActionController::Base) && ::ActionController::Base.respond_to?(:logger=)
         ::ActionMailer::Base.logger = logger if defined?(::ActionMailer::Base) && ::ActionMailer::Base.respond_to?(:logger=)
         ::ActionView::Base.logger = logger if defined?(::ActionView::Base) && ::ActionView::Base.respond_to?(:logger=)
-        ::ActiveRecord::Base.logger = logger if defined?(::ActiveRecord::Base)
+        ::ActiveRecord::Base.logger = logger if defined?(::ActiveRecord::Base) && ::ActiveRecord::Base.respond_to?(:logger=)
         ::Rails.logger = logger
       end
 
       def self.configure_middlewares(middleware)
         var_name = :"@_timber_middlewares_inserted"
\ No newline at end of file