debian-koji/tests/test_www/test_repoinfo.py
Yuming Zhu ca05418fb5 unittest: use unittest.mock instead of mock
because the absence of unittest.mock on python2.7, we still fallback to
mock
2024-10-23 16:35:30 +00:00

46 lines
1.8 KiB
Python

from unittest import mock
import unittest
from .loadwebindex import webidx
class TestRepoInfo(unittest.TestCase):
def setUp(self):
self.get_server = mock.patch.object(webidx, "_getServer").start()
self.gen_html = mock.patch.object(webidx, '_genHTML').start()
self.server = mock.MagicMock()
self.environ = {
'koji.options': {
'SiteName': 'test',
'KojiFilesURL': 'https://server.local/files',
},
'koji.currentUser': None
}
self.repo_id = '5'
def tearDown(self):
mock.patch.stopall()
def test_repoinfo_dist(self):
"""Test repoinfo function - dist repo"""
self.get_server.return_value = self.server
self.server.repoInfo.return_value = {'dist': True, 'id': int(self.repo_id),
'tag_name': 'test-tag'}
self.server.listBuildroots.return_value = []
webidx.repoinfo(self.environ, self.repo_id)
self.server.repoInfo.assert_called_once_with(int(self.repo_id), strict=False)
self.server.listBuildroots.assert_called_once_with(repoID=int(self.repo_id))
def test_repoinfo_not_dist(self):
"""Test repoinfo function - not dist repo"""
self.get_server.return_value = self.server
self.server.repoInfo.return_value = {'dist': False, 'id': int(self.repo_id),
'tag_name': 'test-tag'}
self.server.listBuildroots.return_value = [{'id': 1, 'repo_id': int(self.repo_id)}]
webidx.repoinfo(self.environ, self.repo_id)
self.server.repoInfo.assert_called_once_with(int(self.repo_id), strict=False)
self.server.listBuildroots.assert_called_once_with(repoID=int(self.repo_id))