1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
import unittest
from bottle import static_file, request, response, parse_date, parse_range_header, Bottle, tob
import wsgiref.util
import os
import tempfile
import time
basename = os.path.basename(__file__)
root = os.path.dirname(__file__)
class TestDateParser(unittest.TestCase):
def test_rfc1123(self):
"""DateParser: RFC 1123 format"""
ts = time.time()
rs = time.strftime("%a, %d %b %Y %H:%M:%S GMT", time.gmtime(ts))
self.assertEqual(int(ts), int(parse_date(rs)))
def test_rfc850(self):
"""DateParser: RFC 850 format"""
ts = time.time()
rs = time.strftime("%A, %d-%b-%y %H:%M:%S GMT", time.gmtime(ts))
self.assertEqual(int(ts), int(parse_date(rs)))
def test_asctime(self):
"""DateParser: asctime format"""
ts = time.time()
rs = time.strftime("%a %b %d %H:%M:%S %Y", time.gmtime(ts))
self.assertEqual(int(ts), int(parse_date(rs)))
def test_bad(self):
"""DateParser: Bad format"""
self.assertEqual(None, parse_date('Bad 123'))
class TestSendFile(unittest.TestCase):
def setUp(self):
e = dict()
wsgiref.util.setup_testing_defaults(e)
b = Bottle()
request.bind(e)
response.bind()
def test_valid(self):
""" SendFile: Valid requests"""
out = static_file(basename, root=root)
self.assertEqual(open(__file__,'rb').read(), out.body.read())
def test_invalid(self):
""" SendFile: Invalid requests"""
self.assertEqual(404, static_file('not/a/file', root=root).status_code)
f = static_file(os.path.join('./../', basename), root='./views/')
self.assertEqual(403, f.status_code)
try:
fp, fn = tempfile.mkstemp()
os.chmod(fn, 0)
self.assertEqual(403, static_file(fn, root='/').status_code)
finally:
os.close(fp)
os.unlink(fn)
def test_mime(self):
""" SendFile: Mime Guessing"""
f = static_file(basename, root=root)
self.assertTrue(f.headers['Content-Type'].split(';')[0] in ('application/x-python-code', 'text/x-python'))
f = static_file(basename, root=root, mimetype='some/type')
self.assertEqual('some/type', f.headers['Content-Type'])
f = static_file(basename, root=root, mimetype='text/foo')
self.assertEqual('text/foo; charset=UTF-8', f.headers['Content-Type'])
f = static_file(basename, root=root, mimetype='text/foo', charset='latin1')
self.assertEqual('text/foo; charset=latin1', f.headers['Content-Type'])
def test_ims(self):
""" SendFile: If-Modified-Since"""
request.environ['HTTP_IF_MODIFIED_SINCE'] = time.strftime("%a, %d %b %Y %H:%M:%S GMT", time.gmtime())
res = static_file(basename, root=root)
self.assertEqual(304, res.status_code)
self.assertEqual(int(os.stat(__file__).st_mtime), parse_date(res.headers['Last-Modified']))
self.assertAlmostEqual(int(time.time()), parse_date(res.headers['Date']))
request.environ['HTTP_IF_MODIFIED_SINCE'] = time.strftime("%a, %d %b %Y %H:%M:%S GMT", time.gmtime(100))
self.assertEqual(open(__file__,'rb').read(), static_file(basename, root=root).body.read())
def test_download(self):
""" SendFile: Download as attachment """
f = static_file(basename, root=root, download="foo.mp3")
self.assertEqual('audio/mpeg', f.headers['Content-Type'])
f = static_file(basename, root=root, download=True)
self.assertEqual('attachment; filename="%s"' % basename, f.headers['Content-Disposition'])
request.environ['HTTP_IF_MODIFIED_SINCE'] = time.strftime("%a, %d %b %Y %H:%M:%S GMT", time.gmtime(100))
f = static_file(basename, root=root)
self.assertEqual(open(__file__,'rb').read(), f.body.read())
def test_range(self):
request.environ['HTTP_RANGE'] = 'bytes=10-25,-80'
f = static_file(basename, root=root)
c = open(__file__, 'rb'); c.seek(10)
self.assertEqual(c.read(16), tob('').join(f.body))
self.assertEqual('bytes 10-25/%d' % len(open(__file__, 'rb').read()),
f.headers['Content-Range'])
self.assertEqual('bytes', f.headers['Accept-Ranges'])
def test_range_parser(self):
r = lambda rs: list(parse_range_header(rs, 100))
self.assertEqual([(90, 100)], r('bytes=-10'))
self.assertEqual([(10, 100)], r('bytes=10-'))
self.assertEqual([(5, 11)], r('bytes=5-10'))
self.assertEqual([(10, 100), (90, 100), (5, 11)], r('bytes=10-,-10,5-10'))
|