Merge pull request #318 from benedikt/master

Allow overrides to be `nil`

lib/fabrication/generator/base.rb

@@ -76,7 +76,7 @@ def initialize(klass)
   end
 
   def method_missing(method_name, *args, &block)
-    _attributes[method_name] || super
+    _attributes.fetch(method_name) { super }
   end
 
   protected

spec/fabrication/generator/base_spec.rb

@@ -85,7 +85,7 @@
         let(:schematic) do
            Fabrication::Schematic::Definition.new('ClassWithInit') do
              arg1 10
-             initialize_with { Struct.new(:arg1, :arg2).new(arg1, arg1 + 10) }
+             initialize_with { Struct.new(:arg1, :arg2).new(arg1, arg1.to_i + 10) }
           end
         end
 
@@ -104,6 +104,13 @@
           end
         end
 
+        context "with nil override" do
+          subject { schematic.fabricate(arg1: nil) }
+          it "saves the return value of the block as instance" do
+            expect(subject.arg1).to eq(nil)
+            expect(subject.arg2).to eq(10)
+          end
+        end
       end
     end