test/unit/ardb_tests.rb in ardb-0.29.2 vs test/unit/ardb_tests.rb in ardb-0.30.0

@@ -114,13 +114,20 @@
 
     should "set ActiveRecord::Base attributes" do
       subject.init
 
       assert_equal subject.config.logger, ActiveRecord::Base.logger
-      assert_equal(
-        subject.config.default_timezone,
-        ActiveRecord::Base.default_timezone,
-      )
+      if ActiveRecord.respond_to?(:default_timezone=)
+        assert_equal(
+          subject.config.default_timezone,
+          ActiveRecord.default_timezone,
+        )
+      else
+        assert_equal(
+          subject.config.default_timezone,
+          ActiveRecord::Base.default_timezone,
+        )
+      end
     end
 
     should "optionally establish an AR connection" do
       assert_nil @ardb_adapter
       subject.init