dak/tests/test_utils.py

#! /usr/bin/env python3
#
# Copyright (C) 2017, Niels Thykier <niels@thykier.net>
#
# This program 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.
#
# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

import unittest

import apt_pkg
from base_test import DakTestCase

from daklib.utils import (
    ArchKey,
    _gpg_get_addresses_from_listing,
    extract_component_from_section,
    is_in_debug_section,
    parse_built_using,
)

apt_pkg.init()


class UtilsTest(DakTestCase):

    def test_utils_ArchKey(self):
        source = ArchKey("source")
        arch1 = ArchKey("amd64")
        arch2 = ArchKey("i386")

        assert source.__eq__(source)
        assert not source.__lt__(source)

        assert arch1.__eq__(arch1)
        assert not arch1.__lt__(arch1)

        assert not source.__eq__(arch1)
        assert not arch1.__eq__(source)
        assert source.__lt__(arch1)
        assert not arch1.__lt__(source)

        assert not arch1.__eq__(arch2)
        assert not arch2.__eq__(arch1)
        assert arch1.__lt__(arch2)
        assert not arch2.__lt__(arch1)

    def test_is_in_debug_section(self):
        data = [
            (
                {
                    "Section": "debug",
                    "Auto-Built-Package": "debug-symbols",
                },
                True,
            ),
            (
                {
                    "Section": "non-free/debug",
                    "Auto-Built-Package": "debug-symbols",
                },
                True,
            ),
            (
                {
                    "Section": "debug",
                    "Auto-Built-Package": "other",
                },
                False,
            ),
            (
                {
                    "Section": "non-free/debug",
                    "Auto-Built-Package": "other",
                },
                False,
            ),
            (
                {
                    "Section": "debug",
                },
                False,
            ),
            (
                {
                    "Section": "non-free/debug",
                },
                False,
            ),
        ]
        for ctrl, r in data:
            self.assertEqual(is_in_debug_section(ctrl), r)

    def test_parse_built_using(self):
        field = "binutils (= 1.0-1), gcc-6 (= 6.3-2)"
        expected = [("binutils", "1.0-1"), ("gcc-6", "6.3-2")]
        ctrl = {
            "Built-Using": field,
        }
        self.assertEqual(parse_built_using(ctrl), expected)
        self.assertEqual(parse_built_using({}), [])

    def test_extract_component_from_section(self):
        data = [
            # Argument is passed through as first return value. There
            # is a comment in docs/TODO.old suggesting that it should
            # be changed.
            ("utils", ("utils", "main")),
            ("main/utils", ("main/utils", "main")),
            ("non-free/libs", ("non-free/libs", "non-free")),
            ("contrib/net", ("contrib/net", "contrib")),
            ("non-free/two/slashes", ("non-free/two/slashes", "non-free")),
        ]
        for v, r in data:
            self.assertEqual(extract_component_from_section(v), r)

    def test_gpg_get_addresses_from_listing(self):
        # Test with output containing a uid encoded in Latin-1 and UTF-8 each
        data = (
            b"tru::1:1610798976:1673870895:3:1:5\n"
            b"pub:u:3072:1:4847924E3DEDDF1E:1610798895:1673870895::u:::scESC::::::23::0:\n"
            b"fpr:::::::::AC6DFC4B78223BFCC1C064004847924E3DEDDF1E:\n"
            b"uid:u::::1610798964::1D4AB6BD4AB8A446C196A1801F367B83B6141CC0::Markus Mustermann-S\xf6der <mustermann-soeder@example.org>::::::::::0:\n"
            b"uid:u::::1610798895::F60CF610557A38E2943DECEEA5B02EE57B1ECC24::Markus Mustermann-S\xc3\xb6der <mustermann-soeder@example.com>::::::::::0:\n"
            b"sub:u:3072:1:C3BCC6A42035DDB3:1610798895:1673870895:::::e::::::23:\n"
            b"fpr:::::::::C2CC5D5744746B5E7F5F7286C3BCC6A42035DDB3:\n"
        )

        addresses = _gpg_get_addresses_from_listing(data)
        expected = ["mustermann-soeder@example.org", "mustermann-soeder@example.com"]
        self.assertEqual(addresses, expected)


if __name__ == "__main__":
    unittest.main()