tests/dbtest_packages.py

   1 #!/usr/bin/env python
   2
   3 from db_test import DBDakTestCase
   4
   5 from daklib.dbconn import Architecture, Suite, get_suite_architectures
   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', 'kfreebsd-i386'):
  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 connect_suite_architectures(self):
  46         """
  47         Gonnect all suites and all architectures except for kfreebsd-i386 which
  48         should not be in lenny.
  49         """
  50
  51         for arch_string, architecture in self.arch.items():
  52             if arch_string != 'kfreebsd-i386':
  53                 architecture.suites = self.suite.values()
  54             else:
  55                 architecture.suites = [self.suite['squeeze'], self.suite['sid']]
  56
  57     def test_suite_architecture(self):
  58         # check the id for architectures source and all
  59         self.assertEqual(1, self.arch['source'].arch_id)
  60         self.assertEqual(2, self.arch['all'].arch_id)
  61         # check the many to many relation between Suite and Architecture
  62         self.arch['source'].suites.append(self.suite['lenny'])
  63         self.assertEqual('source', self.suite['lenny'].architectures[0])
  64         self.arch['source'].suites = []
  65         self.assertEqual([], self.suite['lenny'].architectures)
  66         self.connect_suite_architectures()
  67         self.assertEqual(4, len(self.suite['lenny'].architectures))
  68         self.assertEqual(3, len(self.arch['i386'].suites))
  69         # check the function get_suite_architectures()
  70         architectures = get_suite_architectures('lenny', session = self.session)
  71         self.assertEqual(4, len(architectures))
  72         self.assertTrue(self.arch['source'] in architectures)
  73         self.assertTrue(self.arch['all'] in architectures)
  74         self.assertTrue(self.arch['kfreebsd-i386'] not in architectures)
  75         architectures = get_suite_architectures('sid', session = self.session)
  76         self.assertEqual(5, len(architectures))
  77         self.assertTrue(self.arch['kfreebsd-i386'] in architectures)
  78         architectures = get_suite_architectures('lenny', skipsrc = True, session = self.session)
  79         self.assertEqual(3, len(architectures))
  80         self.assertTrue(self.arch['source'] not in architectures)
  81         architectures = get_suite_architectures('lenny', skipall = True, session = self.session)
  82         self.assertEqual(3, len(architectures))
  83         self.assertTrue(self.arch['all'] not in architectures)
  84
  85 if __name__ == '__main__':
  86     unittest.main()