diff --git a/lib/fake_web/stub_socket.rb b/lib/fake_web/stub_socket.rb index 4d23f7e..c1e4c17 100644 --- a/lib/fake_web/stub_socket.rb +++ b/lib/fake_web/stub_socket.rb @@ -9,7 +9,12 @@ def initialize(*args) end def closed? - @closed ||= true + @closed ||= false + @closed + end + + def close + @closed = true end def readuntil(*args) diff --git a/test/test_stub_socket_compatibility.rb b/test/test_stub_socket_compatibility.rb index 6ab9960..02ab308 100644 --- a/test/test_stub_socket_compatibility.rb +++ b/test/test_stub_socket_compatibility.rb @@ -30,4 +30,22 @@ def test_stub_socket_only_responds_to_continue_timeout_under_193_or_later socket = http.instance_variable_get(:@socket) assert_equal RUBY_VERSION >= "1.9.3", socket.respond_to?(:continue_timeout=) end + + def test_stub_socket_responds_to_close_and_always_returns_true + FakeWeb.register_uri(:get, "http://example.com", :status => [200, "OK"]) + http = Net::HTTP.new("example.com", 80) + http.get("/") + socket = http.instance_variable_get(:@socket) + assert_equal socket.close, true + end + + def test_stub_socket_tracks_closed_state + FakeWeb.register_uri(:get, "http://example.com", :status => [200, "OK"]) + http = Net::HTTP.new("example.com", 80) + http.get("/") + socket = http.instance_variable_get(:@socket) + assert !socket.closed? + socket.close + assert socket.closed? + end end