fix unit tests

This commit is contained in:
Mike McLean 2025-03-24 19:46:57 -04:00
parent e7bb3ed14b
commit 1ecba244e6
2 changed files with 3 additions and 0 deletions

View file

@ -13,6 +13,7 @@ class TestGetNextTask(unittest.TestCase):
def setUp(self):
self.options = mock.MagicMock()
self.options.single = False
self.session = mock.MagicMock()
self.tm = koji.daemon.TaskManager(self.options, self.session)
self.tm.readyForTask = mock.MagicMock()
@ -65,6 +66,7 @@ class TestTakeTask(unittest.TestCase):
def setUp(self):
self.options = mock.MagicMock()
self.options.single = False
self.session = mock.MagicMock()
self.tm = koji.daemon.TaskManager(self.options, self.session)
self.tm.readyForTask = mock.MagicMock()