mirror of
https://pagure.io/fm-orchestrator.git
synced 2026-02-06 23:03:18 +08:00
f24cd4222f
Please note that this patch does not change the use of database session in MBS. So, in the frontend, the database session is still managed by Flask-SQLAlchemy, that is the db.session. And the backend, running event handlers, has its own database session created from SQLAclehmy session API directly. This patch aims to reduce the number of scoped_session created when call original function make_db_session. For technical detailed information, please refer to SQLAlchemy documentation Contextual/Thread-local Sessions. As a result, a global scoped_session is accessible from the code running inside backend, both the event handlers and functions called from handlers. The library code shared by frontend and backend, like resolvers, has no change. Similarly, db.session is only used to recreate database for every test. Signed-off-by: Chenxiong Qi <cqi@redhat.com>
58 lines
2.1 KiB
Python
58 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.db_session import db_session
|
|
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):
|
|
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}
|