def test_add_ext_rpm_dup(self):
prev = self.rpminfo.copy()
prev['external_repo_id'] = mock.sentinel.repo_id
prev['external_repo_name'] = self.repo
self.get_rpm.return_value = prev
self.get_external_repo_id.return_value = mock.sentinel.repo_id
# call it (default is strict)
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo)
self.assertEqual(len(self.inserts), 0)
self.nextval.assert_not_called()
# call it without strict
ret = kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(ret, self.get_rpm.return_value)
self.assertEqual(len(self.inserts), 0)
self.nextval.assert_not_called()
# different hash
prev['payloadhash'] = 'different hash'
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(len(self.inserts), 0)
self.nextval.assert_not_called()
def test_add_ext_rpm_dup(self):
prev = self.rpminfo.copy()
prev['external_repo_id'] = mock.sentinel.repo_id
prev['external_repo_name'] = self.repo
self.get_rpm.return_value = prev
self.get_external_repo_id.return_value = mock.sentinel.repo_id
# call it (default is strict)
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo)
self.assertEqual(len(self.inserts), 0)
self.nextval.assert_not_called()
# call it without strict
ret = kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(ret, self.get_rpm.return_value)
self.assertEqual(len(self.inserts), 0)
self.nextval.assert_not_called()
# different hash
prev['payloadhash'] = 'different hash'
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(len(self.inserts), 0)
self.nextval.assert_not_called()
def test_add_ext_rpm(self):
self.get_rpm.return_value = None
self.get_external_repo_id.return_value = mock.sentinel.repo_id
self.nextval.return_value = mock.sentinel.rpm_id
# call it
kojihub.add_external_rpm(self.rpminfo, self.repo)
self.assertEqual(len(self.inserts), 1)
insert = self.inserts[0]
self.assertEqual(insert.data['external_repo_id'], mock.sentinel.repo_id)
self.assertEqual(insert.data['id'], mock.sentinel.rpm_id)
self.assertEqual(insert.table, 'rpminfo')
def test_add_ext_rpm(self):
self.get_rpm.return_value = None
self.get_external_repo_id.return_value = mock.sentinel.repo_id
self.nextval.return_value = mock.sentinel.rpm_id
# call it
kojihub.add_external_rpm(self.rpminfo, self.repo)
self.assertEqual(len(self.inserts), 1)
insert = self.inserts[0]
self.assertEqual(insert.data['external_repo_id'],
mock.sentinel.repo_id)
self.assertEqual(insert.data['id'], mock.sentinel.rpm_id)
self.assertEqual(insert.table, 'rpminfo')
def test_add_ext_rpm_bad_data(self):
rpminfo = self.rpminfo.copy()
del rpminfo['size']
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(rpminfo, self.repo)
self.get_rpm.assert_not_called()
self.nextval.assert_not_called()
self.assertEqual(len(self.inserts), 0)
rpminfo = self.rpminfo.copy()
rpminfo['size'] = ['invalid type']
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(rpminfo, self.repo)
self.get_rpm.assert_not_called()
self.nextval.assert_not_called()
self.assertEqual(len(self.inserts), 0)
def test_add_ext_rpm_bad_data(self):
rpminfo = self.rpminfo.copy()
del rpminfo['size']
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(rpminfo, self.repo)
self.get_rpm.assert_not_called()
self.nextval.assert_not_called()
self.assertEqual(len(self.inserts), 0)
rpminfo = self.rpminfo.copy()
rpminfo['size'] = ['invalid type']
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(rpminfo, self.repo)
self.get_rpm.assert_not_called()
self.nextval.assert_not_called()
self.assertEqual(len(self.inserts), 0)
def test_add_ext_rpm_dup_late(self):
prev = self.rpminfo.copy()
prev['external_repo_id'] = mock.sentinel.repo_id
prev['external_repo_name'] = self.repo
self.get_rpm.side_effect = [None, prev]
self.get_external_repo_id.return_value = mock.sentinel.repo_id
self.insert_execute.side_effect = FakeException('insert failed')
# call it (default is strict)
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()
# call it without strict
self.inserts[:] = []
self.nextval.reset_mock()
self.get_rpm.side_effect = [None, prev]
ret = kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(ret, prev)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()
# different hash
self.inserts[:] = []
self.nextval.reset_mock()
self.get_rpm.side_effect = [None, prev]
prev['payloadhash'] = 'different hash'
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()
# no dup after failed insert
self.inserts[:] = []
self.nextval.reset_mock()
self.get_rpm.side_effect = [None, None]
with self.assertRaises(FakeException):
kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()
def test_add_ext_rpm_dup_late(self):
prev = self.rpminfo.copy()
prev['external_repo_id'] = mock.sentinel.repo_id
prev['external_repo_name'] = self.repo
self.get_rpm.side_effect = [None, prev]
self.get_external_repo_id.return_value = mock.sentinel.repo_id
self.insert_execute.side_effect = FakeException('insert failed')
# call it (default is strict)
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()
# call it without strict
self.inserts[:] = []
self.nextval.reset_mock()
self.get_rpm.side_effect = [None, prev]
ret = kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(ret, prev)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()
# different hash
self.inserts[:] = []
self.nextval.reset_mock()
self.get_rpm.side_effect = [None, prev]
prev['payloadhash'] = 'different hash'
with self.assertRaises(koji.GenericError):
kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()
# no dup after failed insert
self.inserts[:] = []
self.nextval.reset_mock()
self.get_rpm.side_effect = [None, None]
with self.assertRaises(FakeException):
kojihub.add_external_rpm(self.rpminfo, self.repo, strict=False)
self.assertEqual(len(self.inserts), 1)
self.nextval.assert_called_once()