fm-orchestrator/tests/test_models/test_models.py

# 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

from tests.test_models import init_data, module_build_from_modulemd
from tests import init_data as init_data_contexts, clean_database
from module_build_service import conf, Modulemd
from module_build_service.models import ComponentBuild, ModuleBuild, make_session


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 = mmd.dumps()
        build.build_context, build.runtime_context = ModuleBuild.contexts_from_mmd(build.modulemd)
        assert build.build_context == 'f6e2aeec7576196241b9afa0b6b22acf2b6873d7'
        assert build.runtime_context == 'bbc84c7b817ab3dd54916c0bcd6c6bdf512f7f9c'
        assert build.context == 'f1a17afd'

    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(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]


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']