lib/analyst/entities/method.rb in analyst-0.14.2 vs lib/analyst/entities/method.rb in analyst-0.15.0

@@ -1,27 +1,41 @@
 module Analyst
   module Entities
 
     class InstanceMethod < Entity
+
+      handles_node :def
+
+      def kind
+        "Instance Method"
+      end
+
       def name
         ast.children.first.to_s
       end
+
       def full_name
         parent.full_name + '#' + name
       end
     end
 
-    # TODO HERE
     class ClassMethod < Entity
+      def kind
+        "Class Method"
+      end
+
       def name
         ast.children.first.to_s
       end
+
       def full_name
         parent.full_name + '::' + name
       end
     end
 
     class SingletonMethod < Entity
+
+      handles_node :defs
 
       # NOTE: not a public API -- used by Entities::Class
       def target
         target, name, params, content = ast.children
         target