@@ -11,11 +11,23 @@ module JRSplenda end end + def splenda_mock_attr(arg, options = {}) + store_mock_in_attr(arg, options) do |class_to_mock| + splenda_mock(class_to_mock) + end + end + def splenda_partial_mock(arg) internal_mock(arg) do |class_name| partial_mock(class_name) end end + + def splenda_partial_mock_attr(arg, options = {}) + store_mock_in_attr(arg, options) do |class_to_mock| + splenda_partial_mock(class_to_mock) + end + end private def internal_mock(class_name, &block) @@ -63,6 +75,16 @@ module JRSplenda end mock end + + def store_mock_in_attr(class_to_mock, options, &block) + default_field_name = stringify_class(class_to_mock).split('.').last.underscore + attr_name = "@" + options.fetch(:store_in, default_field_name).to_s + should_preserve_attr = options[:preserve_existing_attr] + current_attr_val = instance_variable_get(attr_name) + unless current_attr_val && should_preserve_attr + instance_variable_set(attr_name, yield(class_to_mock)) + end + end def stringify_class(klass) name = nil lib/jrsplenda/jrsplenda_mock_helper.rb @@ -1,12 +1,12 @@ require File.dirname(__FILE__) + '/spec_helper.rb' - + describe "A mock helper" do include JRSplenda::MockHelper it "should create a mock of a Java class when give a Java class name when the Java class has not been imported" do splenda_mock('fixtures.PrivateField').should be_a_kind_of(PrivateField) end - + it "should create a mock of a Java class when give a Java class name when the Java class has been imported" do splenda_mock('fixtures.PrivateField').should be_a_kind_of(PrivateField) end @@ -15,7 +15,7 @@ describe "A mock helper" do import 'fixtures.PrivateField' splenda_mock(PrivateField).should be_a_kind_of(PrivateField) end - + it "should receive calls delegated by Java objects" do import 'fixtures.DelegateToPublicInstanceMethod' delegatingObj = DelegateToPublicInstanceMethod.new @@ -24,4 +24,23 @@ describe "A mock helper" do delegatingObj.publicInstanceMethod = mock delegatingObj.delegateDoSomething().should == "55" end + + describe "when creating a mock object" do + it "should optionally store the created mock in a Ruby member variable by convention" do + splenda_mock_attr('fixtures.PrivateField') + @private_field.should_not be_nil + end + + it "should optionally store the created mock in a Ruby member variable specified by the caller" do + splenda_mock_attr('fixtures.PrivateField', :store_in => :another_field) + @another_field.should_not be_nil + end + + it "should optionally not overwrite existing member variables" do + splenda_mock_attr('fixtures.PrivateField') + p1 = @private_field + splenda_mock_attr('fixtures.PrivateField', :preserve_existing_attr => true) + p1.should equal(@private_field) + end + end end \ No newline at end of file spec/mock_helper_spec.rb 7a22429211a4c6116b5ef236acf69eadc2606250 Evan David Light evan@tiggerpalace.com http://github.com/elight/jrsplenda/commit/7c0ff080b51e1aa79d7181d1e13a640d00a5b756 7c0ff080b51e1aa79d7181d1e13a640d00a5b756 2008-08-15T22:44:38-07:00 2008-08-15T22:44:38-07:00 Restored splenda_mock_attr -- actually useful for creating mocks while iterating over an array of class name. Should leave it in 0a6b940b4ba3fe5a1beac4998b06524868867bf4 Evan David Light evan@tiggerpalace.com