def test_request_cancel(self, mock_from_request):
url = 'http://fakeurl/robots.txt'
req = DownloadRequest(url, mock.Mock())
req.canceled = True
self.downloader._fetch(req)
mock_from_request.return_value.download_canceled.assert_called_once_with()
def test_request_cancel(self, mock_from_request):
url = 'http://fakeurl/robots.txt'
req = DownloadRequest(url, mock.Mock())
req.canceled = True
self.downloader._fetch(req)
mock_from_request.return_value.download_canceled.assert_called_once_with(
)
def test_copy_canceled_single_request(self, mock_canceled, mock_open):
downloader = local.LocalFileDownloader(DownloaderConfig())
request = DownloadRequest('file://' + __file__, '/bar')
request.canceled = True
downloader._copy(request)
# make sure the cancel method was called on the report
mock_canceled.assert_called_once_with()
# make sure the no writing was attempted
self.assertEqual(mock_open.return_value.write.call_count, 0)
def test_copy_canceled_single_request(self, mock_canceled, mock_open):
downloader = local.LocalFileDownloader(DownloaderConfig())
request = DownloadRequest('file://' + __file__, '/bar')
request.canceled = True
downloader._copy(request)
# make sure the cancel method was called on the report
mock_canceled.assert_called_once_with()
# make sure the no writing was attempted
self.assertEqual(mock_open.return_value.write.call_count, 0)
