Add a spec to make sure we're nil-safe. [#41]

marnen · Oct 20, 2009 · b4cfd43 · b4cfd43
1 parent f82d0ee
commit b4cfd43
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/spec/models/address_spec.rb b/spec/models/address_spec.rb
@@ -106,6 +106,12 @@
         country = mock_model(Country, :name => 'US')
         Address.new(:state => mock_model(State, :country => country)).country.should == country
       end
+
+      it "should not complain if methods are called on it, even if nil" do
+        a = Address.new
+        a.country.should be_nil
+        lambda{a.country.code}.should_not raise_error
+      end
     end
 
     describe 'state' do