lib/aequitas/rule/method.rb in aequitas-0.0.1 vs lib/aequitas/rule/method.rb in aequitas-0.0.2

@@ -11,20 +11,25 @@
       attr_reader :method
 
       def initialize(attribute_name, options={})
         super
 
-        @method = options.fetch(:method, attribute_name)
+        @method         = options.fetch(:method,         attribute_name)
+        @violation_type = options.fetch(:violation_type, :unsatisfied_condition)
       end
 
       def validate(resource)
         result, error_message = resource.__send__(method)
 
         if result
           nil
         else
           Violation.new(resource, error_message, self)
         end
+      end
+
+      def violation_type(resource)
+        @violation_type
       end
 
     end # class Method
   end # class Rule
 end # module Aequitas