stump.gemspec @@ -72,6 +72,10 @@ Stubbing and mocking that isn't painful, fanciful, and doesn't inspire any sort MyClass.proxy!(:the_method) # Returns static value but still has side effects of real method call MyClass.proxy!(:that_method, :return => "poop") + +== NOTES: + +* Be sure you call +super+ in +teardown+ if you create your own +teardown+ methods. Otherwise your mocks won't get verified! == REQUIREMENTS: @@ -83,6 +87,11 @@ Stubbing and mocking that isn't painful, fanciful, and doesn't inspire any sort $ sudo gem install jeremymcanally-stump +== ACKNOWLEDGEMENTS: + +* Thanks to Jim Weirich for Flexmock, David Chelmisky et. al. for RSpec's mocking stuff, and Brian Takita for rr. All of those things fed into stump. +* BIG thanks to Nathan Sutton (fowlduck) for helping me work through how to get +proxy!+ to work with ActiveRecord/method_missing. + == LICENSE: (The MIT License) README.rdoc @@ -36,40 +36,68 @@ class Object def proxy!(method, options = {}, &block) Stump::Mocks.add([self, method]) + if respond_to?(method) + proxy_existing_method(method, options, &block) + else + proxy_missing_method(method, options, &block) + end + end + + protected + def proxy_existing_method(method, options = {}, &block) method_alias = "__old_#{method}" meta_eval do module_eval("alias #{method_alias} #{method}") end - behavior = unless options[:return] + behavior = if options[:return] lambda do |*args| raise ArgumentError if args.length != method(method_alias.to_sym).arity - - Stump::Mocks.verify([self, method]) + Stump::Mocks.verify([self, method]) + if method(method_alias.to_sym).arity == 0 - return send(method_alias) + send(method_alias) else - return send(method_alias, *args) + send(method_alias, *args) end + + return options[:return] end else lambda do |*args| raise ArgumentError if args.length != method(method_alias.to_sym).arity - - Stump::Mocks.verify([self, method]) + Stump::Mocks.verify([self, method]) + if method(method_alias.to_sym).arity == 0 - send(method_alias) + return send(method_alias) else - send(method_alias, *args) + return send(method_alias, *args) end + end + end + meta_def method, &behavior + end + + def proxy_missing_method(method, options = {}, &block) + behavior = if options[:return] + lambda do |*args| + Stump::Mocks.verify([self, method]) + + method_missing(method, args) return options[:return] end + else + lambda do |*args| + Stump::Mocks.verify([self, method]) + + method_missing(method, args) + end end - + meta_def method, &behavior end end \ No newline at end of file lib/stump/proxy.rb @@ -1,5 +1,15 @@ require File.dirname(__FILE__) + '/test_helper.rb' +class ProxyPants + def method_missing(method_name, arguments) + if method_name.to_s =~ /fun_/ + return "whoo #{method_name} party!!" + else + super + end + end +end + class TestProxy < Test::Unit::TestCase def setup end @@ -24,7 +34,7 @@ class TestProxy < Test::Unit::TestCase def test_proxy_is_added_to_tracker proxy_me = MyStump.new - proxy_me.proxy!(:tree, "tea") + proxy_me.proxy!(:tree, :return => "tea") assert_equal [[proxy_me, :tree]], Stump::Mocks.failures end @@ -44,7 +54,7 @@ class TestProxy < Test::Unit::TestCase def test_proxy_fail stumply = MyStump.new - stumply.proxy!(:tree, "hi") + stumply.proxy!(:tree, :return => "hi") assert_raise Test::Unit::AssertionFailedError do stumpdown! @@ -53,7 +63,7 @@ class TestProxy < Test::Unit::TestCase def test_proxy_pass stumply = MyStump.new - stumply.proxy!(:tree, "hi") + stumply.proxy!(:tree, :return => "hi") stumply.tree assert_nothing_raised do @@ -61,6 +71,38 @@ class TestProxy < Test::Unit::TestCase end end + def test_proxy_method_missing + obj = ProxyPants.new + + obj.proxy!(:fun_pants) + assert_equal "whoo fun_pants party!!", obj.fun_pants + end + + def test_proxy_method_missing_with_return + obj = ProxyPants.new + + obj.proxy!(:fun_party, :return => "party time in the city!!") + assert_equal "party time in the city!!", obj.fun_party + end + + def test_proxy_method_missing_fails_with_no_method_match + obj = ProxyPants.new + + assert_raises NoMethodError do + obj.proxy!(:fail_pants) + obj.fail_pants + end + end + + def test_proxy_method_missing_fail + stumply = ProxyPants.new + stumply.proxy!(:fun_tree, :return => "hello") + + assert_raise Test::Unit::AssertionFailedError do + stumpdown! + end + end + def teardown Stump::Mocks.clear! end test/test_proxy.rb 2683329c98f6c4768e752717e14520c0f1ad8b8f Jeremy McAnally jeremymcanally@gmail.com http://github.com/jeremymcanally/stump/commit/c3bdba7bcd60f18f25f606c841744d2d8276a1ab c3bdba7bcd60f18f25f606c841744d2d8276a1ab 2008-10-07T20:38:20-07:00 2008-10-07T20:38:20-07:00 Actually add gemspec this time, add acknowledgements to README (including fowlduck for his awesome idea), and fix proxying for AR/method_missing methods 2bf004da27121427a43c11095d7268f40adfaedc Jeremy McAnally jeremymcanally@gmail.com