File: test_package_query.py

package info (click to toggle)
dnf5 5.4.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 17,960 kB
  • sloc: cpp: 94,312; python: 3,370; xml: 1,073; ruby: 600; sql: 250; ansic: 232; sh: 104; perl: 62; makefile: 30
file content (170 lines) | stat: -rw-r--r-- 6,748 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
# Copyright Contributors to the DNF5 project.
# Copyright Contributors to the libdnf project.
# SPDX-License-Identifier: GPL-2.0-or-later
#
# This file is part of libdnf: https://github.com/rpm-software-management/libdnf/
#
# Libdnf is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
#
# Libdnf is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with libdnf.  If not, see <https://www.gnu.org/licenses/>.

import unittest

import libdnf5

import base_test_case


class TestPackageQuery(base_test_case.BaseTestCase):
    def setUp(self):
        super().setUp()
        self.add_repo_repomd("repomd-repo1")

    def test_size(self):
        query = libdnf5.rpm.PackageQuery(self.base)
        self.assertEqual(query.size(), 3)

    def test_iterate_package_query(self):
        query = libdnf5.rpm.PackageQuery(self.base)

        # Iterates over reference "query".
        prev_id = 0
        for pkg in query:
            id = pkg.get_id().id
            self.assertGreater(id, prev_id)
            prev_id = id
        self.assertLess(prev_id, self.package_sack.get_nsolvables())
        self.assertGreaterEqual(prev_id, query.size())

        # Similar to above, but longer notation.
        query_iterator = iter(query)
        prev_id = 0
        while True:
            try:
                pkg = next(query_iterator)
                id = pkg.get_id().id
                self.assertGreater(id, prev_id)
                prev_id = id
            except StopIteration:
                break
        self.assertLess(prev_id, self.package_sack.get_nsolvables())
        self.assertGreaterEqual(prev_id, query.size())

    def test_iterate_package_query2(self):
        # Tests the iteration of an unreferenced PackageQuery object. The iterator must hold a reference
        # to the iterated object, otherwise the gargabe collector can remove the object.

        # Iterates directly over "libdnf5.rpm.PackageQuery(self.base)" result. No helping reference.
        prev_id = 0
        for pkg in libdnf5.rpm.PackageQuery(self.base):
            id = pkg.get_id().id
            self.assertGreater(id, prev_id)
            prev_id = id
        self.assertLess(prev_id, self.package_sack.get_nsolvables())
        self.assertGreaterEqual(
            prev_id, libdnf5.rpm.PackageQuery(self.base).size())

        # Another test. The iterator is created from the "query" reference, but the reference
        # is removed (set to "None") before starting the iteration.
        query = libdnf5.rpm.PackageQuery(self.base)
        query_iterator = iter(query)
        query = None
        prev_id = 0
        while True:
            try:
                pkg = next(query_iterator)
                id = pkg.get_id().id
                self.assertGreater(id, prev_id)
                prev_id = id
            except StopIteration:
                break
        self.assertLess(prev_id, self.package_sack.get_nsolvables())
        self.assertGreaterEqual(
            prev_id, libdnf5.rpm.PackageQuery(self.base).size())

    def test_filter_name(self):
        # Test QueryCmp::EQ
        query = libdnf5.rpm.PackageQuery(self.base)
        query.filter_name(["pkg"])
        self.assertEqual(query.size(), 1)
        # TODO(dmach): implement __str__()
        self.assertEqual([i.get_nevra() for i in query], ["pkg-1.2-3.x86_64"])

        # ---

        # Test QueryCmp::GLOB
        query = libdnf5.rpm.PackageQuery(self.base)
        query.filter_name(["pk*"], libdnf5.common.QueryCmp_GLOB)
        self.assertEqual(query.size(), 2)
        # TODO(dmach): implement __str__()
        self.assertEqual([i.get_nevra() for i in query], [
                         "pkg-1.2-3.x86_64", "pkg-libs-1:1.3-4.x86_64"])

    def test_resolve_pkg_spec(self):
        # Test passing empty forms
        query = libdnf5.rpm.PackageQuery(self.base)
        settings = libdnf5.base.ResolveSpecSettings()

        match, nevra = query.resolve_pkg_spec("pkg*", settings, True)

        self.assertEqual(query.size(), 2)
        self.assertEqual([i.get_nevra() for i in query], [
                         "pkg-1.2-3.x86_64", "pkg-libs-1:1.3-4.x86_64"])
        self.assertTrue(match)
        self.assertEqual(nevra.get_name(), "pkg*")
        self.assertTrue(nevra.has_just_name())

        # Test passing an explicit list of forms
        query = libdnf5.rpm.PackageQuery(self.base)
        settings = libdnf5.base.ResolveSpecSettings()
        settings.set_nevra_forms(
            libdnf5.rpm.VectorNevraForm(1, libdnf5.rpm.Nevra.Form_NA))

        match, nevra = query.resolve_pkg_spec("pkg.x86_64", settings, True)

        self.assertEqual(query.size(), 1)
        self.assertEqual([i.get_nevra() for i in query], ["pkg-1.2-3.x86_64"])
        self.assertTrue(match)
        self.assertEqual(nevra.get_name(), "pkg")
        self.assertEqual(nevra.get_arch(), "x86_64")
        self.assertFalse(nevra.has_just_name())

    def test_pkg_reason_value(self):
        # Test wrapper for reason enum
        query = libdnf5.rpm.PackageQuery(self.base)
        query.filter_name(["pkg"])
        package = next(iter(query))
        self.assertEqual(package.get_reason(),
                         libdnf5.transaction.TransactionItemReason_NONE)

    def test_pkg_query_without_setup(self):
        # Create a new Base object
        base = libdnf5.base.Base()

        # Try to create a package query without running base.setup()
        with self.assertRaisesRegex(libdnf5.exception.UserAssertionError,
                                    'base/base_impl.hpp:[0-9]+: libdnf5::solv::RpmPool& libdnf5::Base::Impl::get_rpm_pool\\(\\):'
                                    ' API Assertion \'pool\' failed:'
                                    ' Base instance was not fully initialized by Base::setup()'):
            libdnf5.rpm.PackageQuery(base)

    def test_pkg_get_changelogs(self):
        # Test wrapper for changelogs vector
        query = libdnf5.rpm.PackageQuery(self.base)
        query.filter_name(["pkg"])
        package = next(iter(query))
        logs = package.get_changelogs()
        self.assertEqual(2, logs.size())
        log = next(iter(logs))
        self.assertEqual('First change', log.get_text())
        self.assertEqual('Joe Black', log.get_author())
        self.assertEqual(1641027600, log.get_timestamp())