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

...

Browse Source
This commit is contained in:
Kovid Goyal 2015-05-21 14:04:03 +05:30
parent 427781683e
commit 4ffda3300c
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/calibre/srv/tests/http.py
View File

@ -6,9 +6,10 @@ from __future__ import (unicode_literals, division, absolute_import,
__license__ = 'GPL v3' __license__ = 'GPL v3'
__copyright__ = '2015, Kovid Goyal <kovid at kovidgoyal.net>' __copyright__ = '2015, Kovid Goyal <kovid at kovidgoyal.net>'
import textwrap, httplib, hashlib, zlib import textwrap, httplib, hashlib, zlib, string
from io import BytesIO from io import BytesIO
from calibre.ptempfile import PersistentTemporaryFile
from calibre.srv.tests.base import BaseTest, TestServer from calibre.srv.tests.base import BaseTest, TestServer
def headers(raw): def headers(raw):
@ -54,7 +55,7 @@ class TestHTTP(BaseTest):
def test_accept_encoding(self): # {{{ def test_accept_encoding(self): # {{{
'Test parsing of Accept-Encoding' 'Test parsing of Accept-Encoding'
from calibre.srv.http import acceptable_encoding from calibre.srv.respond import acceptable_encoding
def test(name, val, ans, allowed={'gzip'}): def test(name, val, ans, allowed={'gzip'}):
self.ae(acceptable_encoding(val, allowed), ans, name + ' failed') self.ae(acceptable_encoding(val, allowed), ans, name + ' failed')
test('Empty field', '', None) test('Empty field', '', None)
@ -170,7 +171,11 @@ class TestHTTP(BaseTest):
'Test HTTP protocol responses' 'Test HTTP protocol responses'
def handler(conn): def handler(conn):
return conn.generate_static_output('test', lambda : ''.join(conn.path)) return conn.generate_static_output('test', lambda : ''.join(conn.path))
with TestServer(handler, timeout=0.1, compress_min_size=0) as server: with TestServer(handler, timeout=0.1, compress_min_size=0) as server, PersistentTemporaryFile('test.epub') as f:
fdata = string.ascii_letters * 100
f.write(fdata)
f.close()
# Test ETag # Test ETag
conn = server.connect() conn = server.connect()
conn.request('GET', '/an_etagged_path') conn.request('GET', '/an_etagged_path')