Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-02-24 00:29:53 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #193 from vbanos/login-tests

Browse Source 
Thanks, @vbanos!
This commit is contained in:
Barbara Miller 2020-04-14 09:42:20 -07:00 committed by GitHub
commit a8734bcc11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 76 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
tests/htdocs/favicon.ico Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 4.2 KiB

12
tests/htdocs/site11/form-no-login.html Normal file
View File

@ -0,0 +1,12 @@
<html>
<head>
<title>Form without proper username/password fields</title>
</head>
<body>
<form action="/login-action" method="POST">
<input type="text" name="username" />
<input type="text" name="foo" />
<input type="submit" name="login" value="login" />
</form>
</body>
</html>

12
tests/htdocs/site11/form1.html Normal file
View File

@ -0,0 +1,12 @@
<html>
<head>
<title>Detect login form and use it</title>
</head>
<body>
<form action="/login-action" method="POST">
<input type="email" name="email" />
<input type="password" name="password" />
<input type="submit" name="login" value="login" />
</form>
</body>
</html>

52
tests/test_brozzling.py
View File

@ -72,6 +72,18 @@ def httpd(request):
else: else:
super().do_GET() super().do_GET()
def do_POST(self):
if self.path == '/login-action':
self.send_response(200)
payload = b'login successfull\n'
self.send_header('Content-Type', 'text/plain;charset=utf-8')
self.send_header('Content-Length', len(payload))
self.end_headers()
self.wfile.write(payload)
else:
super().do_POST()
# SimpleHTTPRequestHandler always uses CWD so we have to chdir # SimpleHTTPRequestHandler always uses CWD so we have to chdir
os.chdir(os.path.join(os.path.dirname(__file__), 'htdocs')) os.chdir(os.path.join(os.path.dirname(__file__), 'htdocs'))
@ -247,3 +259,43 @@ def test_proxy_down():
with pytest.raises(brozzler.ProxyError): with pytest.raises(brozzler.ProxyError):
worker.brozzle_page(browser, site, page) worker.brozzle_page(browser, site, page)
def test_try_login(httpd):
"""Test try_login behavior.
"""
response_urls = []
def on_response(msg):
response_urls.append(msg['params']['response']['url'])
chrome_exe = brozzler.suggest_default_chrome_exe()
form_url = 'http://localhost:%s/site11/form1.html' % httpd.server_port
favicon_url = 'http://localhost:%s/favicon.ico' % httpd.server_port
login_url = 'http://localhost:%s/login-action' % httpd.server_port
# When username and password are defined and initial page has login form,
# detect login form, submit login, and then return to the initial page.
username = 'user1'
password = 'pass1'
with brozzler.Browser(chrome_exe=chrome_exe) as browser:
browser.browse_page(form_url, username=username, password=password,
on_response=on_response)
assert len(response_urls) == 4
assert response_urls[0] == form_url
assert response_urls[1] == favicon_url
assert response_urls[2] == login_url
assert response_urls[3] == form_url
# When username and password are not defined, just load the initial page.
response_urls = []
with brozzler.Browser(chrome_exe=chrome_exe) as browser:
browser.browse_page(form_url, on_response=on_response)
assert len(response_urls) == 2
assert response_urls[0] == form_url
assert response_urls[1] == favicon_url
# when the page doesn't have a form with username/password, don't submit it
response_urls = []
form_without_login_url = 'http://localhost:%s/site11/form-no-login.html' % httpd.server_port
with brozzler.Browser(chrome_exe=chrome_exe) as browser:
browser.browse_page(form_without_login_url, username=username,
password=password, on_response=on_response)
assert len(response_urls) == 2
assert response_urls[0] == form_without_login_url
assert response_urls[1] == favicon_url