tests/dbtest_packages.py

   1 #!/usr/bin/env python
   2
   3 from db_test import DBDakTestCase
   4
   5 from daklib.dbconn import Architecture, Suite
   6
   7 import unittest
   8
   9 class PackageTestCase(DBDakTestCase):
  10     """
  11     PackageTestCase checks the handling of source and binary packages in dak's
  12     database.
  13     """
  14
  15     def setup_architectures(self):
  16         "setup a hash of Architecture objects in self.arch"
  17
  18         self.arch = {}
  19         for arch_string in ('source', 'all', 'i386', 'amd64'):
  20             self.arch[arch_string] = Architecture(arch_string)
  21         # hard code ids for source and all
  22         self.arch['source'].arch_id = 1
  23         self.arch['all'].arch_id = 2
  24         for _, architecture in self.arch.items():
  25             self.session.add(architecture)
  26             self.session.flush()
  27             self.session.refresh(architecture)
  28
  29     def setup_suites(self):
  30         "setup a hash of Suite objects in self.suite"
  31
  32         self.suite = {}
  33         for suite_name in ('lenny', 'squeeze', 'sid'):
  34             suite = Suite(suite_name = suite_name, version = '-')
  35             self.suite[suite_name] = suite
  36             self.session.add(suite)
  37             self.session.flush()
  38             self.session.refresh(suite)
  39
  40     def setUp(self):
  41         super(PackageTestCase, self).setUp()
  42         self.setup_architectures()
  43         self.setup_suites()
  44
  45     def test_packages(self):
  46         # check the id for architectures source and all
  47         self.assertEqual(1, self.arch['source'].arch_id)
  48         self.assertEqual(2, self.arch['all'].arch_id)
  49         # check the many to many relation between Suite and Architecture
  50         self.arch['source'].suites.append(self.suite['lenny'])
  51         self.assertEqual('source', self.suite['lenny'].architectures[0])
  52
  53 if __name__ == '__main__':
  54     unittest.main()