3 from db_test import DBDakTestCase
5 from daklib.dbconn import Architecture, Suite
9 class PackageTestCase(DBDakTestCase):
14 def setup_architectures(self):
15 "setup a hash of Architecture objects in 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)
26 self.session.refresh(architecture)
29 super(PackageTestCase, self).setUp()
30 self.setup_architectures()
32 def test_packages(self):
33 self.assertEqual(1, self.arch['source'].arch_id)
34 self.assertEqual(2, self.arch['all'].arch_id)
36 if __name__ == '__main__':