Rspec Mocks：从方法调用中模拟/产生块

Question

我有这样的代码：

Net::SSH.start(@server, @username, :password => @password) do |ssh|
            output = ssh.exec!(@command)
            @logger.info 'SSH output: '
            @logger.info output
        end

我可以像这样使用 RSpec 的模拟框架来模拟 SSH.Start，告诉我我已经启动了 SSH 会话：

Net::SSH.should_receive(:start).with("server", "user", :password => "secret") do
            @started = true
        end

这告诉我是否已经 @started ssh 会话。现在我需要模拟 ssh.exec！方法，这很简单：

Net::SSH.should_receive(:exec!).with("command")...

但是我如何产生/调用包含 ssh.exec 的块！方法调用，因为我已经模拟了 SSH.start 方法？我可能可以调用一些简单的方法来执行这个块，但我不知道它是什么，也找不到关于 rspec 模拟框架的任何真正好的解释/文档。

Answer 1

Net::SSH.should_receive(:start).with(
  "server", "user", :password => "secret").and_yield(
  "whatever value the block should yield")

不确定为什么需要设置 @started，因为 should_receive 验证该方法已被调用。