def test_download_succeeded(self):
request = Request("T1", {"A": 1}, "http://test", "/tmp/test")
request.data = Mock()
content_listener = Mock()
# test
listener = ContainerListener(content_listener)
listener.download_succeeded(request)
# validation
calls = content_listener.download_succeeded.mock_calls
self.assertEqual(len(calls), 1)
report = calls[0][1][0]
self.assertEqual(report.url, request.url)
self.assertEqual(report.destination, request.destination)
self.assertEqual(report.data, request.data)
def test_on_succeeded(self):
request = Request('T1', {'A': 1}, 'http://test', '/tmp/test')
request.data = Mock()
content_listener = Mock()
# test
listener = ContainerListener(content_listener)
listener.on_succeeded(request)
# validation
calls = content_listener.download_succeeded.mock_calls
self.assertEqual(len(calls), 1)
report = calls[0][1][0]
self.assertEqual(report.url, request.url)
self.assertEqual(report.destination, request.destination)
self.assertEqual(report.data, request.data)
def test_download_failed(self):
request = Request('T1', {'A': 1}, 'http://test', '/tmp/test')
request.data = Mock()
request.errors = [1, 2, 3]
content_listener = Mock()
# test
listener = ContainerListener(content_listener)
listener.download_failed(request)
# validation
calls = content_listener.download_failed.mock_calls
self.assertEqual(len(calls), 1)
report = calls[0][1][0]
self.assertEqual(report.url, request.url)
self.assertEqual(report.destination, request.destination)
self.assertEqual(report.data, request.data)
self.assertEqual(report.error_report['errors'], request.errors)
def test_on_failed(self):
request = Request("T1", {"A": 1}, "http://test", "/tmp/test")
request.data = Mock()
request.errors = [1, 2, 3]
content_listener = Mock()
# test
listener = ContainerListener(content_listener)
listener.on_failed(request)
# validation
calls = content_listener.download_failed.mock_calls
self.assertEqual(len(calls), 1)
report = calls[0][1][0]
self.assertEqual(report.url, request.url)
self.assertEqual(report.destination, request.destination)
self.assertEqual(report.data, request.data)
self.assertEqual(report.error_report["errors"], request.errors)
def test_construction(self):
content_listener = Mock()
listener = ContainerListener(content_listener)
self.assertEqual(listener.content_listener, content_listener)
