mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-06 00:02:08 -04:00
77 lines
2.3 KiB
Ruby
77 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "test_helper"
|
|
|
|
class ResponseTest < Minitest::Test
|
|
include HTTPX
|
|
|
|
def test_response_status
|
|
r1 = Response.new(request, 200, "1.1", {})
|
|
assert r1.status == 200, "unexpected status code (#{r1.status})"
|
|
r2 = Response.new(request, "200", "1.1", {})
|
|
assert r2.status == 200, "unexpected status code (#{r2.status})"
|
|
end
|
|
|
|
def test_response_headers
|
|
assert resource.headers.is_a?(Headers), "headers should have been coerced"
|
|
end
|
|
|
|
def test_response_body_write
|
|
assert resource.body.empty?, "body should be empty after init"
|
|
resource << "data"
|
|
assert resource.body == "data", "body should have been updated"
|
|
end
|
|
|
|
def test_response_body_to_s
|
|
opts = { threshold_size: 1024 }
|
|
body1 = Response::Body.new(Response.new(request, 200, "2.0", {}), opts)
|
|
assert body1.empty?, "body must be empty after initialization"
|
|
body1.write("foo")
|
|
assert body1 == "foo", "body must be updated"
|
|
body1.write("foo")
|
|
body1.write("bar")
|
|
assert body1 == "foobar", "body must buffer subsequent chunks"
|
|
|
|
body3 = Response::Body.new(Response.new(request("head"), 200, "2.0", {}), opts)
|
|
assert body3.empty?, "body must be empty after initialization"
|
|
assert body3 == "", "HEAD requets body must be empty"
|
|
end
|
|
|
|
def test_response_body_each
|
|
opts = { threshold_size: 1024 }
|
|
body1 = Response::Body.new(Response.new(request, 200, "2.0", {}), opts)
|
|
body1.write("foo")
|
|
assert body1.each.to_a == %w[foo], "must yield buffer"
|
|
body1.write("foo")
|
|
body1.write("bar")
|
|
assert body1.each.to_a == %w[foobar], "must yield buffers"
|
|
end
|
|
|
|
def test_response_body_buffer
|
|
opts = { threshold_size: 10 }
|
|
body = Response::Body.new(Response.new(request, 200, "2.0", {}), opts)
|
|
body.extend(Module.new do
|
|
attr_reader :buffer
|
|
end)
|
|
assert body.buffer.nil?, "body should not buffer anything"
|
|
body.write("hello")
|
|
assert body.buffer.is_a?(StringIO), "body should buffer to memory"
|
|
body.write(" world")
|
|
assert body.buffer.is_a?(Tempfile), "body should buffer to file after going over threshold"
|
|
end
|
|
|
|
private
|
|
|
|
def request(verb = :get, uri = "http://google.com")
|
|
Request.new(verb, uri)
|
|
end
|
|
|
|
def response(*args)
|
|
Response.new(*args)
|
|
end
|
|
|
|
def resource
|
|
@resource ||= Response.new(request, 200, "2.0", {})
|
|
end
|
|
end
|