test/upgrade_generator_test.rb in audited-4.10.0 vs test/upgrade_generator_test.rb in audited-5.0.0

@@ -1,23 +1,19 @@
-require 'test_helper'
+require "test_helper"
 
-require 'generators/audited/upgrade_generator'
+require "generators/audited/upgrade_generator"
 
 class UpgradeGeneratorTest < Rails::Generators::TestCase
-  destination File.expand_path('../../tmp', __FILE__)
+  destination File.expand_path("../../tmp", __FILE__)
   setup :prepare_destination
   tests Audited::Generators::UpgradeGenerator
-  if Rails::VERSION::MAJOR == 4
-    self.use_transactional_fixtures = false
-  else
-    self.use_transactional_tests = false
-  end
+  self.use_transactional_tests = false
 
   test "should add 'comment' to audits table" do
     load_schema 1
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_migration "db/migrate/add_comment_to_audits.rb" do |content|
       assert_match(/add_column :audits, :comment, :string/, content)
     end
 
@@ -25,11 +21,11 @@
   end
 
   test "should rename 'changes' to 'audited_changes'" do
     load_schema 2
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_no_migration "db/migrate/add_comment_to_audits.rb"
 
     assert_migration "db/migrate/rename_changes_to_audited_changes.rb" do |content|
       assert_match(/rename_column :audits, :changes, :audited_changes/, content)
@@ -37,66 +33,65 @@
   end
 
   test "should add a 'remote_address' to audits table" do
     load_schema 3
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_migration "db/migrate/add_remote_address_to_audits.rb" do |content|
       assert_match(/add_column :audits, :remote_address, :string/, content)
     end
   end
 
   test "should add 'association_id' and 'association_type' to audits table" do
     load_schema 4
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_migration "db/migrate/add_association_to_audits.rb" do |content|
       assert_match(/add_column :audits, :association_id, :integer/, content)
       assert_match(/add_column :audits, :association_type, :string/, content)
     end
   end
 
   test "should rename 'association_id' to 'associated_id' and 'association_type' to 'associated_type'" do
     load_schema 5
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_migration "db/migrate/rename_association_to_associated.rb" do |content|
       assert_match(/rename_column :audits, :association_id, :associated_id/, content)
       assert_match(/rename_column :audits, :association_type, :associated_type/, content)
     end
   end
 
   test "should add 'request_uuid' to audits table" do
     load_schema 6
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_migration "db/migrate/add_request_uuid_to_audits.rb" do |content|
       assert_match(/add_column :audits, :request_uuid, :string/, content)
       assert_match(/add_index :audits, :request_uuid/, content)
     end
   end
 
   test "should add 'version' to auditable_index" do
     load_schema 6
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_migration "db/migrate/add_version_to_auditable_index.rb" do |content|
       assert_match(/add_index :audits, \[:auditable_type, :auditable_id, :version\]/, content)
     end
   end
 
   test "generate migration with correct AR migration parent" do
     load_schema 1
 
-    run_generator %w(upgrade)
+    run_generator %w[upgrade]
 
     assert_migration "db/migrate/add_comment_to_audits.rb" do |content|
-      parent = Rails::VERSION::MAJOR == 4 ? 'ActiveRecord::Migration' : "ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]"
-      assert_includes(content, "class AddCommentToAudits < #{parent}\n")
+      assert_includes(content, "class AddCommentToAudits < ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]\n")
     end
   end
 end