dak/tests/test_packagelist.py

#! /usr/bin/env python3
#
# Copyright (C) 2014, Ansgar Burchardt <ansgar@debian.org>
#
# 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

from base_test import DakTestCase

from daklib.packagelist import PackageList


class FakeArchitecture:
    def __init__(self, name):
        self.arch_string = name


class FakeSuite:
    def __init__(self, *architectures):
        self.architectures = [FakeArchitecture(a) for a in architectures]


source_all = {
    "Package-List": "\n libdune-common-doc deb doc optional arch=all\n",
    "Binary": "libdune-common-doc\n",
}

source_any = {
    "Package-List": "\n libdune-common-dev deb libdevel optional arch=any\n",
    "Binary": "libdune-common-dev\n",
}

source_all_any = {
    "Package-List": "\n libdune-common-dev deb libdevel optional arch=any\nlibdune-common-doc deb doc optional arch=all\n",
    "Binary": "libdune-common-dev, libdune-common-doc\n",
}

source_amd64 = {
    "Package-List": "\n libdune-common-dev deb libdevel optional arch=amd64\n",
    "Binary": "libdune-common-dev\n",
}

source_linuxany = {
    "Package-List": "\n libdune-common-dev deb libdevel optional arch=linux-any\n",
    "Binary": "libdune-common-dev\n",
}

source_noarch = {
    "Package-List": "\n libdune-common-dev deb libdevel optional\n",
    "Binary": "libdune-common-dev\n",
}

source_fallback = {
    "Binary": "libdune-common-dev\n",
}

source_profiles = {
    "Package-List": "\n pkg-a deb misc optional arch=any profile=!stage1"
    "\n pkg-b deb misc optional arch=any profile=!stage1,!stage2"
    "\n pkg-c deb misc optional arch=any profile=stage1"
    "\n pkg-d deb misc optional arch=any profile=stage1,stage2"
    "\n pkg-e deb misc optional arch=any profile=stage1+stage2"
    "\n pkg-f deb misc optional arch=any profile=!stage1+!stage2"
    "\n pkg-g deb misc optional arch=any profile=!stage1+stage2"
    "\n pkg-h deb misc optional arch=any profile=stage1+!stage2"
    "\n",
    "Binary": "pkg-a, pkg-b, pkg-c, pkg-d, pkg-e, pkg-f, pkg-g, pkg-h\n",
}


