Files
fm-orchestrator/tests/test_mmd_resolver.py
Igor Gnatenko 85af781920 mmd_resolver: add support for streams exclusion
Also fix support for dependencies with empty streams list.

Signed-off-by: Igor Gnatenko <ignatenko@redhat.com>
2018-04-03 09:58:57 -04:00

271 lines
10 KiB
Python

# -*- coding: utf-8 -*-
#
# Copyright © 2018 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 Jan Kaluža <jkaluza@redhat.com>
# Igor Gnatenko <ignatenko@redhat.com>
import collections
import gi
gi.require_version("Modulemd", "1.0") # noqa
from gi.repository import Modulemd
import pytest
from module_build_service.mmd_resolver import MMDResolver
class TestMMDResolver:
def setup_method(self, test_method):
self.mmd_resolver = MMDResolver()
def teardown_method(self, test_method):
pass
@staticmethod
def _make_mmd(nsvc, requires):
name, stream, version = nsvc.split(":", 2)
mmd = Modulemd.Module()
mmd.set_mdversion(2)
mmd.set_name(name)
mmd.set_stream(stream)
mmd.set_summary("foo")
mmd.set_description("foo")
licenses = Modulemd.SimpleSet()
licenses.add("GPL")
mmd.set_module_licenses(licenses)
if ":" in version:
version, context = version.split(":")
mmd.set_context(context)
add_requires = Modulemd.Dependencies.add_requires
else:
add_requires = Modulemd.Dependencies.add_buildrequires
mmd.set_version(int(version))
if not isinstance(requires, list):
requires = [requires]
else:
requires = requires
deps_list = []
for reqs in requires:
deps = Modulemd.Dependencies()
for req_name, req_streams in reqs.items():
add_requires(deps, req_name, req_streams)
deps_list.append(deps)
mmd.set_dependencies(deps_list)
return mmd
@pytest.mark.parametrize(
"deps, expected", (
([], "None"),
([{"x": []}], "module(x)"),
([{"x": ["1"]}], "(module(x) with module(x:1))"),
([{"x": ["-1"]}], "(module(x) without module(x:1))"),
([{"x": ["1", "2"]}], "(module(x) with (module(x:1) or module(x:2)))"),
([{"x": ["-1", "2"]}], "(module(x) with module(x:2))"),
([{"x": [], "y": []}], "(module(x) and module(y))"),
([{"x": []}, {"y": []}], "(module(x) or module(y))"),
)
)
def test_deps2reqs(self, deps, expected):
# Sort by keys here to avoid unordered dicts
deps = [collections.OrderedDict(sorted(dep.items())) for dep in deps]
reqs = self.mmd_resolver._deps2reqs(deps)
assert str(reqs) == expected
@classmethod
def _default_mmds(cls):
return [
cls._make_mmd("gtk:1:0:c2", {"platform": ["f28"]}),
cls._make_mmd("gtk:1:0:c3", {"platform": ["f29"]}),
cls._make_mmd("gtk:2:0:c4", {"platform": ["f28"]}),
cls._make_mmd("gtk:2:0:c5", {"platform": ["f29"]}),
cls._make_mmd("foo:1:0:c2", {"platform": ["f28"]}),
cls._make_mmd("foo:1:0:c3", {"platform": ["f29"]}),
cls._make_mmd("foo:2:0:c4", {"platform": ["f28"]}),
cls._make_mmd("foo:2:0:c5", {"platform": ["f29"]}),
cls._make_mmd("platform:f28:0:c10", {}),
cls._make_mmd("platform:f29:0:c11", {}),
]
@classmethod
def _default_mmds_with_multiple_requires(cls):
return [
cls._make_mmd("gtk:1:0:c2", {"font": ["a", "b"], "platform": ["f28"]}),
cls._make_mmd("gtk:1:0:c3", {"font": ["a", "b"], "platform": ["f29"]}),
cls._make_mmd("gtk:2:0:c4", {"font": ["a", "b"], "platform": ["f28"]}),
cls._make_mmd("gtk:2:0:c5", {"font": ["a", "b"], "platform": ["f29"]}),
cls._make_mmd("font:a:0:c6", {"platform": ["f28"]}),
cls._make_mmd("font:a:0:c7", {"platform": ["f29"]}),
cls._make_mmd("font:b:0:c8", {"platform": ["f28"]}),
cls._make_mmd("font:b:0:c9", {"platform": ["f29"]}),
cls._make_mmd("platform:f28:0:c10", {}),
cls._make_mmd("platform:f29:0:c11", {}),
]
def test_solve_tree(self):
for mmd in self._default_mmds_with_multiple_requires():
self.mmd_resolver.add_modules(mmd)
app = self._make_mmd("app:1:0", {"gtk": ["1", "2"]})
expanded = self.mmd_resolver.solve(app)
expected = set([
frozenset(["app:1:0:0:src",
"font:a:0:c6:x86_64",
"gtk:1:0:c2:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"font:a:0:c6:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
])
assert expanded == expected
def test_solve_tree_buildrequire_platform(self):
for mmd in self._default_mmds_with_multiple_requires():
self.mmd_resolver.add_modules(mmd)
app = self._make_mmd("app:1:0", {"gtk": ["1", "2"], "platform": ["f28"]})
expanded = self.mmd_resolver.solve(app)
expected = set([
frozenset(["app:1:0:0:src",
"font:a:0:c6:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"font:a:0:c6:x86_64",
"gtk:1:0:c2:x86_64",
"platform:f28:0:c10:x86_64"]),
])
assert expanded == expected
def test_solve_tree_multiple_build_requires(self):
for mmd in self._default_mmds():
self.mmd_resolver.add_modules(mmd)
app = self._make_mmd("app:1:0", {"gtk": ["1", "2"], "foo": ["1", "2"]})
expanded = self.mmd_resolver.solve(app)
expected = set([
frozenset(["app:1:0:0:src",
"foo:1:0:c2:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"foo:2:0:c4:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"foo:1:0:c2:x86_64",
"gtk:1:0:c2:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"foo:2:0:c4:x86_64",
"gtk:1:0:c2:x86_64",
"platform:f28:0:c10:x86_64"]),
])
assert expanded == expected
def test_solve_multiple_requires_pairs(self):
for mmd in self._default_mmds():
self.mmd_resolver.add_modules(mmd)
app = self._make_mmd(
"app:1:0",
[{"gtk": ["1"], "foo": ["1"]},
{"gtk": ["2"], "foo": ["1", "2"]}])
expanded = self.mmd_resolver.solve(app)
expected = set([
frozenset(["app:1:0:1:src",
"foo:1:0:c2:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:1:src",
"foo:2:0:c4:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"foo:1:0:c2:x86_64",
"gtk:1:0:c2:x86_64",
"platform:f28:0:c10:x86_64"]),
])
assert expanded == expected
def test_solve_multiple_requires_pairs_buildrequire_platform(self):
for mmd in self._default_mmds():
self.mmd_resolver.add_modules(mmd)
app = self._make_mmd(
"app:1:0",
[{"gtk": ["1"], "foo": ["1"]},
{"gtk": ["2"], "foo": ["1", "2"], "platform": ["f28"]}])
expanded = self.mmd_resolver.solve(app)
expected = set([
frozenset(["app:1:0:1:src",
"foo:1:0:c2:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"foo:1:0:c2:x86_64",
"gtk:1:0:c2:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:1:src",
"foo:2:0:c4:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
])
assert expanded == expected
def test_solve_multiple_requires_pairs_multiple_requires(self):
for mmd in self._default_mmds_with_multiple_requires():
self.mmd_resolver.add_modules(mmd)
app = self._make_mmd(
"app:1:0",
[{"gtk": ["1"], "font": ["a"]},
{"gtk": ["2"], "font": ["b"]}])
expanded = self.mmd_resolver.solve(app)
expected = set([
frozenset(["app:1:0:1:src",
"font:b:0:c8:x86_64",
"gtk:2:0:c4:x86_64",
"platform:f28:0:c10:x86_64"]),
frozenset(["app:1:0:0:src",
"font:a:0:c6:x86_64",
"gtk:1:0:c2:x86_64",
"platform:f28:0:c10:x86_64"]),
])
assert expanded == expected