operator method works for ruby/prefer-endless-method

Author: flyerhzm

lib/ruby/prefer-endless-method.rb

@@ -23,7 +23,7 @@ def one_plus_one = 1 + 1
 
   within_files Synvert::ALL_RUBY_FILES + Synvert::ALL_RAKE_FILES do
     find_node '.def_node[body!=nil][body.body.length=1]' do
-      break if node.name.to_s.end_with?('=')
+      break if node.name.to_s.match?(/[a-zA-Z]/) && node.name.to_s.end_with?('=')
       break if !node.parameters.nil? && node.lparen.nil? && node.rparen.nil?
 
       first_body_node = node.body.body.first

spec/ruby/prefer-endless-method_spec.rb

@@ -118,13 +118,13 @@ def index
 
   context 'do not process for or_node' do
     let(:test_content) { <<~EOS }
-      def new_rating=(new_rating)
-        self.rating = NEW_TO_OLD_MAPPER[new_rating] or raise "Unknown new rating"
+      def new_rating(old_rating)
+        NEW_TO_OLD_MAPPER[old_rating] or raise "Unknown new rating"
       end
     EOS
     let(:test_rewritten_content) { <<~EOS }
-      def new_rating=(new_rating)
-        self.rating = NEW_TO_OLD_MAPPER[new_rating] or raise "Unknown new rating"
+      def new_rating(old_rating)
+        NEW_TO_OLD_MAPPER[old_rating] or raise "Unknown new rating"
       end
     EOS
 
@@ -178,6 +178,19 @@ def remove_item_ids=(item_ids)
     include_examples 'convertable'
   end
 
+  context 'process for operator method' do
+    let(:test_content) { <<~EOS }
+      def ==(other)
+        true
+      end
+    EOS
+    let(:test_rewritten_content) { <<~EOS }
+      def ==(other) = true
+    EOS
+
+    include_examples 'convertable'
+  end
+
   context 'do not process if parameter without parentheses' do
     let(:test_content) { <<~EOS }
       def initialize item