add ClassDefinition#find_method_by_name

Author: flyerhzm

lib/helpers/parse_ruby.rb

@@ -161,6 +161,10 @@ def find_classes_by_ancestor(superclass)
       @modules.flat_map { |mod| mod.find_classes_by_ancestor(superclass) }
   end
 
+  def find_method_by_name(name)
+    methods.find { |method_definition| method_definition.name == name  }
+  end
+
   def full_name
     names = [name]
     this = self

spec/helpers/parse_ruby_spec.rb

@@ -99,7 +99,7 @@ def user_type
     })
   end
 
-  it 'finds classes by full_name' do
+  it 'finds class by full_name' do
     rewriter =
       Synvert::Rewriter.new 'test', 'ruby_parse_helper' do
         call_helper 'ruby/parse'
@@ -148,4 +148,28 @@ class UserJob < SynvertJob
     classes = definitions.find_classes_by_ancestor('ApplicationJob')
     expect(classes.map(&:full_name)).to eq(['SynvertJob', 'Synvert::UserJob'])
   end
+
+  it 'finds method by name' do
+    rewriter =
+      Synvert::Rewriter.new 'test', 'ruby_parse_helper' do
+        call_helper 'ruby/parse'
+      end
+
+    FileUtils.mkdir_p('app/models')
+    File.write('app/models/user.rb', <<~EOF)
+      class User < ApplicationRecord
+        def activate
+        end
+
+        def deactivate
+        end
+      end
+    EOF
+
+    rewriter.process
+
+    definitions = rewriter.load_data(:ruby_definitions)
+    class_definition = definitions.find_class_by_full_name('User')
+    expect(class_definition.find_method_by_name('deactivate').name).to eq('deactivate')
+  end
 end