def test_download_started_canceled(self):
batch = Mock()
batch.is_canceled = True
batch.listener = Mock()
report = Mock()
report.data = Mock()
# test
listener = NectarListener(batch)
listener.download_started(report)
# validation
self.assertFalse(batch.listener.download_started.called)
def test_download_started(self, event):
batch = Mock()
batch.listener = Mock()
report = Mock()
report.data = Mock()
# test
listener = NectarListener(batch)
listener.download_started(report)
# validation
event.assert_called_once_with(report.data)
event.return_value.assert_called_once_with(batch.listener)
def test_download_started_canceled(self):
batch = Mock()
batch.is_canceled = True
batch.listener = Mock()
report = Mock()
report.data = Mock()
# test
listener = NectarListener(batch)
listener.download_started(report)
# validation
self.assertFalse(batch.listener.download_started.called)
def test_download_started(self, event):
batch = Mock()
batch.listener = Mock()
report = Mock()
report.data = Mock()
# test
listener = NectarListener(batch)
listener.download_started(report)
# validation
event.assert_called_once_with(report.data)
event.return_value.assert_called_once_with(batch.listener)
def test_download_started_no_listener(self):
canceled = Mock()
canceled.isSet.return_value = False
downloader = Mock()
listener = None
report = Mock()
report.data = {'A': 1}
# test
nectar_listener = NectarListener(canceled, downloader, listener)
nectar_listener.download_started(report)
# validation
canceled.isSet.assert_called_with()
def test_download_started_no_listener(self):
canceled = Mock()
canceled.isSet.return_value = False
downloader = Mock()
listener = None
report = Mock()
report.data = {'A': 1}
# test
nectar_listener = NectarListener(canceled, downloader, listener)
nectar_listener.download_started(report)
# validation
canceled.isSet.assert_called_with()
def test_download_started_no_listener(self):
batch = Mock()
batch.is_canceled = False
batch.listener = Mock()
batch.listener.__nonzero__ = Mock(return_value=False)
report = Mock()
report.data = Mock()
# test
listener = NectarListener(batch)
listener.download_started(report)
# validation
self.assertFalse(batch.listener.download_started.called)
def test_download_started_no_listener(self):
batch = Mock()
batch.is_canceled = False
batch.listener = Mock()
batch.listener.__nonzero__ = Mock(return_value=False)
report = Mock()
report.data = Mock()
# test
listener = NectarListener(batch)
listener.download_started(report)
# validation
self.assertFalse(batch.listener.download_started.called)
def test_notification_no_listener(self, *unused):
request = Request('', {}, '', '')
container = ContentContainer('')
event = Event()
nectar_listener = NectarListener(event, Mock())
nectar_listener._notify = Mock()
report = Mock()
report.data = request
# started
nectar_listener.download_started(report)
self.assertFalse(nectar_listener._notify.called)
# succeeded
nectar_listener.download_succeeded(report)
self.assertFalse(nectar_listener._notify.called)
# failed
nectar_listener.download_failed(report)
self.assertFalse(nectar_listener._notify.called)
def test_notification(self, *unused):
request = Request('', {}, '', '')
listener = MockListener()
listener.download_started = Mock(side_effect=ValueError)
container = ContentContainer('')
event = Event()
nectar_listener = NectarListener(event, Mock(), listener)
report = Mock()
report.data = request
# started
nectar_listener.download_started(report)
listener.download_started.assert_called_with(request)
# succeeded
nectar_listener.download_succeeded(report)
listener.download_succeeded.assert_called_with(request)
# failed
nectar_listener.download_failed(report)
listener.download_failed.assert_called_with(request)
