ruby - RSpec 和 Open-URI 我如何模拟引发 SocketError/TimeoutError

Question

我希望能够说明当 Open-Uri open() 调用超时或引发诸如 SocketError 之类的异常时，我正在按预期处理事情，但是我在这方面遇到了麻烦。

这是我的规范(针对 SocketError):

@obj.should_receive(:open).with("some_url").and_raise(SocketError)

以及我使用 open-uri 的对象部分:

begin
  resp = open(url)
  resp = resp.read
rescue SocketError
  something = true
end

然而，在这种情况下，规范会失败，出现 nil.read 错误。

这是我这周第二次遇到这个问题，上一次我试图在用 timeout() {} 包装 open() 时模拟 TimeoutError ，那次我放弃了，只是通过打开类(class)导致实际超时发生。我可以显然通过尝试调用无效的 URL 导致它抛出 SocketError，但我确信有一种正确的方法可以用 RSpec 模拟它。

更新: 深夜我显然没有想清楚，错误实际上是当我在 SocketError 之后重试 URL 时，and_raise(SocketError) 部分工作正常。

Answer 1

根据您提供的信息，您提供的行应该有效:我制作了一个小型测试类和规范(见下文)，仅具有所描述的功能，并且一切都按预期运行。如果您能提供更多上下文，这可能会有所帮助 - 例如，规范中完整的“it” block 可能会暴露一些其他问题。

如前所述，以下规范通过了，我相信它捕获了您试图验证的逻辑:

require 'rubygems'
require 'spec'

class Foo
  attr_accessor :socket_error

  def get(url)
    @socket_error = false
    begin
      resp = open(url)
      resp = resp.read
    rescue SocketError
      @socket_error = true
    end
  end
end

describe Foo do
  before do
    @foo = Foo.new
  end

  it "should handle socket errors" do
    @foo.should_receive(:open).with("http://www.google.com").and_raise(SocketError)
    @foo.get("http://www.google.com")
    @foo.socket_error.should be_true
  end
end