1 files changed, 20 insertions, 25 deletions
diff --git a/spec/rubocop/cop/migration/timestamps_spec.rb b/spec/rubocop/cop/migration/timestamps_spec.rb
index 2f4154907d2..91bb5c1b05b 100644
--- a/spec/rubocop/cop/migration/timestamps_spec.rb
+++ b/spec/rubocop/cop/migration/timestamps_spec.rb
@@ -1,12 +1,9 @@
 # frozen_string_literal: true
 
 require 'fast_spec_helper'
-require 'rubocop'
 require_relative '../../../../rubocop/cop/migration/timestamps'
 
 RSpec.describe RuboCop::Cop::Migration::Timestamps do
-  include CopHelper
-
   subject(:cop) { described_class.new }
 
   let(:migration_with_timestamps) do
@@ -62,38 +59,36 @@ RSpec.describe RuboCop::Cop::Migration::Timestamps do
     end
 
     it 'registers an offense when the "timestamps" method is used' do
-      inspect_source(migration_with_timestamps)
-
-      aggregate_failures do
-        expect(cop.offenses.size).to eq(1)
-        expect(cop.offenses.map(&:line)).to eq([8])
-      end
+      expect_offense(<<~RUBY)
+        class Users < ActiveRecord::Migration[4.2]
+          DOWNTIME = false
+
+          def change
+            create_table :users do |t|
+              t.string :username, null: false
+              t.timestamps null: true
+                ^^^^^^^^^^ Do not use `timestamps`, use `timestamps_with_timezone` instead
+              t.string :password
+            end
+          end
+        end
+      RUBY
     end
 
     it 'does not register an offense when the "timestamps" method is not used' do
-      inspect_source(migration_without_timestamps)
-
-      aggregate_failures do
-        expect(cop.offenses.size).to eq(0)
-      end
+      expect_no_offenses(migration_without_timestamps)
     end
 
     it 'does not register an offense when the "timestamps_with_timezone" method is used' do
-      inspect_source(migration_with_timestamps_with_timezone)
-
-      aggregate_failures do
-        expect(cop.offenses.size).to eq(0)
-      end
+      expect_no_offenses(migration_with_timestamps_with_timezone)
     end
   end
 
   context 'outside of migration' do
-    it 'registers no offense' do
-      inspect_source(migration_with_timestamps)
-      inspect_source(migration_without_timestamps)
-      inspect_source(migration_with_timestamps_with_timezone)
-
-      expect(cop.offenses.size).to eq(0)
+    it 'registers no offense', :aggregate_failures do
+      expect_no_offenses(migration_with_timestamps)
+      expect_no_offenses(migration_without_timestamps)
+      expect_no_offenses(migration_with_timestamps_with_timezone)
     end
   end
 end