Method: JavaMethod#add_easymock_stub_methods_to

Defined in:

lib/java_testing_guff/qdox_extensions.rb

#add_easymock_stub_methods_to(generated_clazz) ⇒ Object

# File 'lib/java_testing_guff/qdox_extensions.rb', line 407

def add_easymock_stub_methods_to(generated_clazz)
    if (setter?)
        return
    end

    if (getter?)
        add_easymock_getter_stub_to(generated_clazz)
    else
        method_builder = generated_clazz.add_method("with#{name.camelize}").returns('T')
        with_parameters_and_return do |p|
            p.add_takes_clause_to(method_builder)
        end
        with_exceptions do |e|
            e.add_throws_clause_to(method_builder)
        end

        method_builder.body {|body|
            if (returns_something?)
                body.line("#{easyMock}.expect(mock.#{name}(#{parameter_list})).andStubReturn(result);")
            else
                body.line("mock.#{name}(#{parameter_list});")
                body.line("#{easyMock}.expectLastCall().asStub();")
            end
            body.line('return (T) this;')
        }
    end
end