Skip to content

Commit fc12a66

Browse files
author
ravishankar
committed
Added cookies test cases
1 parent 1787d08 commit fc12a66

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

tests/test_driver.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
from s_tool.driver import SeleniumDriver
2+
3+
from . import TEST_URL
4+
5+
6+
def test_cookies():
7+
with SeleniumDriver("firefox", headless=True) as obj:
8+
obj.get("http://google.com")
9+
10+
# Validate cookies type
11+
assert type(obj.cookies()) is dict
12+
13+
# Drop all cookies
14+
obj.set_cookies(drop_all=True)
15+
assert obj.cookies() == {}
16+
17+
# Set new cookies
18+
19+
obj.set_cookies(name="s-tool", version="1.0")
20+
cookies = obj.cookies()
21+
assert cookies.get("name") == "s-tool"
22+
23+
# Drop cookie by name
24+
obj.set_cookies(drop_keys={"version"})
25+
cookies = obj.cookies()
26+
assert cookies.get("version", 1) == 1

0 commit comments

Comments
 (0)