def test_download_successful(self, mock_copy): self.sync_call_config.get.return_value = False content_listener = listener.ContentListener(self.sync_conduit, self.progress_report, self.sync_call_config, self.metadata_files) content_listener.download_succeeded(self.report) self.progress_report['content'].success.assert_called_once_with( self.report.data)
def test_download_successful_with_validation(self, mock_copy, mock_verify_size, mock_verify_checksum, mock_open): self.sync_call_config.get.return_value = True content_listener = listener.ContentListener(self.sync_conduit, self.progress_report, self.sync_call_config, self.metadata_files) content_listener.download_succeeded(self.report) self.progress_report['content'].success.assert_called_once_with( self.report.data) mock_verify_size.assert_called_once() mock_verify_checksum.assert_called_once()
def test_download_successful_invalid_checksum_verification( self, mock_copy, mock_verify_size, mock_verify_checksum, mock_open): self.sync_call_config.get.return_value = True mock_verify_checksum.side_effect = verification.VerificationException( 'bad') content_listener = listener.ContentListener(self.sync_conduit, self.progress_report, self.sync_call_config, self.metadata_files) content_listener.download_succeeded(self.report) mock_verify_checksum.assert_called_once() self.assertFalse(self.progress_report['content'].success.called)