dak/tests/dbtest_cruft.py

#! /usr/bin/env python3

import unittest

from db_test import DBDakTestCase

from daklib.cruft import DejavuBinary, NamedSource, get_package_names, newer_version
from daklib.dbconn import DBBinary, DBSource, PoolFile, get_suite


class CruftTestCase(DBDakTestCase):
    """
    This class checks various functions of cruft-report.
    """

    def setUp(self):
        super(CruftTestCase, self).setUp()
        self.install_date = self.now()
        self.setup_binaries()
        # flush to make sure that the setup is correct
        self.session.flush()

    def test_newer_version(self):
        "tests newer_version()"

        list = newer_version("squeeze", "sid", self.session)
        self.assertEqual([], list)
        self.file["sl_3.03-17.dsc"] = PoolFile(
            filename="main/s/sl/sl_3.03-17.dsc", filesize=0, md5sum=""
        )
        self.file["sl_3.03-17.dsc"].sha1sum = "sha1sum"
        self.file["sl_3.03-17.dsc"].sha256sum = "sha256sum"
        self.source["sl_3.03-17"] = DBSource(
            source="sl",
            version="3.03-17",
            maintainer=self.maintainer["maintainer"],
            changedby=self.maintainer["uploader"],
            poolfile=self.file["sl_3.03-17.dsc"],
            install_date=self.install_date,
        )
        self.source["sl_3.03-17"].suites.append(self.suite["squeeze"])
        list = newer_version("squeeze", "sid", self.session)
        self.assertEqual([("sl", "3.03-16", "3.03-17")], list)

    def test_multiple_source(self):
        "tests functions related to report_multiple_source()"

        # test get_package_names()
        suite = get_suite("sid", self.session)
        self.assertEqual([("gnome-hello",), ("hello",)], get_package_names(suite).all())
        # test class NamedSource
        src = NamedSource(suite, "hello")
        self.assertEqual("hello", src.source)
        self.assertEqual(["2.2-1", "2.2-2"], src.versions)
        self.assertEqual("hello(2.2-1, 2.2-2)", str(src))
        # test class DejavuBinary
        bin = DejavuBinary(suite, "hello")
        self.assertEqual(False, bin.has_multiple_sources())
        # add another binary
        self.file["hello_2.2-3"] = PoolFile(
            filename="main/s/sl/hello_2.2-3_i386.deb", filesize=0, md5sum=""
        )
        self.file["hello_2.2-3"].sha1sum = "sha1sum"
        self.file["hello_2.2-3"].sha256sum = "sha256sum"
        self.binary["hello_2.2-3_i386"] = DBBinary(
            package="hello",
            source=self.source["sl_3.03-16"],
            version="2.2-3",
            maintainer=self.maintainer["maintainer"],
            architecture=self.arch["i386"],
            poolfile=self.file["hello_2.2-3"],
        )
        self.session.add(self.binary["hello_2.2-3_i386"])
        bin = DejavuBinary(suite, "hello")
        self.assertEqual(False, bin.has_multiple_sources())
        # add it to suite sid
        self.binary["hello_2.2-3_i386"].suites.append(self.suite["sid"])
        bin = DejavuBinary(suite, "hello")
        self.assertEqual(True, bin.has_multiple_sources())
        self.assertEqual("hello built by: hello(2.2-1, 2.2-2), sl(3.03-16)", str(bin))


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