mirror of
https://pagure.io/fm-orchestrator.git
synced 2026-02-11 00:55:00 +08:00
8c6cfb702d
This also removes the outdated comments around authorship of each file. If there is still interest in this information, one can just look at the git history.
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
# SPDX-License-Identifier: MIT
|
|
import mock
|
|
|
|
import module_build_service.models
|
|
import module_build_service.builder
|
|
import module_build_service.resolver
|
|
|
|
from tests import init_data
|
|
|
|
from module_build_service.builder import GenericBuilder
|
|
from mock import patch
|
|
|
|
|
|
class TestGenericBuilder:
|
|
def setup_method(self, test_method):
|
|
init_data(1)
|
|
|
|
@patch("module_build_service.resolver.DBResolver")
|
|
@patch("module_build_service.builder.base.GenericResolver")
|
|
def test_default_buildroot_groups_cache(self, generic_resolver, resolver, db_session):
|
|
mbs_groups = {"buildroot": [], "srpm-buildroot": []}
|
|
|
|
resolver = mock.MagicMock()
|
|
resolver.backend = "mbs"
|
|
resolver.resolve_profiles.return_value = mbs_groups
|
|
|
|
expected_groups = {"build": [], "srpm-build": []}
|
|
|
|
module = module_build_service.models.ModuleBuild.get_by_id(db_session, 1)
|
|
|
|
generic_resolver.create.return_value = resolver
|
|
# Call default_buildroot_groups, the result should be cached.
|
|
ret = GenericBuilder.default_buildroot_groups(db_session, module)
|
|
assert ret == expected_groups
|
|
resolver.resolve_profiles.assert_called_once()
|
|
resolver.resolve_profiles.reset_mock()
|
|
|
|
# Now try calling it again to verify resolve_profiles is not called,
|
|
# because it is cached.
|
|
generic_resolver.create.return_value = resolver
|
|
ret = GenericBuilder.default_buildroot_groups(db_session, module)
|
|
assert ret == expected_groups
|
|
resolver.resolve_profiles.assert_not_called()
|
|
resolver.resolve_profiles.reset_mock()
|
|
|
|
# And now try clearing the cache and call it again.
|
|
generic_resolver.create.return_value = resolver
|
|
GenericBuilder.clear_cache(module)
|
|
ret = GenericBuilder.default_buildroot_groups(db_session, module)
|
|
assert ret == expected_groups
|
|
resolver.resolve_profiles.assert_called_once()
|
|
|
|
def test_get_build_weights(self):
|
|
weights = GenericBuilder.get_build_weights(["httpd", "apr"])
|
|
assert weights == {"httpd": 1.5, "apr": 1.5}
|