test/unit/timestamp_test.rb in paper_trail-3.0.1 vs test/unit/timestamp_test.rb in paper_trail-3.0.2

@@ -3,12 +3,10 @@
 class TimestampTest < ActiveSupport::TestCase
 
   setup do
     PaperTrail.timestamp_field = :custom_created_at
     change_schema
-    PaperTrail::Version.connection.schema_cache.clear!
-    PaperTrail::Version.reset_column_information
 
     Fluxor.instance_eval <<-END
       has_paper_trail
     END
 
@@ -17,9 +15,10 @@
     @fluxor.update_attributes :name => 'Even more text.'
   end
 
   teardown do
     PaperTrail.timestamp_field = :created_at
+    restore_schema
   end
 
   test 'versions works with custom timestamp field' do
     # Normal behaviour
     assert_equal 3, @fluxor.versions.length