Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup header tests

Browse Source
This commit is contained in:
Kovid Goyal 2024-08-17 10:00:08 +05:30
parent b2745092c6
commit 592f250644
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 18 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/calibre/scraper/test_fetch_backend.py
View File

@ -127,38 +127,43 @@ class TestFetchBackend(unittest.TestCase):
self.dont_send_body = False self.dont_send_body = False
self.dont_send_response = False self.dont_send_response = False
def header(name, *expected):
name = name.lower()
ans = []
for k, v in r['headers'].items():
if k.lower() == name:
ans.extend(v)
self.ae(expected, tuple(ans))
try: try:
r = get() r = get()
self.ae(r['method'], 'GET') self.ae(r['method'], 'GET')
self.ae(r['request_count'], 1) self.ae(r['request_count'], 1)
self.ae(r['headers']['th'], ['1']) header('th', '1')
self.ae(r['headers']['User-Agent'], ['test-ua']) header('User-Agent', 'test-ua')
self.assertIn('Accept-Encoding', r['headers']) self.assertIn('Accept-Encoding', r['headers'])
r = get() r = get()
self.ae(r['request_count'], 2) self.ae(r['request_count'], 2)
self.ae(r['headers']['Cookie'], ['sc=1']) header('Cookie', 'sc=1')
test_with_timeout(True) test_with_timeout(True)
test_with_timeout(False) test_with_timeout(False)
r = get('/redirect') r = get('/redirect')
self.ae(r['path'], '/redirected') self.ae(r['path'], '/redirected')
self.ae(r['headers']['th'], ['1']) header('th', '1')
self.assertTrue(r['final_url'].endswith('/redirected')) self.assertTrue(r['final_url'].endswith('/redirected'))
self.ae(r['headers']['User-Agent'], ['test-ua']) header('User-Agent', 'test-ua')
r = get(headers={'th': '2', 'tc': '1'}) r = get(headers={'th': '2', 'tc': '1'})
if browser_class is Browser: header('Th', '1, 2')
self.ae(r['headers']['Th'], ['1, 2']) header('Tc', '1')
self.ae(r['headers']['Tc'], ['1'])
else:
self.ae(r['headers']['th'], ['1, 2'])
self.ae(r['headers']['tc'], ['1'])
br.set_simple_cookie('cook', 'ie') br.set_simple_cookie('cook', 'ie')
br.set_user_agent('man in black') br.set_user_agent('man in black')
r = get() r = get()
self.ae(r['headers']['User-Agent'], ['man in black']) header('User-Agent', 'man in black')
self.ae(r['headers']['Cookie'], ['sc=1; cook=ie']) header('Cookie', 'sc=1; cook=ie')
r = get(data=b'1234') r = get(data=b'1234')
self.ae(r['method'], 'POST') self.ae(r['method'], 'POST')
self.ae(r['data'], '1234') self.ae(r['data'], '1234')
header('Content-Type', 'application/x-www-form-urlencoded')
finally: finally:
br.shutdown() br.shutdown()