dak/tests/dbtest_debversion.py

#! /usr/bin/env python3

from db_test import DBDakTestCase

from daklib.dbconn import DebVersion

from sqlalchemy import Table, Column, Integer, func
from sqlalchemy.orm import mapper
import unittest


class Version:

    def __init__(self, version):
        self.version = version

    def __repr__(self):
        return "<Version('%s')>" % self.version


class DebVersionTestCase(DBDakTestCase):

    """
    The DebVersionTestCase tests both comparison (<=, ==, >, !=), the in_()
    method and aggregate functions (min, max) for the DebVersion type. To
    simplify the test it creates a separate table 'version' which is not used
    by dak itself.
    """

    def setUp(self):
        super(DebVersionTestCase, self).setUp()
        self.version_table = Table('version', self.metadata,
                                   Column('id', Integer, primary_key=True),
                                   Column('version', DebVersion),
                                   )
        self.version_table.create(checkfirst=True)
        mapper(Version, self.version_table)

    def test_debversion(self):
        v = Version('0.5~')
        self.session.add(v)
        v = Version('0.5')
        self.session.add(v)
        v = Version('1.0')
        self.session.add(v)
        q = self.session.query(Version)
        self.assertEqual(3, q.count())
        self.assertEqual(2, q.filter(Version.version <= '0.5').count())
        self.assertEqual(1, q.filter(Version.version == '0.5').count())
        self.assertEqual(2, q.filter(Version.version > '0.5~').count())
        self.assertEqual(1, q.filter(Version.version > '0.5').count())
        self.assertEqual(0, q.filter(Version.version > '1.0').count())
        self.assertEqual(2, q.filter(Version.version != '1.0').count())
        self.assertEqual(
            2, q.filter(Version.version.in_(['0.5~', '1.0'])).count())
        q = self.session.query(func.min(Version.version))
        self.assertEqual('0.5~', q.scalar())
        q = self.session.query(func.max(Version.version))
        self.assertEqual('1.0', q.scalar())

    def tearDown(self):
        self.session.rollback()
        self.version_table.drop()
        super(DebVersionTestCase, self).tearDown()


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