stub kojid tests

This commit is contained in:
Mike McLean 2016-10-19 20:23:47 -04:00
parent 144a4f1b30
commit 695ded30c8
3 changed files with 54 additions and 0 deletions

View file

@ -0,0 +1,41 @@
import unittest
import os
import sys
import mock
import rpm
import tempfile
from loadkojid import kojid
class FakeHeader(dict):
def __init__(self, **kwargs):
for key in kwargs:
kname = "RPMTAG_%s" % key.upper()
hkey = getattr(rpm, kname)
self.__setitem__(hkey, kwargs[key])
class TestChooseTaskarch(unittest.TestCase):
def setUp(self):
task_id = 99
method = 'build'
params = []
self.session = mock.MagicMock()
self.options = mock.MagicMock()
workdir = tempfile.mkdtemp()
self.handler = kojid.BuildTask(task_id, method, params, self.session,
self.options, workdir)
self.readSRPMHeader = mock.MagicMock()
self.handler.readSRPMHeader = self.readSRPMHeader
def test_noarch(self):
self.readSRPMHeader.return_value = FakeHeader(buildarchs=['noarch'],
exclusivearch=[], excludearch=[])
self.handler.choose_taskarch('noarch', 'srpm', 'build_tag')