mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-06 00:02:08 -04:00
79 lines
2.6 KiB
Ruby
79 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Requests
|
|
module Plugins
|
|
module Cookies
|
|
def test_plugin_cookies_get
|
|
client = HTTPX.plugin(:cookies)
|
|
assert client.respond_to?(:cookies), "client should be cookie-enabled"
|
|
response = client.get(cookies_uri)
|
|
assert response.respond_to?(:cookies), "response should have cookies"
|
|
body = json_body(response)
|
|
assert body.key?("cookies")
|
|
assert body["cookies"].empty?
|
|
|
|
session_response = client.cookies("abc" => "def").get(cookies_uri)
|
|
body = json_body(session_response)
|
|
assert body.key?("cookies")
|
|
assert body["cookies"]["abc"] == "def", "abc wasn't properly set"
|
|
end
|
|
|
|
def test_plugin_cookies_set
|
|
client = HTTPX.plugin(:cookies)
|
|
session_cookies = { "a" => "b", "c" => "d" }
|
|
session_uri = cookies_set_uri(session_cookies)
|
|
session_response = client.get(session_uri)
|
|
verify_status(session_response, 302)
|
|
verify_cookies(session_response.cookies, session_cookies)
|
|
|
|
# first iteration sets the session
|
|
response = client.get(cookies_uri)
|
|
body = json_body(response)
|
|
assert body.key?("cookies")
|
|
verify_cookies(body["cookies"], session_cookies)
|
|
|
|
extra_cookie_response = client.cookies("c" => "d").get(cookies_uri)
|
|
body = json_body(extra_cookie_response)
|
|
assert body.key?("cookies")
|
|
verify_cookies(body["cookies"], session_cookies.merge("c" => "d"))
|
|
end
|
|
|
|
def test_plugin_cookies_follow
|
|
client = HTTPX.plugins(:follow_redirects, :cookies)
|
|
session_cookies = { "a" => "b", "c" => "d" }
|
|
session_uri = cookies_set_uri(session_cookies)
|
|
|
|
response = client.get(session_uri)
|
|
verify_status(response, 200)
|
|
assert response.uri.to_s == cookies_uri
|
|
body = json_body(response)
|
|
assert body.key?("cookies")
|
|
verify_cookies(body["cookies"], session_cookies)
|
|
end
|
|
|
|
private
|
|
|
|
def cookies_uri
|
|
build_uri("/cookies")
|
|
end
|
|
|
|
def cookies_set_uri(cookies)
|
|
build_uri("/cookies/set?" + URI.encode_www_form(cookies))
|
|
end
|
|
|
|
def verify_cookies(jar, cookies)
|
|
assert !jar.nil? && !jar.empty?, "there should be cookies in the response"
|
|
assert jar.all? { |cookie|
|
|
case cookie
|
|
when HTTP::Cookie
|
|
cookies.one? { |k, v| k == cookie.name && v == cookie.value }
|
|
else
|
|
cookie_name, cookie_value = cookie
|
|
cookies.one? { |k, v| k == cookie_name && v == cookie_value }
|
|
end
|
|
}, "jar should contain all expected cookies"
|
|
end
|
|
end
|
|
end
|
|
end
|