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     xxx
  12     """
  13
  14     def setup_architectures(self):
  15         "setup a hash of Architecture objects in self.arch"
  16
  17         self.arch = {}
  18         for arch_string in ('source', 'all', 'i386', 'amd64'):
  19             self.arch[arch_string] = Architecture(arch_string)
  20         # hard code ids for source and all
  21         self.arch['source'].arch_id = 1
  22         self.arch['all'].arch_id = 2
  23         for _, architecture in self.arch.items():
  24             self.session.add(architecture)
  25             self.session.flush()
  26             self.session.refresh(architecture)
  27
  28     def setUp(self):
  29         super(PackageTestCase, self).setUp()
  30         self.setup_architectures()
  31
  32     def test_packages(self):
  33         self.assertEqual(1, self.arch['source'].arch_id)
  34         self.assertEqual(2, self.arch['all'].arch_id)
  35
  36 if __name__ == '__main__':
  37     unittest.main()