Allow initialize method and single ivar

1 files changed, 29 insertions, 0 deletions

diff --git a/spec/rubocop/cop/module_with_instance_variables_spec.rb b/spec/rubocop/cop/module_with_instance_variables_spec.rb
index bac39117dba..72f6b01aa81 100644
--- a/spec/rubocop/cop/module_with_instance_variables_spec.rb
+++ b/spec/rubocop/cop/module_with_instance_variables_spec.rb
@@ -137,6 +137,35 @@ describe RuboCop::Cop::ModuleWithInstanceVariables do
     it_behaves_like 'not registering offense'
   end
 
+  context 'when source is using simple ivar' do
+    let(:source) do
+      <<~RUBY
+        module M
+          def f?
+            @f
+          end
+        end
+      RUBY
+    end
+
+    it_behaves_like 'not registering offense'
+  end
+
+  context 'when source is defining initialize' do
+    let(:source) do
+      <<~RUBY
+        module M
+          def initialize
+            @a = 1
+            @b = 2
+          end
+        end
+      RUBY
+    end
+
+    it_behaves_like 'not registering offense'
+  end
+
   context 'when source is using simple or ivar assignment with other ivar' do
     let(:source) do
       <<~RUBY