add cookie path attribute test

master
Kanezoh 2021-08-06 16:58:31 +09:00
parent f0afbf2c5f
commit 4f845757f9
2 changed files with 24 additions and 2 deletions

View File

@ -4,6 +4,9 @@ WebMock.stub(:get, "example.com/cookies1").to_return(headers: {"Set-Cookie" => "
WebMock.stub(:get, "example.com/cookies1_domain").to_return(headers: {"Set-Cookie" => "id=123; Domain=example.com"}) WebMock.stub(:get, "example.com/cookies1_domain").to_return(headers: {"Set-Cookie" => "id=123; Domain=example.com"})
WebMock.stub(:get, "example.com/cookies2").to_return(headers: {"Set-Cookie" => "name=kanezoh"}) WebMock.stub(:get, "example.com/cookies2").to_return(headers: {"Set-Cookie" => "name=kanezoh"})
WebMock.stub(:get, "example.com/cookies3").to_return(headers: {"Set-Cookie" => "id=456"}) WebMock.stub(:get, "example.com/cookies3").to_return(headers: {"Set-Cookie" => "id=456"})
WebMock.stub(:get, "example.com/paths").to_return(headers: {"Set-Cookie" => "id=123; Path=/paths"})
WebMock.stub(:get, "example.com/paths/hoge").to_return()
WebMock.stub(:get, "example.com/hoge/paths").to_return()
WebMock.stub(:get, "www.example.com").to_return() WebMock.stub(:get, "www.example.com").to_return()
WebMock.stub(:get, "example.com/meta_cookie").to_return(body: WebMock.stub(:get, "example.com/meta_cookie").to_return(body:
<<-BODY <<-BODY
@ -74,4 +77,17 @@ describe "Mechanize Cookie test" do
agent.get("http://www.example.com/") agent.get("http://www.example.com/")
agent.request_headers.has_key?("Cookie").should eq true agent.request_headers.has_key?("Cookie").should eq true
end end
it "doesn't send not-matched path if path attribute is set" do
agent = Mechanize.new
agent.get("http://example.com/paths")
agent.get("http://example.com/")
agent.request_headers.has_key?("Cookie").should eq false
agent.get("http://example.com/paths")
agent.request_headers.has_key?("Cookie").should eq true
agent.get("http://example.com/paths/hoge")
agent.request_headers.has_key?("Cookie").should eq true
agent.get("http://example.com/hoge/paths")
agent.request_headers.has_key?("Cookie").should eq false
end
end end

View File

@ -131,7 +131,7 @@ module MechanizeCr
host = uri.host host = uri.host
valid_cookies = ::HTTP::Cookies.new valid_cookies = ::HTTP::Cookies.new
request_cookies.each do |cookie| request_cookies.each do |cookie|
valid_cookies << cookie if cookie.valid_cookie?(host) valid_cookies << cookie if cookie.valid_cookie?(uri)
end end
valid_cookies valid_cookies
end end
@ -157,7 +157,13 @@ class HTTP::Cookie
@origin = origin @origin = origin
end end
def valid_cookie?(host) def valid_cookie?(uri)
host = uri.host
if path
bool = uri.path.try &.=~(/^#{path}.*/)
return false if bool.nil?
end
if domain if domain
host.try &.=~(/.*#{domain.try &.gsub(".", "\.")}$/) host.try &.=~(/.*#{domain.try &.gsub(".", "\.")}$/)
else else