class TestPackageList(DakTestCase):
    def testArchAll(self):
        pl = PackageList(source_all)

        self.assertTrue(pl.has_arch_indep_packages())
        self.assertFalse(pl.has_arch_dep_packages())

        suite_amd64 = FakeSuite("amd64")
        p_amd64 = pl.packages_for_suite(suite_amd64)
        self.assertEqual(len(p_amd64), 0)

        suite_all = FakeSuite("all")
        p_all = pl.packages_for_suite(suite_all)
        self.assertEqual(len(p_all), 1)

        suite_all_amd64 = FakeSuite("amd64", "all")
        p_all_amd64 = pl.packages_for_suite(suite_all_amd64)
        self.assertEqual(len(p_all_amd64), 1)

        p = p_all[0]
        self.assertEqual(p.name, "libdune-common-doc")
        self.assertEqual(p.type, "deb")
        self.assertEqual(p.section, "doc")
        self.assertEqual(p.component, "main")
        self.assertEqual(p.priority, "optional")
        self.assertEqual(p.architectures, ["all"])

    def testArchAny(self):
        pl = PackageList(source_any)

        self.assertFalse(pl.has_arch_indep_packages())
        self.assertTrue(pl.has_arch_dep_packages())

        suite_amd64 = FakeSuite("amd64")
        p_amd64 = pl.packages_for_suite(suite_amd64)
        self.assertEqual(len(p_amd64), 1)

        suite_all = FakeSuite("all")
        p_all = pl.packages_for_suite(suite_all)
        self.assertEqual(len(p_all), 0)

        suite_all_amd64 = FakeSuite("amd64", "all")
        p_all_amd64 = pl.packages_for_suite(suite_all_amd64)
        self.assertEqual(len(p_all_amd64), 1)

    def testArchAnyAll(self):
        pl = PackageList(source_all_any)

        self.assertTrue(pl.has_arch_indep_packages())
        self.assertTrue(pl.has_arch_dep_packages())

        suite_amd64 = FakeSuite("amd64")
        p_amd64 = pl.packages_for_suite(suite_amd64)
        self.assertEqual(len(p_amd64), 1)

        suite_amd64_i386 = FakeSuite("amd64", "i386")
        p_amd64_i386 = pl.packages_for_suite(suite_amd64_i386)
        self.assertEqual(len(p_amd64_i386), 1)

        suite_all = FakeSuite("all")
        p_all = pl.packages_for_suite(suite_all)
        self.assertEqual(len(p_all), 1)

        suite_all_amd64 = FakeSuite("amd64", "all")
        p_all_amd64 = pl.packages_for_suite(suite_all_amd64)
        self.assertEqual(len(p_all_amd64), 2)

    def testArchAmd64(self):
        pl = PackageList(source_amd64)

        self.assertFalse(pl.has_arch_indep_packages())
        self.assertTrue(pl.has_arch_dep_packages())

        suite_amd64 = FakeSuite("amd64")
        p_amd64 = pl.packages_for_suite(suite_amd64)
        self.assertEqual(len(p_amd64), 1)

        suite_i386 = FakeSuite("i386")
        p_i386 = pl.packages_for_suite(suite_i386)
        self.assertEqual(len(p_i386), 0)

    def testArchLinuxAny(self):
        pl = PackageList(source_linuxany)

        self.assertFalse(pl.has_arch_indep_packages())
        self.assertTrue(pl.has_arch_dep_packages())

        suite_amd64 = FakeSuite("amd64")
        p_amd64 = pl.packages_for_suite(suite_amd64)
        self.assertEqual(len(p_amd64), 1)

        suite_i386 = FakeSuite("i386")
        p_i386 = pl.packages_for_suite(suite_i386)
        self.assertEqual(len(p_i386), 1)

        suite_kfreebsdi386 = FakeSuite("kfreebsd-i386")
        p_kfreebsdi386 = pl.packages_for_suite(suite_kfreebsdi386)
        self.assertEqual(len(p_kfreebsdi386), 0)

        suite_source = FakeSuite("source")
        p_source = pl.packages_for_suite(suite_source)
        self.assertEqual(len(p_source), 0)

    def testNoArch(self):
        pl = PackageList(source_noarch)

        self.assertIsNone(pl.has_arch_indep_packages())
        self.assertIsNone(pl.has_arch_dep_packages())

        suite_amd64 = FakeSuite("amd64")
        p_amd64 = pl.packages_for_suite(suite_amd64)
        self.assertEqual(len(p_amd64), 1)

    def testFallback(self):
        pl = PackageList(source_fallback)

        self.assertIsNone(pl.has_arch_indep_packages())
        self.assertIsNone(pl.has_arch_dep_packages())

        suite_amd64 = FakeSuite("amd64")
        p_amd64 = pl.packages_for_suite(suite_amd64)
        self.assertEqual(len(p_amd64), 1)

    def testProfiles(self):
        pl = PackageList(source_profiles)

        self.assertEqual(len(pl.package_list), 8)

        built_in_default_profile = {"pkg-a", "pkg-b", "pkg-f", "pkg-g", "pkg-h"}
        not_built_in_default_profile = {"pkg-c", "pkg-d", "pkg-e"}

        for entry in pl.package_list:
            if entry.built_in_default_profile():
                self.assertIn(entry.name, built_in_default_profile)
            else:
                self.assertIn(entry.name, not_built_in_default_profile)

        suite_amd64 = FakeSuite("amd64")
        ps_only_default = pl.packages_for_suite(suite_amd64)
        ps_all = pl.packages_for_suite(suite_amd64, only_default_profile=False)
        self.assertSetEqual({p.name for p in ps_only_default}, built_in_default_profile)
        self.assertSetEqual(
            {p.name for p in ps_all},
            built_in_default_profile | not_built_in_default_profile,
        )


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