1786.1.8
by John Arbash Meinel
[merge] Johan Rydberg test updates |
1 |
# Copyright (C) 2005, 2006 by Canonical Ltd
|
2 |
#
|
|
3 |
# This program is free software; you can redistribute it and/or modify
|
|
4 |
# it under the terms of the GNU General Public License as published by
|
|
5 |
# the Free Software Foundation; either version 2 of the License, or
|
|
6 |
# (at your option) any later version.
|
|
7 |
#
|
|
8 |
# This program is distributed in the hope that it will be useful,
|
|
9 |
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
10 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
11 |
# GNU General Public License for more details.
|
|
12 |
#
|
|
13 |
# You should have received a copy of the GNU General Public License
|
|
14 |
# along with this program; if not, write to the Free Software
|
|
15 |
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
16 |
||
17 |
"""Tests from HTTP response parsing."""
|
|
18 |
||
19 |
from cStringIO import StringIO |
|
1786.1.27
by John Arbash Meinel
Fix up the http transports so that tests pass with the new configuration. |
20 |
import mimetools |
1786.1.8
by John Arbash Meinel
[merge] Johan Rydberg test updates |
21 |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
22 |
from bzrlib import errors |
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
23 |
from bzrlib.transport import http |
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
24 |
from bzrlib.transport.http import response |
1786.1.8
by John Arbash Meinel
[merge] Johan Rydberg test updates |
25 |
from bzrlib.tests import TestCase |
26 |
||
27 |
||
1786.1.12
by John Arbash Meinel
Add tests for ResponseRange and streamline class |
28 |
class TestResponseRange(TestCase): |
29 |
"""Test the ResponseRange class."""
|
|
30 |
||
31 |
def test_cmp(self): |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
32 |
RR = response.ResponseRange |
33 |
r1 = RR(0, 10, 0) |
|
34 |
r2 = RR(15, 20, 10) |
|
1786.1.12
by John Arbash Meinel
Add tests for ResponseRange and streamline class |
35 |
self.assertTrue(r1 < r2) |
36 |
self.assertFalse(r1 > r2) |
|
37 |
self.assertTrue(r1 < 5) |
|
38 |
self.assertFalse(r2 < 5) |
|
39 |
||
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
40 |
self.assertEqual(RR(0, 10, 5), RR(0, 10, 5)) |
41 |
self.assertNotEqual(RR(0, 10, 5), RR(0, 8, 5)) |
|
42 |
self.assertNotEqual(RR(0, 10, 5), RR(0, 10, 6)) |
|
1786.1.12
by John Arbash Meinel
Add tests for ResponseRange and streamline class |
43 |
|
44 |
def test_sort_list(self): |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
45 |
"""Ensure longer ranges are sorted after shorter ones"""
|
46 |
RR = response.ResponseRange |
|
47 |
lst = [RR(3, 8, 0), 5, RR(3, 7, 0), 6] |
|
1786.1.12
by John Arbash Meinel
Add tests for ResponseRange and streamline class |
48 |
lst.sort() |
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
49 |
self.assertEqual([RR(3,7,0), RR(3,8,0), 5, 6], lst) |
1786.1.12
by John Arbash Meinel
Add tests for ResponseRange and streamline class |
50 |
|
51 |
||
1786.1.8
by John Arbash Meinel
[merge] Johan Rydberg test updates |
52 |
class TestRangeFile(TestCase): |
53 |
"""Test RangeFile."""
|
|
54 |
||
55 |
def setUp(self): |
|
56 |
content = "abcdefghijklmnopqrstuvwxyz" |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
57 |
self.fp = response.RangeFile('foo', StringIO(content)) |
1786.1.8
by John Arbash Meinel
[merge] Johan Rydberg test updates |
58 |
self.fp._add_range(0, 9, 0) |
59 |
self.fp._add_range(20, 29, 10) |
|
60 |
self.fp._add_range(30, 39, 15) |
|
61 |
||
62 |
def test_valid_accesses(self): |
|
63 |
"""Test so that valid accesses work to the file."""
|
|
64 |
self.fp.seek(0, 0) |
|
65 |
self.assertEquals(self.fp.read(3), 'abc') |
|
66 |
self.assertEquals(self.fp.read(3), 'def') |
|
67 |
self.assertEquals(self.fp.tell(), 6) |
|
68 |
self.fp.seek(20, 0) |
|
69 |
self.assertEquals(self.fp.read(3), 'klm') |
|
70 |
self.assertEquals(self.fp.read(2), 'no') |
|
71 |
self.assertEquals(self.fp.tell(), 25) |
|
72 |
# should wrap over to 30-39 entity
|
|
73 |
self.assertEquals(self.fp.read(3), 'pqr') |
|
74 |
self.fp.seek(3) |
|
75 |
self.assertEquals(self.fp.read(3), 'def') |
|
76 |
self.assertEquals(self.fp.tell(), 6) |
|
77 |
||
78 |
def test_invalid_accesses(self): |
|
79 |
"""Test so that invalid accesses trigger errors."""
|
|
80 |
self.fp.seek(9) |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
81 |
self.assertRaises(errors.InvalidRange, self.fp.read, 2) |
1786.1.8
by John Arbash Meinel
[merge] Johan Rydberg test updates |
82 |
self.fp.seek(39) |
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
83 |
self.assertRaises(errors.InvalidRange, self.fp.read, 2) |
1786.1.8
by John Arbash Meinel
[merge] Johan Rydberg test updates |
84 |
self.fp.seek(19) |
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
85 |
self.assertRaises(errors.InvalidRange, self.fp.read, 2) |
1786.1.12
by John Arbash Meinel
Add tests for ResponseRange and streamline class |
86 |
|
87 |
def test__finish_ranges(self): |
|
88 |
"""Test that after RangeFile._finish_ranges the list is sorted."""
|
|
89 |
self.fp._add_range(1, 2, 3) |
|
90 |
self.fp._add_range(8, 9, 10) |
|
91 |
self.fp._add_range(3, 4, 5) |
|
92 |
||
93 |
# TODO: jam 20060706 If we switch to inserting
|
|
94 |
# in sorted order, remove this test
|
|
95 |
self.assertNotEqual(self.fp._ranges, sorted(self.fp._ranges)) |
|
96 |
||
97 |
self.fp._finish_ranges() |
|
98 |
self.assertEqual(self.fp._ranges, sorted(self.fp._ranges)) |
|
99 |
||
100 |
def test_seek_and_tell(self): |
|
101 |
# Check for seeking before start
|
|
102 |
self.fp.seek(-2, 0) |
|
103 |
self.assertEqual(0, self.fp.tell()) |
|
104 |
||
105 |
self.fp.seek(5, 0) |
|
106 |
self.assertEqual(5, self.fp.tell()) |
|
107 |
||
108 |
self.fp.seek(-2, 1) |
|
109 |
self.assertEqual(3, self.fp.tell()) |
|
110 |
||
111 |
# TODO: jam 20060706 following tests will fail if this
|
|
112 |
# is not true, and would be difficult to debug
|
|
113 |
# but it is a layering violation
|
|
114 |
self.assertEqual(39, self.fp._len) |
|
115 |
||
116 |
self.fp.seek(0, 2) |
|
117 |
self.assertEqual(39, self.fp.tell()) |
|
118 |
||
119 |
self.fp.seek(-10, 2) |
|
120 |
self.assertEqual(29, self.fp.tell()) |
|
121 |
||
122 |
self.assertRaises(ValueError, self.fp.seek, 0, 4) |
|
123 |
self.assertRaises(ValueError, self.fp.seek, 0, -1) |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
124 |
|
125 |
||
1786.1.16
by John Arbash Meinel
Refactor tests |
126 |
class TestRegexes(TestCase): |
127 |
||
128 |
def assertRegexMatches(self, groups, text): |
|
129 |
"""Check that the regex matches and returns the right values"""
|
|
130 |
m = self.regex.match(text) |
|
131 |
self.assertNotEqual(None, m, "text %s did not match regex" % (text,)) |
|
132 |
||
133 |
self.assertEqual(groups, m.groups()) |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
134 |
|
135 |
def test_range_re(self): |
|
136 |
"""Test that we match valid ranges."""
|
|
1786.1.24
by John Arbash Meinel
Move the functions/regexes to be static members |
137 |
self.regex = response.HttpRangeResponse._CONTENT_RANGE_RE |
1786.1.16
by John Arbash Meinel
Refactor tests |
138 |
self.assertRegexMatches(('bytes', '1', '10', '11'), |
139 |
'bytes 1-10/11') |
|
140 |
self.assertRegexMatches(('bytes', '1', '10', '11'), |
|
141 |
'\tbytes 1-10/11 ') |
|
142 |
self.assertRegexMatches(('bytes', '2123', '4242', '1231'), |
|
143 |
'\tbytes 2123-4242/1231 ') |
|
144 |
self.assertRegexMatches(('chars', '1', '2', '3'), |
|
145 |
' chars 1-2/3') |
|
146 |
||
147 |
def test_content_type_re(self): |
|
1786.1.24
by John Arbash Meinel
Move the functions/regexes to be static members |
148 |
self.regex = response.HttpMultipartRangeResponse._CONTENT_TYPE_RE |
1979.1.2
by John Arbash Meinel
Use the regex, rather than stripping off the boundary later. |
149 |
self.assertRegexMatches(('', 'xxyyzz'), |
1786.1.16
by John Arbash Meinel
Refactor tests |
150 |
'multipart/byteranges; boundary = xxyyzz') |
1979.1.2
by John Arbash Meinel
Use the regex, rather than stripping off the boundary later. |
151 |
self.assertRegexMatches(('', 'xxyyzz'), |
1786.1.16
by John Arbash Meinel
Refactor tests |
152 |
'multipart/byteranges;boundary=xxyyzz') |
1979.1.2
by John Arbash Meinel
Use the regex, rather than stripping off the boundary later. |
153 |
self.assertRegexMatches(('', 'xx yy zz'), |
1786.1.16
by John Arbash Meinel
Refactor tests |
154 |
' multipart/byteranges ; boundary= xx yy zz ') |
1979.1.2
by John Arbash Meinel
Use the regex, rather than stripping off the boundary later. |
155 |
self.assertRegexMatches(('"', 'xx yy zz'), |
156 |
' multipart/byteranges ; boundary= "xx yy zz" ') |
|
157 |
self.assertEqual(None, |
|
158 |
self.regex.match( |
|
159 |
' multipart/byteranges ; boundary= "xx yy zz ')) |
|
160 |
self.assertEqual(None, |
|
161 |
self.regex.match( |
|
162 |
' multipart/byteranges ; boundary= xx yy zz" ')) |
|
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
163 |
self.assertEqual(None, |
1786.1.16
by John Arbash Meinel
Refactor tests |
164 |
self.regex.match('multipart byteranges;boundary=xx')) |
165 |
||
166 |
||
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
167 |
simple_data = """ |
168 |
--xxyyzz\r |
|
169 |
foo\r |
|
170 |
Content-range: bytes 1-10/20\r |
|
171 |
\r
|
|
172 |
1234567890
|
|
173 |
--xxyyzz\r |
|
174 |
Content-Range: bytes 21-30/20\r |
|
175 |
bar\r |
|
176 |
\r
|
|
177 |
abcdefghij
|
|
178 |
--xxyyzz\r |
|
179 |
content-range: bytes 41-50/20\r |
|
180 |
\r
|
|
181 |
zyxwvutsrq
|
|
182 |
--xxyyzz\r |
|
1786.1.18
by John Arbash Meinel
Add tests for HttpMultiRangeResponse |
183 |
content-range: bytes 51-60/20\r |
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
184 |
\r
|
185 |
xxyyzz fbd
|
|
186 |
"""
|
|
187 |
||
188 |
||
189 |
class TestHelpers(TestCase): |
|
190 |
"""Test the helper functions"""
|
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
191 |
|
192 |
def test__parse_range(self): |
|
193 |
"""Test that _parse_range acts reasonably."""
|
|
194 |
content = StringIO('') |
|
1786.1.24
by John Arbash Meinel
Move the functions/regexes to be static members |
195 |
parse_range = response.HttpRangeResponse._parse_range |
1786.1.15
by John Arbash Meinel
Factor out _parse_range into a separate function for easier testing/better hierarchy. |
196 |
self.assertEqual((1,2), parse_range('bytes 1-2/3')) |
197 |
self.assertEqual((10,20), parse_range('bytes 10-20/2')) |
|
198 |
||
199 |
self.assertRaises(errors.InvalidHttpRange, parse_range, 'char 1-3/2') |
|
200 |
self.assertRaises(errors.InvalidHttpRange, parse_range, 'bytes a-3/2') |
|
201 |
||
202 |
try: |
|
203 |
parse_range('bytes x-10/3', path='http://foo/bar') |
|
204 |
except errors.InvalidHttpRange, e: |
|
205 |
self.assertContainsRe(str(e), 'http://foo/bar') |
|
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
206 |
self.assertContainsRe(str(e), 'bytes x-10/3') |
1786.1.15
by John Arbash Meinel
Factor out _parse_range into a separate function for easier testing/better hierarchy. |
207 |
else: |
208 |
self.fail('Did not raise InvalidHttpRange') |
|
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
209 |
|
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
210 |
def test__parse_boundary_simple(self): |
211 |
"""Test that _parse_boundary handles Content-type properly"""
|
|
1786.1.24
by John Arbash Meinel
Move the functions/regexes to be static members |
212 |
parse_boundary = response.HttpMultipartRangeResponse._parse_boundary |
213 |
m = parse_boundary(' multipart/byteranges; boundary=xxyyzz') |
|
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
214 |
self.assertNotEqual(None, m) |
215 |
# Check that the returned regex is capable of splitting simple_data
|
|
216 |
matches = list(m.finditer(simple_data)) |
|
217 |
self.assertEqual(4, len(matches)) |
|
218 |
||
219 |
# match.group() should be the content-range entry
|
|
220 |
# and match.end() should be the start of the content
|
|
221 |
self.assertEqual(' bytes 1-10/20', matches[0].group(1)) |
|
222 |
self.assertEqual(simple_data.find('1234567890'), matches[0].end()) |
|
223 |
self.assertEqual(' bytes 21-30/20', matches[1].group(1)) |
|
224 |
self.assertEqual(simple_data.find('abcdefghij'), matches[1].end()) |
|
225 |
self.assertEqual(' bytes 41-50/20', matches[2].group(1)) |
|
226 |
self.assertEqual(simple_data.find('zyxwvutsrq'), matches[2].end()) |
|
1786.1.18
by John Arbash Meinel
Add tests for HttpMultiRangeResponse |
227 |
self.assertEqual(' bytes 51-60/20', matches[3].group(1)) |
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
228 |
self.assertEqual(simple_data.find('xxyyzz fbd'), matches[3].end()) |
229 |
||
230 |
def test__parse_boundary_invalid(self): |
|
1786.1.24
by John Arbash Meinel
Move the functions/regexes to be static members |
231 |
parse_boundary = response.HttpMultipartRangeResponse._parse_boundary |
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
232 |
try: |
1786.1.24
by John Arbash Meinel
Move the functions/regexes to be static members |
233 |
parse_boundary(' multipart/bytes;boundary=xxyyzz', |
234 |
path='http://foo/bar') |
|
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
235 |
except errors.InvalidHttpContentType, e: |
236 |
self.assertContainsRe(str(e), 'http://foo/bar') |
|
237 |
self.assertContainsRe(str(e), 'multipart/bytes;boundary=xxyyzz') |
|
238 |
else: |
|
239 |
self.fail('Did not raise InvalidHttpContentType') |
|
240 |
||
241 |
||
242 |
class TestHttpRangeResponse(TestCase): |
|
243 |
||
1786.1.13
by John Arbash Meinel
Found a few bugs in error handling code, updated tests |
244 |
def test_smoketest(self): |
245 |
"""A basic test that HttpRangeResponse is reasonable."""
|
|
246 |
content = StringIO('0123456789') |
|
247 |
f = response.HttpRangeResponse('http://foo', 'bytes 1-10/9', content) |
|
248 |
self.assertEqual([response.ResponseRange(1,10,0)], f._ranges) |
|
249 |
||
250 |
f.seek(0) |
|
251 |
self.assertRaises(errors.InvalidRange, f.read, 2) |
|
252 |
f.seek(1) |
|
253 |
self.assertEqual('012345', f.read(6)) |
|
1786.1.14
by John Arbash Meinel
Testing basic functionality of HttpMultipartRangeResponse |
254 |
|
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
255 |
def test_invalid(self): |
256 |
try: |
|
257 |
f = response.HttpRangeResponse('http://foo', 'bytes x-10/9', |
|
258 |
StringIO('0123456789')) |
|
259 |
except errors.InvalidHttpRange, e: |
|
260 |
self.assertContainsRe(str(e), 'http://foo') |
|
261 |
self.assertContainsRe(str(e), 'bytes x-10/9') |
|
262 |
else: |
|
263 |
self.fail('Failed to raise InvalidHttpRange') |
|
264 |
||
1786.1.14
by John Arbash Meinel
Testing basic functionality of HttpMultipartRangeResponse |
265 |
|
266 |
class TestHttpMultipartRangeResponse(TestCase): |
|
1786.1.17
by John Arbash Meinel
Adding tests for _parse_boundary. |
267 |
"""Test the handling of multipart range responses"""
|
1786.1.18
by John Arbash Meinel
Add tests for HttpMultiRangeResponse |
268 |
|
269 |
def test_simple(self): |
|
270 |
content = StringIO(simple_data) |
|
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
271 |
multi = response.HttpMultipartRangeResponse('http://foo', |
1786.1.18
by John Arbash Meinel
Add tests for HttpMultiRangeResponse |
272 |
'multipart/byteranges; boundary = xxyyzz', content) |
273 |
||
274 |
self.assertEqual(4, len(multi._ranges)) |
|
275 |
||
276 |
multi.seek(1) |
|
277 |
self.assertEqual('1234567890', multi.read(10)) |
|
278 |
multi.seek(21) |
|
279 |
self.assertEqual('abcdefghij', multi.read(10)) |
|
280 |
multi.seek(41) |
|
281 |
self.assertEqual('zyxwvutsrq', multi.read(10)) |
|
282 |
multi.seek(51) |
|
283 |
self.assertEqual('xxyyzz fbd', multi.read(10)) |
|
284 |
# TODO: jam 20060706 Currently RangeFile does not support
|
|
285 |
# reading across ranges. Consider adding it.
|
|
286 |
multi.seek(41) |
|
287 |
# self.assertEqual('zyxwvutsrqxxyyzz fbd', multi.read(20))
|
|
288 |
self.assertRaises(errors.InvalidRange, multi.read, 20) |
|
289 |
||
290 |
multi.seek(21) |
|
291 |
self.assertRaises(errors.InvalidRange, multi.read, 11) |
|
292 |
multi.seek(31) |
|
293 |
self.assertRaises(errors.InvalidRange, multi.read, 10) |
|
1786.1.20
by John Arbash Meinel
Added an exception check for HttpMulti |
294 |
|
295 |
def test_invalid(self): |
|
296 |
content = StringIO('') |
|
297 |
try: |
|
298 |
response.HttpMultipartRangeResponse('http://foo', |
|
299 |
'multipart/byte;boundary=invalid', content) |
|
300 |
except errors.InvalidHttpContentType, e: |
|
301 |
self.assertContainsRe(str(e), 'http://foo') |
|
302 |
self.assertContainsRe(str(e), 'multipart/byte;') |
|
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
303 |
|
304 |
||
305 |
# Taken from real request responses
|
|
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
306 |
_full_text_response = (200, """HTTP/1.1 200 OK\r |
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
307 |
Date: Tue, 11 Jul 2006 04:32:56 GMT\r |
308 |
Server: Apache/2.0.54 (Fedora)\r |
|
309 |
Last-Modified: Sun, 23 Apr 2006 19:35:20 GMT\r |
|
310 |
ETag: "56691-23-38e9ae00"\r |
|
311 |
Accept-Ranges: bytes\r |
|
312 |
Content-Length: 35\r |
|
313 |
Connection: close\r |
|
314 |
Content-Type: text/plain; charset=UTF-8\r |
|
315 |
\r
|
|
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
316 |
""", """Bazaar-NG meta directory, format 1 |
317 |
""") |
|
318 |
||
319 |
||
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
320 |
_missing_response = (404, """HTTP/1.1 404 Not Found\r |
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
321 |
Date: Tue, 11 Jul 2006 04:32:56 GMT\r |
322 |
Server: Apache/2.0.54 (Fedora)\r |
|
323 |
Content-Length: 336\r |
|
324 |
Connection: close\r |
|
325 |
Content-Type: text/html; charset=iso-8859-1\r |
|
326 |
\r
|
|
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
327 |
""", """<!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML 2.0//EN"> |
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
328 |
<html><head>
|
329 |
<title>404 Not Found</title>
|
|
330 |
</head><body>
|
|
331 |
<h1>Not Found</h1>
|
|
332 |
<p>The requested URL /branches/bzr/jam-integration/.bzr/repository/format was not found on this server.</p>
|
|
333 |
<hr>
|
|
334 |
<address>Apache/2.0.54 (Fedora) Server at bzr.arbash-meinel.com Port 80</address>
|
|
335 |
</body></html>
|
|
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
336 |
""") |
337 |
||
338 |
||
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
339 |
_single_range_response = (206, """HTTP/1.1 206 Partial Content\r |
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
340 |
Date: Tue, 11 Jul 2006 04:45:22 GMT\r |
341 |
Server: Apache/2.0.54 (Fedora)\r |
|
342 |
Last-Modified: Thu, 06 Jul 2006 20:22:05 GMT\r |
|
343 |
ETag: "238a3c-16ec2-805c5540"\r |
|
344 |
Accept-Ranges: bytes\r |
|
345 |
Content-Length: 100\r |
|
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
346 |
Content-Range: bytes 100-199/93890\r |
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
347 |
Connection: close\r |
348 |
Content-Type: text/plain; charset=UTF-8\r |
|
349 |
\r
|
|
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
350 |
""", """mbp@sourcefrog.net-20050309040815-13242001617e4a06 |
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
351 |
mbp@sourcefrog.net-20050309040929-eee0eb3e6d1e762""") |
352 |
||
353 |
||
354 |
_multipart_range_response = (206, """HTTP/1.1 206 Partial Content\r |
|
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
355 |
Date: Tue, 11 Jul 2006 04:49:48 GMT\r |
356 |
Server: Apache/2.0.54 (Fedora)\r |
|
357 |
Last-Modified: Thu, 06 Jul 2006 20:22:05 GMT\r |
|
358 |
ETag: "238a3c-16ec2-805c5540"\r |
|
359 |
Accept-Ranges: bytes\r |
|
360 |
Content-Length: 1534\r |
|
361 |
Connection: close\r |
|
362 |
Content-Type: multipart/byteranges; boundary=418470f848b63279b\r |
|
363 |
\r
|
|
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
364 |
\r""", """--418470f848b63279b\r |
1786.1.21
by John Arbash Meinel
(broken) Work on factoring out handle_response so we can test with fake headers. |
365 |
Content-type: text/plain; charset=UTF-8\r |
366 |
Content-range: bytes 0-254/93890\r |
|
367 |
\r
|
|
368 |
mbp@sourcefrog.net-20050309040815-13242001617e4a06
|
|
369 |
mbp@sourcefrog.net-20050309040929-eee0eb3e6d1e7627
|
|
370 |
mbp@sourcefrog.net-20050309040957-6cad07f466bb0bb8
|
|
371 |
mbp@sourcefrog.net-20050309041501-c840e09071de3b67
|
|
372 |
mbp@sourcefrog.net-20050309044615-c24a3250be83220a
|
|
373 |
\r
|
|
374 |
--418470f848b63279b\r |
|
375 |
Content-type: text/plain; charset=UTF-8\r |
|
376 |
Content-range: bytes 1000-2049/93890\r |
|
377 |
\r
|
|
378 |
40-fd4ec249b6b139ab
|
|
379 |
mbp@sourcefrog.net-20050311063625-07858525021f270b
|
|
380 |
mbp@sourcefrog.net-20050311231934-aa3776aff5200bb9
|
|
381 |
mbp@sourcefrog.net-20050311231953-73aeb3a131c3699a
|
|
382 |
mbp@sourcefrog.net-20050311232353-f5e33da490872c6a
|
|
383 |
mbp@sourcefrog.net-20050312071639-0a8f59a34a024ff0
|
|
384 |
mbp@sourcefrog.net-20050312073432-b2c16a55e0d6e9fb
|
|
385 |
mbp@sourcefrog.net-20050312073831-a47c3335ece1920f
|
|
386 |
mbp@sourcefrog.net-20050312085412-13373aa129ccbad3
|
|
387 |
mbp@sourcefrog.net-20050313052251-2bf004cb96b39933
|
|
388 |
mbp@sourcefrog.net-20050313052856-3edd84094687cb11
|
|
389 |
mbp@sourcefrog.net-20050313053233-e30a4f28aef48f9d
|
|
390 |
mbp@sourcefrog.net-20050313053853-7c64085594ff3072
|
|
391 |
mbp@sourcefrog.net-20050313054757-a86c3f5871069e22
|
|
392 |
mbp@sourcefrog.net-20050313061422-418f1f73b94879b9
|
|
393 |
mbp@sourcefrog.net-20050313120651-497bd231b19df600
|
|
394 |
mbp@sourcefrog.net-20050314024931-eae0170ef25a5d1a
|
|
395 |
mbp@sourcefrog.net-20050314025438-d52099f915fe65fc
|
|
396 |
mbp@sourcefrog.net-20050314025539-637a636692c055cf
|
|
397 |
mbp@sourcefrog.net-20050314025737-55eb441f430ab4ba
|
|
398 |
mbp@sourcefrog.net-20050314025901-d74aa93bb7ee8f62
|
|
399 |
mbp@source\r |
|
1979.1.1
by John Arbash Meinel
Fix bug #57723, parse boundary="" correctly, since Squid uses it |
400 |
--418470f848b63279b--\r |
401 |
""") |
|
402 |
||
403 |
_multipart_squid_range_response = (206, """HTTP/1.0 206 Partial Content\r |
|
404 |
Date: Thu, 31 Aug 2006 21:16:22 GMT\r |
|
405 |
Server: Apache/2.2.2 (Unix) DAV/2\r |
|
406 |
Last-Modified: Thu, 31 Aug 2006 17:57:06 GMT\r |
|
407 |
Accept-Ranges: bytes\r |
|
408 |
Content-Type: multipart/byteranges; boundary="squid/2.5.STABLE12:C99323425AD4FE26F726261FA6C24196"\r |
|
409 |
Content-Length: 598\r |
|
410 |
X-Cache: MISS from localhost.localdomain\r |
|
411 |
X-Cache-Lookup: HIT from localhost.localdomain:3128\r |
|
412 |
Proxy-Connection: keep-alive\r |
|
413 |
\r
|
|
414 |
""", |
|
415 |
"""\r
|
|
416 |
--squid/2.5.STABLE12:C99323425AD4FE26F726261FA6C24196\r
|
|
417 |
Content-Type: text/plain\r
|
|
418 |
Content-Range: bytes 0-99/18672\r
|
|
419 |
\r
|
|
420 |
# bzr knit index 8
|
|
421 |
||
422 |
scott@netsplit.com-20050708230047-47c7868f276b939f fulltext 0 863 :
|
|
423 |
scott@netsp\r
|
|
424 |
--squid/2.5.STABLE12:C99323425AD4FE26F726261FA6C24196\r
|
|
425 |
Content-Type: text/plain\r
|
|
426 |
Content-Range: bytes 300-499/18672\r
|
|
427 |
\r
|
|
428 |
com-20050708231537-2b124b835395399a :
|
|
429 |
scott@netsplit.com-20050820234126-551311dbb7435b51 line-delta 1803 479 .scott@netsplit.com-20050820232911-dc4322a084eadf7e :
|
|
430 |
scott@netsplit.com-20050821213706-c86\r
|
|
431 |
--squid/2.5.STABLE12:C99323425AD4FE26F726261FA6C24196--\r
|
|
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
432 |
""") |
433 |
||
434 |
||
1786.1.41
by John Arbash Meinel
parse redirect headers properly. |
435 |
_redirect_response = (206, """HTTP/1.1 301 Moved Permanently\r |
436 |
Date: Tue, 18 Jul 2006 20:29:22 GMT\r |
|
437 |
Server: Apache/2.0.54 (Ubuntu) PHP/4.4.0-3ubuntu1 mod_ssl/2.0.54 OpenSSL/0.9.7g\r |
|
438 |
Location: http://bazaar-vcs.org/bzr/bzr.dev/.bzr/repository/inventory.knit\r |
|
439 |
Content-Length: 272\r |
|
440 |
Keep-Alive: timeout=15, max=100\r |
|
441 |
Connection: Keep-Alive\r |
|
442 |
Content-Type: text/html; charset=iso-8859-1\r |
|
443 |
\r
|
|
444 |
HTTP/1.1 206 Partial Content\r |
|
445 |
Date: Tue, 18 Jul 2006 20:29:23 GMT\r |
|
446 |
Server: Apache/2.0.54 (Ubuntu) PHP/4.4.0-3ubuntu1 mod_ssl/2.0.54 OpenSSL/0.9.7g\r |
|
447 |
Last-Modified: Tue, 18 Jul 2006 20:24:59 GMT\r |
|
448 |
ETag: "be8213-83958c-f0d3dcc0"\r |
|
449 |
Accept-Ranges: bytes\r |
|
450 |
Content-Length: 425\r |
|
451 |
Content-Range: bytes 8623075-8623499/8623500\r |
|
452 |
Keep-Alive: timeout=15, max=100\r |
|
453 |
Connection: Keep-Alive\r |
|
454 |
Content-Type: text/plain; charset=UTF-8\r |
|
455 |
\r
|
|
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
456 |
""", """this data intentionally removed, |
457 |
this is not meant to be tested by
|
|
458 |
handle_response, just _extract_headers
|
|
1786.1.41
by John Arbash Meinel
parse redirect headers properly. |
459 |
""") |
460 |
||
461 |
||
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
462 |
# This is made up
|
463 |
_invalid_response = (444, """HTTP/1.1 444 Bad Response\r |
|
464 |
Date: Tue, 11 Jul 2006 04:32:56 GMT\r |
|
465 |
Connection: close\r |
|
466 |
Content-Type: text/html; charset=iso-8859-1\r |
|
467 |
\r
|
|
468 |
""", """<!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML 2.0//EN"> |
|
469 |
<html><head>
|
|
470 |
<title>404 Not Found</title>
|
|
471 |
</head><body>
|
|
472 |
<h1>Not Found</h1>
|
|
473 |
<p>I don't know what I'm doing</p>
|
|
474 |
<hr>
|
|
475 |
</body></html>
|
|
476 |
""") |
|
477 |
||
478 |
||
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
479 |
# This should be in test_http.py, but the headers we
|
480 |
# want to parse are here
|
|
481 |
class TestExtractHeader(TestCase): |
|
482 |
||
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
483 |
def use_response(self, response): |
484 |
self.headers = http._extract_headers(response[1], 'http://foo') |
|
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
485 |
|
486 |
def check_header(self, header, value): |
|
487 |
self.assertEqual(value, self.headers[header]) |
|
488 |
||
489 |
def test_full_text(self): |
|
490 |
self.use_response(_full_text_response) |
|
491 |
||
492 |
self.check_header('Date', 'Tue, 11 Jul 2006 04:32:56 GMT') |
|
493 |
self.check_header('date', 'Tue, 11 Jul 2006 04:32:56 GMT') |
|
494 |
self.check_header('Content-Length', '35') |
|
495 |
self.check_header('Content-Type', 'text/plain; charset=UTF-8') |
|
496 |
self.check_header('content-type', 'text/plain; charset=UTF-8') |
|
497 |
||
498 |
def test_missing_response(self): |
|
499 |
self.use_response(_missing_response) |
|
500 |
||
501 |
self.check_header('Content-Length', '336') |
|
502 |
self.check_header('Content-Type', 'text/html; charset=iso-8859-1') |
|
503 |
||
504 |
def test_single_range(self): |
|
505 |
self.use_response(_single_range_response) |
|
506 |
||
507 |
self.check_header('Content-Length', '100') |
|
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
508 |
self.check_header('Content-Range', 'bytes 100-199/93890') |
1786.1.25
by John Arbash Meinel
Test that we can extract headers properly. |
509 |
self.check_header('Content-Type', 'text/plain; charset=UTF-8') |
510 |
||
511 |
def test_multi_range(self): |
|
512 |
self.use_response(_multipart_range_response) |
|
513 |
||
514 |
self.check_header('Content-Length', '1534') |
|
515 |
self.check_header('Content-Type', |
|
516 |
'multipart/byteranges; boundary=418470f848b63279b') |
|
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
517 |
|
1979.1.1
by John Arbash Meinel
Fix bug #57723, parse boundary="" correctly, since Squid uses it |
518 |
def test_multi_squid_range(self): |
519 |
self.use_response(_multipart_squid_range_response) |
|
520 |
||
521 |
self.check_header('Content-Length', '598') |
|
522 |
self.check_header('Content-Type', |
|
523 |
'multipart/byteranges; '\
|
|
524 |
'boundary="squid/2.5.STABLE12:C99323425AD4FE26F726261FA6C24196"') |
|
525 |
||
1786.1.41
by John Arbash Meinel
parse redirect headers properly. |
526 |
def test_redirect(self): |
527 |
"""We default to returning the last group of headers in the file."""
|
|
528 |
self.use_response(_redirect_response) |
|
529 |
self.check_header('Content-Range', 'bytes 8623075-8623499/8623500') |
|
530 |
self.check_header('Content-Type', 'text/plain; charset=UTF-8') |
|
531 |
||
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
532 |
def test_empty(self): |
533 |
self.assertRaises(errors.InvalidHttpResponse, |
|
534 |
http._extract_headers, '', 'bad url') |
|
535 |
||
536 |
def test_no_opening_http(self): |
|
537 |
# Remove the HTTP line from the header
|
|
538 |
first, txt = _full_text_response[1].split('\r\n', 1) |
|
539 |
self.assertRaises(errors.InvalidHttpResponse, |
|
540 |
http._extract_headers, txt, 'missing HTTTP') |
|
541 |
||
542 |
def test_trailing_whitespace(self): |
|
543 |
# Test that we ignore bogus whitespace on the end
|
|
544 |
code, txt, body = _full_text_response |
|
545 |
txt += '\r\n\n\n\n\n' |
|
546 |
self.use_response((code, txt, body)) |
|
547 |
||
548 |
self.check_header('Date', 'Tue, 11 Jul 2006 04:32:56 GMT') |
|
549 |
self.check_header('Content-Length', '35') |
|
550 |
self.check_header('Content-Type', 'text/plain; charset=UTF-8') |
|
551 |
||
552 |
def test_trailing_non_http(self): |
|
553 |
# Test that we ignore bogus stuff on the end
|
|
554 |
code, txt, body = _full_text_response |
|
555 |
txt = txt + 'Foo: Bar\r\nBaz: Bling\r\n\r\n' |
|
556 |
self.use_response((code, txt, body)) |
|
557 |
||
558 |
self.check_header('Date', 'Tue, 11 Jul 2006 04:32:56 GMT') |
|
559 |
self.check_header('Content-Length', '35') |
|
560 |
self.check_header('Content-Type', 'text/plain; charset=UTF-8') |
|
561 |
self.assertRaises(KeyError, self.headers.__getitem__, 'Foo') |
|
562 |
||
563 |
def test_extra_whitespace(self): |
|
564 |
# Test that we read an HTTP response, even with extra whitespace
|
|
565 |
code, txt, body = _redirect_response |
|
566 |
# Find the second HTTP location
|
|
567 |
loc = txt.find('HTTP', 5) |
|
568 |
txt = txt[:loc] + '\r\n\n' + txt[loc:] |
|
569 |
self.use_response((code, txt, body)) |
|
570 |
self.check_header('Content-Range', 'bytes 8623075-8623499/8623500') |
|
571 |
self.check_header('Content-Type', 'text/plain; charset=UTF-8') |
|
572 |
||
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
573 |
|
574 |
class TestHandleResponse(TestCase): |
|
575 |
||
576 |
def get_response(self, a_response): |
|
577 |
"""Process a supplied response, and return the result."""
|
|
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
578 |
headers = http._extract_headers(a_response[1], 'http://foo') |
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
579 |
return response.handle_response('http://foo', a_response[0], headers, |
580 |
StringIO(a_response[2])) |
|
581 |
||
582 |
def test_full_text(self): |
|
583 |
out = self.get_response(_full_text_response) |
|
584 |
# It is a StringIO from the original data
|
|
585 |
self.assertEqual(_full_text_response[2], out.read()) |
|
586 |
||
587 |
def test_missing_response(self): |
|
588 |
self.assertRaises(errors.NoSuchFile, |
|
589 |
self.get_response, _missing_response) |
|
590 |
||
591 |
def test_single_range(self): |
|
592 |
out = self.get_response(_single_range_response) |
|
593 |
self.assertIsInstance(out, response.HttpRangeResponse) |
|
594 |
||
595 |
self.assertRaises(errors.InvalidRange, out.read, 20) |
|
596 |
||
597 |
out.seek(100) |
|
598 |
self.assertEqual(_single_range_response[2], out.read(100)) |
|
599 |
||
600 |
def test_multi_range(self): |
|
601 |
out = self.get_response(_multipart_range_response) |
|
602 |
self.assertIsInstance(out, response.HttpMultipartRangeResponse) |
|
603 |
||
604 |
# Just make sure we can read the right contents
|
|
605 |
out.seek(0) |
|
606 |
out.read(255) |
|
607 |
||
608 |
out.seek(1000) |
|
609 |
out.read(1050) |
|
610 |
||
1979.1.1
by John Arbash Meinel
Fix bug #57723, parse boundary="" correctly, since Squid uses it |
611 |
def test_multi_squid_range(self): |
612 |
out = self.get_response(_multipart_squid_range_response) |
|
613 |
self.assertIsInstance(out, response.HttpMultipartRangeResponse) |
|
614 |
||
615 |
# Just make sure we can read the right contents
|
|
616 |
out.seek(0) |
|
617 |
out.read(100) |
|
618 |
||
619 |
out.seek(300) |
|
620 |
out.read(200) |
|
621 |
||
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
622 |
def test_invalid_response(self): |
623 |
self.assertRaises(errors.InvalidHttpResponse, |
|
624 |
self.get_response, _invalid_response) |
|
625 |
||
626 |
def test_full_text_no_content_type(self): |
|
627 |
# We should not require Content-Type for a full response
|
|
628 |
a_response = _full_text_response |
|
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
629 |
headers = http._extract_headers(a_response[1], 'http://foo') |
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
630 |
del headers['Content-Type'] |
631 |
out = response.handle_response('http://foo', a_response[0], headers, |
|
632 |
StringIO(a_response[2])) |
|
633 |
self.assertEqual(_full_text_response[2], out.read()) |
|
634 |
||
635 |
def test_missing_no_content_type(self): |
|
636 |
# Without Content-Type we should still raise NoSuchFile on a 404
|
|
637 |
a_response = _missing_response |
|
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
638 |
headers = http._extract_headers(a_response[1], 'http://missing') |
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
639 |
del headers['Content-Type'] |
640 |
self.assertRaises(errors.NoSuchFile, |
|
641 |
response.handle_response, 'http://missing', a_response[0], headers, |
|
642 |
StringIO(a_response[2])) |
|
643 |
||
644 |
def test_missing_content_type(self): |
|
645 |
a_response = _single_range_response |
|
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
646 |
headers = http._extract_headers(a_response[1], 'http://nocontent') |
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
647 |
del headers['Content-Type'] |
648 |
self.assertRaises(errors.InvalidHttpContentType, |
|
649 |
response.handle_response, 'http://nocontent', a_response[0], |
|
650 |
headers, StringIO(a_response[2])) |
|
651 |
||
652 |
def test_missing_content_range(self): |
|
653 |
a_response = _single_range_response |
|
1786.1.42
by John Arbash Meinel
Update _extract_headers, make it less generic, and non recursive. |
654 |
headers = http._extract_headers(a_response[1], 'http://nocontent') |
1786.1.26
by John Arbash Meinel
Update and test handle_response. |
655 |
del headers['Content-Range'] |
656 |
self.assertRaises(errors.InvalidHttpResponse, |
|
657 |
response.handle_response, 'http://nocontent', a_response[0], |
|
658 |
headers, StringIO(a_response[2])) |