1 files changed, 41 insertions, 4 deletions

diff --git a/spec/rubocop/cop/gitlab/json_spec.rb b/spec/rubocop/cop/gitlab/json_spec.rb
index e4ec107747d..70f63a78dd1 100644
--- a/spec/rubocop/cop/gitlab/json_spec.rb
+++ b/spec/rubocop/cop/gitlab/json_spec.rb
@@ -5,12 +5,41 @@ require_relative '../../../../rubocop/cop/gitlab/json'
 
 RSpec.describe RuboCop::Cop::Gitlab::Json do
   context 'when ::JSON is called' do
-    it 'registers an offense' do
+    it 'registers an offense and autocorrects' do
       expect_offense(<<~RUBY)
         class Foo
           def bar
             JSON.parse('{ "foo": "bar" }')
-            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Avoid calling `JSON` directly. [...]
+            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Prefer `Gitlab::Json` over calling `JSON` directly. [...]
+          end
+        end
+      RUBY
+
+      expect_correction(<<~RUBY)
+        class Foo
+          def bar
+            Gitlab::Json.parse('{ "foo": "bar" }')
+          end
+        end
+      RUBY
+    end
+  end
+
+  context 'when ::JSON is called in EE' do
+    it 'registers an offense and autocorrects' do
+      expect_offense(<<~RUBY, '/path/to/ee/foo.rb')
+        class Foo
+          def bar
+            JSON.parse('{ "foo": "bar" }')
+            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Prefer `Gitlab::Json` over calling `JSON` directly. [...]
+          end
+        end
+      RUBY
+
+      expect_correction(<<~RUBY)
+        class Foo
+          def bar
+            ::Gitlab::Json.parse('{ "foo": "bar" }')
           end
         end
       RUBY
@@ -18,12 +47,20 @@ RSpec.describe RuboCop::Cop::Gitlab::Json do
   end
 
   context 'when ActiveSupport::JSON is called' do
-    it 'registers an offense' do
+    it 'registers an offense and autocorrects' do
       expect_offense(<<~RUBY)
         class Foo
           def bar
             ActiveSupport::JSON.parse('{ "foo": "bar" }')
-            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Avoid calling `JSON` directly. [...]
+            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Prefer `Gitlab::Json` over calling `JSON` directly. [...]
+          end
+        end
+      RUBY
+
+      expect_correction(<<~RUBY)
+        class Foo
+          def bar
+            Gitlab::Json.parse('{ "foo": "bar" }')
           end
         end
       RUBY