karchnu/mechanize.cr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add cookie test

Browse Source
This commit is contained in:
Kanezoh 2021-06-13 07:58:50 +09:00
parent d96d597547
commit f3bd9845e3
2 changed files with 33 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
spec/agent_spec.cr
View File

@ -1,4 +1,7 @@
require "./spec_helper"
WebMock.stub(:get, "example.com/")
WebMock.stub(:get, "example.com/cookies1").to_return(headers: {"Set-Cookie" => "id=123"})
WebMock.stub(:get, "example.com/cookies2").to_return(headers: {"Set-Cookie" => "name=kanezoh"})
WebMock.stub(:get, "html_example.com").to_return(body:
<<-BODY
<html>
@ -21,6 +24,7 @@ WebMock.stub(:post, "http://html_example.com/post_path").
to_return(body: "success")
describe "Mechanize Agent test" do
it "fill and submit form" do
agent = Mechanize.new
page = agent.get("http://html_example.com/")
form = page.forms[0]
@ -30,3 +34,23 @@ describe "Mechanize Agent test" do
page.not_nil!.code.should eq 200
page.not_nil!.body.should eq "success"
end
it "receive and send cookie" do
agent = Mechanize.new
# receive cookies
agent.get("http://example.com/cookies1")
# send cookies
agent.get("http://example.com/")
agent.request_headers["Cookie"].should eq "id=123"
end
it "receive and send multiple cookies" do
agent = Mechanize.new
# receive cookies1
agent.get("http://example.com/cookies1")
# receive cookies2
agent.get("http://example.com/cookies2")
agent.get("http://example.com/")
agent.request_headers["Cookie"].should eq "id=123; name=kanezoh"
end
end

2
src/mechanize/http/agent.cr
View File

@ -74,7 +74,7 @@ module MechanizeCr
# end
#end
header_cookies = response.try &.cookies
if (header_cookies.nil?)
if (header_cookies.nil? || header_cookies.try &.empty?)
request_headers
else
if cookies.fetch(uri.host.to_s, ::HTTP::Cookies.new).empty?