debian-koji/tests/test_hub/test_new_build.py

import mock
import unittest

import koji
from koji.util import dslice
import kojihub

IP = kojihub.InsertProcessor


class TestNewBuild(unittest.TestCase):
    def setUp(self):
        self.get_rpm = mock.patch('kojihub.kojihub.get_rpm').start()
        self.get_external_repo_id = mock.patch('kojihub.kojihub.get_external_repo_id').start()
        self.nextval = mock.patch('kojihub.kojihub.nextval').start()
        self.Savepoint = mock.patch('kojihub.kojihub.Savepoint').start()
        self.InsertProcessor = mock.patch('kojihub.kojihub.InsertProcessor',
                                          side_effect=self.getInsert).start()
        self.inserts = []
        self.insert_execute = mock.MagicMock()
        self.lookup_package = mock.patch('kojihub.kojihub.lookup_package').start()
        self.new_package = mock.patch('kojihub.kojihub.new_package').start()
        self.get_user = mock.patch('kojihub.kojihub.get_user').start()
        self.get_build = mock.patch('kojihub.kojihub.get_build').start()
        self.recycle_build = mock.patch('kojihub.kojihub.recycle_build').start()
        self.context = mock.patch('kojihub.kojihub.context').start()
        self.find_build_id = mock.patch('kojihub.kojihub.find_build_id').start()

    def tearDown(self):
        mock.patch.stopall()

    def getInsert(self, *args, **kwargs):
        insert = IP(*args, **kwargs)
        insert.execute = self.insert_execute
        self.inserts.append(insert)
        return insert

    def test_valid(self):
        self.get_build.return_value = None
        self.nextval.return_value = 65  # free build id
        self.new_package.return_value = 54
        self.get_user.return_value = {'id': 123}
        data = {
            'name': 'test_name',
            'version': 'test_version',
            'release': 'test_release',
            'epoch': 'test_epoch',
            'owner': 'test_owner',
            'extra': {'extra_key': 'extra_value'},
        }

        kojihub.new_build(data)

        self.assertEqual(len(self.inserts), 1)
        insert = self.inserts[0]
        self.assertEqual(insert.table, 'build')
        self.assertEqual(insert.data, {
            'completion_time': 'NOW',
            'epoch': 'test_epoch',
            'extra': '{"extra_key": "extra_value"}',
            'id': 65,
            'owner': 123,
            'pkg_id': 54,
            'release': 'test_release',
            'source': None,
            'start_time': 'NOW',
            'state': 1,
            'task_id': None,
            'draft': False,
            'version': 'test_version',
            'volume_id': 0
        })

    def test_empty_data(self):
        with self.assertRaises(koji.GenericError):
            kojihub.new_build({})
        self.assertEqual(len(self.inserts), 0)

    def test_wrong_pkg_id(self):
        self.lookup_package.side_effect = koji.GenericError
        data = {
            'pkg_id': 444,
            'name': 'test_name',
            'version': 'test_version',
            'release': 'test_release',
            'epoch': 'test_epoch',
            'owner': 'test_owner',
            'extra': {'extra_key': 'extra_value'},
        }

        with self.assertRaises(koji.GenericError):
            kojihub.new_build(data)

        self.assertEqual(len(self.inserts), 0)

    def test_missing_pkg_id_name(self):
        data = {
            'version': 'test_version',
            'release': 'test_release',
            'epoch': 'test_epoch',
            'owner': 'test_owner',
            'extra': {'extra_key': 'extra_value'},
        }

        with self.assertRaises(koji.GenericError) as cm:
            kojihub.new_build(data)

        self.assertEqual(len(self.inserts), 0)
        self.assertEqual("No name or package id provided for build", str(cm.exception))

    def test_wrong_owner(self):
        self.get_user.side_effect = koji.GenericError
        data = {
            'owner': 123456,
            'name': 'test_name',
            'version': 'test_version',
            'release': 'test_release',
            'epoch': 'test_epoch',
            'extra': {'extra_key': 'extra_value'},
        }

        with self.assertRaises(koji.GenericError):
            kojihub.new_build(data)

        self.assertEqual(len(self.inserts), 0)

    def test_missing_vre(self):
        data = {
            'name': 'test_name',
            'version': 'test_version',
            'release': 'test_release',
            'epoch': 'test_epoch',
        }

        for item in ('version', 'release', 'epoch'):
            d = data.copy()
            del d[item]
            with self.assertRaises(koji.GenericError):
                kojihub.new_build(d)

        self.assertEqual(len(self.inserts), 0)

    def test_wrong_extra(self):
        # extra dict contains unserializable data
        class CantDoJSON(object):
            pass

        data = {
            'owner': 123456,
            'name': 'test_name',
            'version': 'test_version',
            'release': 'test_release',
            'epoch': 'test_epoch',
            'extra': {'extra_key': CantDoJSON()},
        }

        with self.assertRaises(koji.GenericError) as cm:
            kojihub.new_build(data)

        self.assertEqual(len(self.inserts), 0)
        self.assertEqual("No such build extra data: %(extra)r" % data, str(cm.exception))

    def test_draft(self):
        data = {
            'owner': 123456,
            'name': 'test_name',
            'version': 'test_version',
            'release': 'test_release',
            'epoch': 'test_epoch',
            'draft': True
        }
        insert_data = {
            'completion_time': 'NOW',
            'epoch': 'test_epoch',
            'extra': '{"draft": {"target_release": "test_release", "promoted": false}}',
            'id': 108,
            'owner': 123,
            'pkg_id': 54,
            'release': 'test_release#draft_108',
            'source': None,
            'start_time': 'NOW',
            'state': 1,
            'task_id': None,
            'draft': True,
            'version': 'test_version',
            'volume_id': 0
        }
        self.nextval.return_value = 108
        self.new_package.return_value = 54
        self.get_user.return_value = {'id': 123}
        self.find_build_id.return_value = None

        kojihub.new_build(data)

        self.assertEqual(len(self.inserts), 1)
        insert = self.inserts[0]
        self.assertEqual(insert.table, 'build')
        self.assertEqual(insert.data, insert_data)
        self.get_build.assert_called_once_with(108, strict=True)
        self.assertEqual(self.get_build.call_count, 1)
        self.find_build_id.assert_called_once_with(
            {
                'name': 'test_name',
                'version': 'test_version',
                'release': 'test_release#draft_108'
            })