lib/simplabs/excellent/parsing/call_context.rb in simplabs-excellent-1.4.2 vs lib/simplabs/excellent/parsing/call_context.rb in simplabs-excellent-1.5.0

@@ -14,19 +14,35 @@
         def initialize(exp, parent)
           super
           @receiver  = exp[1].is_a?(Sexp) ? (exp[1][1].nil? ? exp[1][2].to_s : exp[1][1].to_s) : nil
           @method    = exp[2].to_s
           @full_name = [@receiver, @method].compact.join('.')
+          record_validation
+          record_call
         end
 
         def <=>(other)
           @full_name <=> other.full_name
         end
         alias :eql? :'<=>'
 
         def hash
           @full_name.hash
         end
+
+        private
+
+          def record_call
+            parent = @parent
+            parent = parent.parent until parent.nil? || parent.is_a?(MethodContext) || parent.is_a?(BlockContext) || parent.is_a?(SingletonMethodContext)
+            parent.record_call_to(self) if parent
+          end
+
+          def record_validation
+            if ClassContext::VALIDATIONS.include?(@method) && @parent && @parent.is_a?(ClassContext) && @parent.active_record_model?
+              @parent.validations << @method
+            end
+          end
 
       end
 
     end