allow instance_variable_or_write_node and class_variable_or_write_node in ruby/prefer-endless-method

Author: flyerhzm

lib/ruby/prefer-endless-method.rb

@@ -28,7 +28,7 @@ def one_plus_one = 1 + 1
 
       first_body_node = node.body.body.first
       break if %i[if_node unless_node].include?(first_body_node.type) && first_body_node.end_keyword.nil?
-      break if %i[multi_write_node instance_variable_or_write_node class_variable_or_write_node or_node and_node hash_node].include?(first_body_node.type)
+      break if %i[multi_write_node or_node and_node hash_node].include?(first_body_node.type)
       break if first_body_node.type == :call_node && first_body_node.opening.nil? && !first_body_node.arguments.nil? && first_body_node.closing.nil? && !first_body_node.block.nil?
 
       body_column = mutation_adapter.get_start_loc(first_body_node).column

spec/ruby/prefer-endless-method_spec.rb

@@ -116,29 +116,6 @@ def index
     include_examples 'convertable'
   end
 
-  context 'do not process for class_variable_or_write_node, instance_variable_or_write_node' do
-    let(:test_content) { <<~EOS }
-      def class_logger
-        @@logger ||= nil
-      end
-
-      def instance_logger
-        @logger ||= nil
-      end
-    EOS
-    let(:test_rewritten_content) { <<~EOS }
-      def class_logger
-        @@logger ||= nil
-      end
-
-      def instance_logger
-        @logger ||= nil
-      end
-    EOS
-
-    include_examples 'convertable'
-  end
-
   context 'do not process for or_node' do
     let(:test_content) { <<~EOS }
       def new_rating=(new_rating)