如何从 Ruby 中的 Mocha 模拟返回动态值

Question

我的问题的要点如下：-

我正在用 Ruby 为下面表示为“post_to_embassy”的方法编写一个 Mocha 模拟。为了描述问题，实际方法的作用并不是我们真正关心的。但我需要模拟返回动态值。下面的过程“&prc”代替实际方法正确执行。但 Mocha 中的“with”方法只允许返回布尔值。所以下面的代码输出 nil。我需要它输出通过 orderInfoXml 传递的值。有谁知道我可以使用的替代方法？

require 'rubygems'
require 'mocha'
include Mocha::API

class EmbassyInterface 
  def post_to_embassy(xml)
    puts "This is from the original class:-"
    puts xml
    return xml
  end
end

orderInfoXml = "I am THE XML"

mock = EmbassyInterface.new
prc = Proc.new do |orderXml| 
  puts "This is from the mocked proc:-"
  puts orderXml
  orderXml
end

mock.stubs(:post_to_embassy).with(&prc)
mock_result = mock.post_to_embassy(orderInfoXml)
p mock_result
#p prc.call("asd")

输出：-

This is from the mocked proc:-
I am THE XML
nil

Answer 1

我不确定是否有一个完美的方法。但为了让生活更轻松，而不是存根每个可能的响应（如另一个答案中所述），您可以使用 Mocha 的 yields 方法。

require "rubygems"
require "mocha"

include Mocha::API

class EmbassyInterface

  def post_to_embassy(xml)
    puts "This is form the original class:-"
    puts xml
    xml
  end
end

order_info_xml = "I am the xml"

mock = EmbassyInterface.new

prc = Proc.new do |order_xml|
  puts "This is from the mocked proc:-"
  puts order_xml
  order_xml
end

mock.stubs(:post_to_embassy).yields prc

prc_return = nil

mock.post_to_embassy { |value| prc_return = value.call("hello world") }
puts prc_return

mock.post_to_embassy { |value| prc_return = value.call("foo") }
puts prc_return

输出：

This is from the mocked proc:-
hello world
hello world

This is from the mocked proc:-
foo
foo

这将要求您分配 prc 的返回值，而且它并不完全漂亮（imo）。但是，您不必取消每个期望，这会给您很大的自由。

Answer 2

一般来说，您通常最好在测试中指定显式返回值。如果您引入单独的逻辑来确定返回什么值，它往往会使测试难以理解和维护。

我建议您使用 Expectation#with 和合适的 ParameterMatchers 显式定义不同参数值的返回值或使用 StateMachine 功能。

Answer 3

Mocha 似乎不支持这一点。将其添加到您的 test_helper.rb：

# Replace klass's method_name with method_implementation
def stub_replace(klass, method_name, &method_implementation)
  klass.singleton_class.send(:alias_method, "#{method_name}_mock_backup", method_name)
  klass.define_singleton_method(method_name, method_implementation)
end

def undo_stub_replace(klass, method_name)
  klass.singleton_class.send(:alias_method, method_name, "#{method_name}_mock_backup")
end

然后将测试的最后 4 行替换为：

stub_replace(EmbassyInterface, :post_to_embassy, &prc)
mock_result = mock.post_to_embassy(orderInfoXml)
p mock_result

# cleanup
undo_stub_replace(EmbassyInterface, :post_to_embassy)

Answer 4

我还没有找到一种方法使模拟方法的输出完全动态，但如果您的输入数量有限且已知，则可以使输出正常工作。

require 'rubygems'
require 'mocha'
include Mocha::API

class EmbassyInterface 
  def post_to_embassy(xml)
    "original: #{xml}"
  end
end

to_mock = EmbassyInterface.new
orderInfoXml1 = "I am the first XML."
orderInfoXml2 = "I am the second XML."

p to_mock.post_to_embassy(orderInfoXml1)

prc = Proc.new do |xml| 
  "mocked: #{xml}"
end
to_mock.stubs(:post_to_embassy).with(orderInfoXml1).returns(prc.call(orderInfoXml1))
to_mock.stubs(:post_to_embassy).with(orderInfoXml2).returns(prc.call(orderInfoXml2))

p to_mock.post_to_embassy(orderInfoXml1)
p to_mock.post_to_embassy(orderInfoXml2)
p to_mock.post_to_embassy(orderInfoXml1)

输出：

"original: I am the first XML."
"mocked: I am the first XML."
"mocked: I am the second XML."
"mocked: I am the first XML."