Added missing configuration for assign shams.

panthomakos · Jun 13, 2012 · 44ae4ef · 44ae4ef
1 parent e8cebb7
commit 44ae4ef
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/sham/config.rb b/lib/sham/config.rb
@@ -1,4 +1,5 @@
 require 'sham/shammable'
+require 'sham/config/assign'
 require 'sham/config/attributes'
 require 'sham/config/parameters'
 require 'sham/config/empty'
@@ -26,6 +27,10 @@ def initialize klass, name
       @name = name
     end
 
+    def assign(&config)
+      @klass.add_sham_config(@name, Sham::Config::Assign.new(config))
+    end
+
     def attributes(&config)
       @klass.add_sham_config(@name, Sham::Config::Attributes.new(config))
     end

diff --git a/spec/lib/sham/config_spec.rb b/spec/lib/sham/config_spec.rb
@@ -151,6 +151,17 @@ class B; end
     end
   end
 
+  it 'configures assign shams' do
+    Sham.config(parent) do |c|
+      c.assign{ { :first => 'first' } }
+    end
+
+    instance = stub
+    parent.stub(:new){ instance }
+    instance.should_receive(:first=).with('first')
+    parent.sham!.should == instance
+  end
+
   it 'configures attribute shams' do
     attributes = { :first => 'first', :last => 'last' }
     Sham.config(parent) do |c|