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

new test test_needs_browsing

Browse Source 
currently exposes bug in resolving "location" response header
This commit is contained in:
Noah Levitt 2018-01-26 10:59:18 -08:00
parent 67d5a0e671
commit bf5401283e
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/test_units.py
View File

@ -310,3 +310,30 @@ def test_thread_raise_second_with_block():
th.join()
assert isinstance(thread_caught_exception, Exception2)
def test_needs_browsing():
# only one test case here right now, which exposed a bug
class ConvenientHeaders(http.client.HTTPMessage):
def __init__(self, headers):
http.client.HTTPMessage.__init__(self)
for (k, v) in headers.items():
self.add_header(k, v)
page = brozzler.Page(None, {
'url':'http://example.com/a'})
spy = brozzler.worker.YoutubeDLSpy()
spy.transactions.append({
'url': 'http://example.com/a',
'method': 'HEAD',
'status_code': 301,
'response_headers': ConvenientHeaders({'Location': '/b'})})
spy.transactions.append({
'url': 'http://example.com/b',
'method': 'GET',
'status_code': 200,
'response_headers': ConvenientHeaders({
'Content-Type': 'application/pdf'})})
assert not brozzler.worker.BrozzlerWorker._needs_browsing(None, page, spy)