@@ -2,7 +2,7 @@ module IgnoreNil def ignore_nil(&block) begin yield - rescue NoMethodError => e + rescue NoMethodError, RuntimeError => e if (e.message =~ /You have a nil object when you didn't expect it/) || (e.message =~ /undefined method `.*?' for nil:NilClass/) || (e.message =~ /^Called id for nil/) return nil lib/ignore_nil.rb @@ -23,4 +23,11 @@ class IgnoreNilTest < Test::Unit::TestCase } end + def test_from_inside_map + assert_nil ignore_nil { [1,2,3,nil].map{|i| i * 3 } } + end + + def test_whiny_nil + assert_nil ignore_nil { raise RuntimeError.new("Called id for nil, which would mistakenly be 4 -- if you really wanted the id of nil, use object_id") } + end end \ No newline at end of file test/ignore_nil_test.rb 2c73a4e5503773f3122417e7ef8a213eb83467b4 Steven Soroka ssoroka78@gmail.com http://github.com/ssoroka/ignore_nil/commit/d8f73517caed3d8e781467f8e7660ada9cb6c57e d8f73517caed3d8e781467f8e7660ada9cb6c57e 2008-12-11T09:17:01-08:00 2008-12-11T09:17:01-08:00 added support for nil.id raising RuntimeError with whinynil de11a6e2eef9ec68929874ba82f6942a6e9fc275 Steven Soroka ssoroka78@gmail.com