lib/active_null.rb in active_null-1.0.0 vs lib/active_null.rb in active_null-1.1.0

@@ -12,12 +12,18 @@
 
   def polymorphic_null_defaults
     @polymorphic_null_defaults || {}
   end
 
-  def null_model(&block)
+  def null_model(method_name=nil, &block)
     @null_model_overrides = if block_given?
       Module.new.tap { |m| m.module_eval(&block) }
+    end
+
+    if method_name
+      singleton_class.class_eval do
+        define_method(method_name) { null }
+      end
     end
   end
 
   def find_by(*args, &block)
     super || null