forked from thoughtbot/capybara-webkit
-
Notifications
You must be signed in to change notification settings - Fork 2
/
cookie_jar.rb
55 lines (44 loc) · 1.21 KB
/
cookie_jar.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
require 'webrick'
# A simple cookie jar implementation.
# Does not take special cookie attributes
# into account like expire, max-age, httponly, secure
class Capybara::Driver::Webkit::CookieJar
attr_reader :browser
def initialize(browser)
@browser = browser
end
def [](*args)
cookie = find(*args)
cookie && cookie.value
end
def find(name, domain = nil, path = "/")
# we are sorting by path size because more specific paths take
# precendence
cookies.sort_by { |c| -c.path.size }.find { |c|
c.name.downcase == name.downcase &&
(!domain || valid_domain?(c, domain)) &&
(!path || valid_path?(c, path))
}
end
protected
def valid_domain?(cookie, domain)
ends_with?(("." + domain).downcase,
normalize_domain(cookie.domain).downcase)
end
def normalize_domain(domain)
domain = "." + domain unless domain[0,1] == "."
domain
end
def valid_path?(cookie, path)
starts_with?(path, cookie.path)
end
def ends_with?(str, suffix)
str[-suffix.size..-1] == suffix
end
def starts_with?(str, prefix)
str[0, prefix.size] == prefix
end
def cookies
browser.get_cookies.map { |c| WEBrick::Cookie.parse_set_cookie(c) }
end
end