def test_httpurl_next_ok(self, mock_req):
content = ['fake content #1', 'fake content #2']
req_mock = mock.Mock(headers={'content-length': 30})
req_mock.raw.read.side_effect = content
mock_req.return_value = req_mock
httpurl = au.HttpUrl('fake_url')
for data in enumerate(httpurl):
self.assertEqual(content[data[0]], data[1])
def test_httpurl_next_slow_connection(self, mock_req):
url = "http://fake_url"
content = ['fake content #1', '', 'fake content #2']
req_mock = mock.Mock(headers={'content-length': 30})
req_mock.raw.read.side_effect = content
mock_req.return_value = req_mock
http_url = au.HttpUrl(url)
for data in content:
if data:
self.assertEqual(data, next(http_url))
expected_calls = [
mock.call(url),
mock.call(url, byte_range=len(content[0]))
]
self.assertEqual(expected_calls, mock_req.call_args_list)
def test_httpurl_init_ok(self, mock_req):
mock_req.return_value = mock.Mock(headers={'content-length': 123})
httpurl = au.HttpUrl('fake_url')
self.assertEqual(123, httpurl.length)
mock_req.assert_called_once_with('fake_url')