def test_download_successful(self, mock_nevra, mock_copy, mock_assoc):
self.config.get.return_value = False
content_listener = listener.PackageListener(self, self.metadata_files)
content_listener.download_succeeded(self.report)
self.progress_report['content'].success.assert_called_once_with(
self.report.data)
mock_nevra.assert_called_once_with(self.conduit.init_unit().unit_key,
self.conduit.init_unit().type_id,
self.conduit.repo_id)
def test_download_successful_invalid_checksum_verification(
self, mock_copy, mock_verify_size, mock_verify_checksum,
mock_open):
self.config.get.return_value = True
mock_verify_checksum.side_effect = verification.VerificationException(
'bad')
content_listener = listener.PackageListener(self, self.metadata_files)
self.assertRaises(verification.VerificationException,
content_listener.download_succeeded, self.report)
mock_verify_checksum.assert_called_once()
self.assertFalse(self.progress_report['content'].success.called)
def test_download_successful_invalid_checksum_type(
self, mock_copy, mock_verify_size, mock_validate_checksumtype,
mock_open):
self.config.get.return_value = True
self.report.data = models.RPM()
mock_validate_checksumtype.side_effect = util.InvalidChecksumType()
content_listener = listener.PackageListener(self, self.metadata_files)
with self.assertRaises(util.InvalidChecksumType):
content_listener.download_succeeded(self.report)
mock_validate_checksumtype.assert_called_once()
self.assertFalse(self.progress_report['content'].success.called)
def test_download_successful_with_validation(self, mock_copy, mock_nevra,
mock_verify_size,
mock_verify_checksum,
mock_open, mock_assoc):
self.config.get.return_value = True
content_listener = listener.PackageListener(self, 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()
mock_nevra.assert_called_once_with(self.conduit.init_unit().unit_key,
self.conduit.init_unit().type_id,
self.conduit.repo_id)
def test_download_successful_invalid_checksum_verification(
self, mock_copy, mock_verify_size, mock_validate_checksumtype,
mock_verify_checksum, mock_open):
self.config.get.return_value = True
self.report.data = models.RPM()
mock_verify_checksum.side_effect = verification.VerificationException(
'bad')
content_listener = listener.PackageListener(self, self.metadata_files)
with self.assertRaises(verification.VerificationException) as cm:
content_listener.download_succeeded(self.report)
self.assertEqual(cm.exception.args[0], 'bad')
mock_verify_checksum.assert_called_once()
self.assertFalse(self.progress_report['content'].success.called)
def setUp(self):
self.mock_sync = mock.MagicMock()
self.mock_metadata_files = mock.MagicMock()
self.listener = listener.PackageListener(self.mock_sync,
self.mock_metadata_files)
