Files
fm-orchestrator/tests/test_models/test_models.py

188 lines
8.9 KiB
Python

# Copyright (c) 2016 Red Hat, Inc.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
#
# Written by Ralph Bean <rbean@redhat.com>
import os
import pytest
from mock import patch
from module_build_service import conf, Modulemd
from module_build_service.models import ComponentBuild, ModuleBuild, make_session
from module_build_service.utils import to_text_type
from tests import (init_data as init_data_contexts, clean_database, make_module)
from tests.test_models import init_data, module_build_from_modulemd
class TestModels:
def setup_method(self, test_method):
init_data()
def test_app_sqlalchemy_events(self):
with make_session(conf) as session:
component_build = ComponentBuild()
component_build.package = 'before_models_committed'
component_build.scmurl = \
('git://pkgs.domain.local/rpms/before_models_committed?'
'#9999999999999999999999999999999999999999')
component_build.format = 'rpms'
component_build.task_id = 999999999
component_build.state = 1
component_build.nvr = ('before_models_committed-0.0.0-0'
'.module_before_models_committed_0_0')
component_build.batch = 1
component_build.module_id = 1
session.add(component_build)
session.commit()
with make_session(conf) as session:
c = session.query(ComponentBuild).filter(ComponentBuild.id == 1).one()
assert c.component_builds_trace[0].id == 1
assert c.component_builds_trace[0].component_id == 1
assert c.component_builds_trace[0].state == 1
assert c.component_builds_trace[0].state_reason is None
assert c.component_builds_trace[0].task_id == 999999999
def test_context_functions(self):
""" Test that the build_context, runtime_context, and context hashes are correctly
determined"""
build = ModuleBuild.query.filter_by(id=1).one()
yaml_path = os.path.join(
os.path.dirname(__file__), '..', 'staged_data', 'testmodule_dependencies.yaml')
mmd = Modulemd.Module.new_from_file(yaml_path)
mmd.upgrade()
build.modulemd = to_text_type(mmd.dumps())
(build.ref_build_context, build.build_context, build.runtime_context,
build.context) = ModuleBuild.contexts_from_mmd(build.modulemd)
assert build.ref_build_context == 'f6e2aeec7576196241b9afa0b6b22acf2b6873d7'
assert build.build_context == '089df24993c037e10174f3fa7342ab4dc191a4d4'
assert build.runtime_context == 'bbc84c7b817ab3dd54916c0bcd6c6bdf512f7f9c'
assert build.context == '3ee22b28'
def test_siblings_property(self):
""" Tests that the siblings property returns the ID of all modules with
the same name:stream:version
"""
clean_database()
yaml_path = os.path.join(
os.path.dirname(__file__), '..', 'staged_data', 'formatted_testmodule.yaml')
mmd = Modulemd.Module.new_from_file(yaml_path)
mmd.upgrade()
with make_session(conf) as session:
for i in range(3):
build = module_build_from_modulemd(to_text_type(mmd.dumps()))
build.build_context = 'f6e2aeec7576196241b9afa0b6b22acf2b6873d' + str(i)
build.runtime_context = 'bbc84c7b817ab3dd54916c0bcd6c6bdf512f7f9c' + str(i)
session.add(build)
session.commit()
build_one = ModuleBuild.query.get(2)
assert build_one.siblings == [3, 4]
@pytest.mark.parametrize('stream,right_pad,expected', [
['f27', True, 270000.0],
['f27.02.30', True, 270230.0],
['f27', False, 27.0],
['f27.02.30', False, 270230.0],
['el8', True, 080000.0],
['el8.1.0', True, 080100.0],
['el8.z', True, 080000.2],
['el8.1.0.z', True, 080100.3],
])
@patch.object(conf, 'stream_suffixes', new={
r'el\d+\.z': 0.2, r'el\d+\.\d+\.\d+\.z': 0.3
})
def test_get_stream_version(self, stream, right_pad, expected):
assert expected == ModuleBuild.get_stream_version(stream, right_pad)
class TestModelsGetStreamsContexts:
def test_get_last_build_in_all_streams(self):
init_data_contexts(contexts=True)
with make_session(conf) as session:
builds = ModuleBuild.get_last_build_in_all_streams(
session, "nginx")
builds = ["%s:%s:%s" % (build.name, build.stream, str(build.version))
for build in builds]
assert builds == ["nginx:%d:%d" % (i, i + 2) for i in range(10)]
def test_get_last_build_in_all_stream_last_version(self):
init_data_contexts(contexts=False)
with make_session(conf) as session:
builds = ModuleBuild.get_last_build_in_all_streams(
session, "nginx")
builds = ["%s:%s:%s" % (build.name, build.stream, str(build.version))
for build in builds]
assert builds == ["nginx:1:11"]
def test_get_last_builds_in_stream(self):
init_data_contexts(contexts=True)
with make_session(conf) as session:
builds = ModuleBuild.get_last_builds_in_stream(
session, "nginx", "1")
builds = ["%s:%s:%s:%s" % (build.name, build.stream, str(build.version),
build.context) for build in builds]
assert builds == ['nginx:1:3:d5a6c0fa', 'nginx:1:3:795e97c1']
def test_get_last_builds_in_stream_version_lte(self):
init_data_contexts(1, multiple_stream_versions=True)
with make_session(conf) as session:
builds = ModuleBuild.get_last_builds_in_stream_version_lte(
session, "platform", 290100)
builds = set(["%s:%s:%s:%s" % (build.name, build.stream, str(build.version),
build.context) for build in builds])
assert builds == set(['platform:f29.0.0:3:00000000', 'platform:f29.1.0:3:00000000'])
def test_get_last_builds_in_stream_version_lte_different_versions(self):
"""
Tests that get_last_builds_in_stream_version_lte works in case the
name:stream_ver modules have different versions.
"""
clean_database(False)
make_module("platform:f29.1.0:10:old_version", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.1.0:15:c11.another", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.1.0:15:c11", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.2.0:0:old_version", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.2.0:1:c11", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.3.0:15:old_version", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.3.0:20:c11", {}, {}, virtual_streams=["f29"])
with make_session(conf) as session:
builds = ModuleBuild.get_last_builds_in_stream_version_lte(
session, "platform", 290200)
builds = set(["%s:%s:%s:%s" % (build.name, build.stream, str(build.version),
build.context) for build in builds])
assert builds == set(['platform:f29.1.0:15:c11', 'platform:f29.1.0:15:c11.another',
'platform:f29.2.0:1:c11'])
def test_add_virtual_streams_filter(self):
clean_database(False)
make_module("platform:f29.1.0:10:c1", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.1.0:15:c1", {}, {}, virtual_streams=["f29"])
make_module("platform:f29.3.0:15:old_version", {}, {}, virtual_streams=["f28", "f29"])
make_module("platform:f29.3.0:20:c11", {}, {}, virtual_streams=["f30"])
with make_session(conf) as session:
query = session.query(ModuleBuild).filter_by(name="platform")
query = ModuleBuild._add_virtual_streams_filter(session, query, ["f28", "f29"])
count = query.count()
assert count == 3