X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_packages.py;h=e97253213c38ad8ea4366967a0f1b74329e72659;hb=fa9204359367665156ba36756acda4095864016a;hp=439c9a2e0f49382debb375cb41a6d96d811f3677;hpb=22b0b5b1963b6c298f6f619ad7cbbd262bf9ea77;p=dak.git diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py index 439c9a2e..e9725321 100755 --- a/tests/dbtest_packages.py +++ b/tests/dbtest_packages.py @@ -3,7 +3,8 @@ from db_test import DBDakTestCase from daklib.dbconn import Architecture, Suite, get_suite_architectures, \ - get_architecture_suites, Maintainer, DBSource, Location, PoolFile + get_architecture_suites, Maintainer, DBSource, Location, PoolFile, \ + check_poolfile, get_poolfile_like_name import unittest @@ -22,21 +23,15 @@ class PackageTestCase(DBDakTestCase): # hard code ids for source and all self.arch['source'].arch_id = 1 self.arch['all'].arch_id = 2 - for _, architecture in self.arch.items(): - self.session.add(architecture) - self.session.flush() - self.session.refresh(architecture) + self.session.add_all(self.arch.values()) def setup_suites(self): "setup a hash of Suite objects in self.suite" self.suite = {} for suite_name in ('lenny', 'squeeze', 'sid'): - suite = Suite(suite_name = suite_name, version = '-') - self.suite[suite_name] = suite - self.session.add(suite) - self.session.flush() - self.session.refresh(suite) + self.suite[suite_name] = Suite(suite_name = suite_name, version = '-') + self.session.add_all(self.suite.values()) def setUp(self): super(PackageTestCase, self).setUp() @@ -101,6 +96,7 @@ class PackageTestCase(DBDakTestCase): def setup_poolfiles(self): 'create some PoolFile objects' + self.setup_locations() self.file = {} self.file['hello'] = PoolFile(filename = 'main/h/hello/hello_2.2-2.dsc', \ location = self.loc['main'], filesize = 0, md5sum = '') @@ -120,7 +116,7 @@ class PackageTestCase(DBDakTestCase): somelocation.files.append(somefile) ''' - self.setup_locations() + self.setup_poolfiles() location = self.session.query(Location)[0] self.assertEqual('/srv/ftp-master.debian.org/ftp/pool/', location.path) @@ -129,6 +125,11 @@ class PackageTestCase(DBDakTestCase): filter(PoolFile.filename.like('%/hello/hello%')).one() self.assertEqual('main/h/hello/hello_2.2-2.dsc', poolfile.filename) self.assertEqual(location, poolfile.location) + # test get() + self.assertEqual(poolfile, \ + self.session.query(PoolFile).get(poolfile.file_id)) + self.assertEqual(None, self.session.query(PoolFile).get(-1)) + # test remove() and append() location.files.remove(self.file['sl']) # TODO: deletion should cascade automatically self.session.delete(self.file['sl']) @@ -140,20 +141,44 @@ class PackageTestCase(DBDakTestCase): location.files.append(self.file['sl']) self.session.refresh(location) self.assertEqual(2, location.files.count()) + # test fullpath + self.assertEqual('/srv/ftp-master.debian.org/ftp/pool/main/s/sl/sl_3.03-16.dsc', \ + self.file['sl'].fullpath) + # test check_poolfile() + self.assertEqual((True, self.file['sl']), \ + check_poolfile('main/s/sl/sl_3.03-16.dsc', 0, '', \ + location.location_id, self.session)) + self.assertEqual((False, None), \ + check_poolfile('foobar', 0, '', location.location_id, self.session)) + self.assertEqual((False, self.file['sl']), \ + check_poolfile('main/s/sl/sl_3.03-16.dsc', 42, '', \ + location.location_id, self.session)) + self.assertEqual((False, self.file['sl']), \ + check_poolfile('main/s/sl/sl_3.03-16.dsc', 0, 'deadbeef', \ + location.location_id, self.session)) + # test get_poolfile_like_name() + self.assertEqual([self.file['sl']], \ + get_poolfile_like_name('sl_3.03-16.dsc', self.session)) + self.assertEqual([], get_poolfile_like_name('foobar', self.session)) def setup_maintainers(self): 'create some Maintainer objects' - self.maintainer = Maintainer(name = 'Mr. Maintainer') - self.uploader = Maintainer(name = 'Mrs. Uploader') - self.lazyguy = Maintainer(name = 'Lazy Guy') - self.session.add_all([self.maintainer, self.uploader, self.lazyguy]) + self.maintainer = {} + self.maintainer['maintainer'] = Maintainer(name = 'Mr. Maintainer') + self.maintainer['uploader'] = Maintainer(name = 'Mrs. Uploader') + self.maintainer['lazyguy'] = Maintainer(name = 'Lazy Guy') + self.session.add_all(self.maintainer.values()) def setup_sources(self): 'create a DBSource object; but it cannot be stored in the DB yet' - self.source = DBSource(maintainer = self.maintainer, - changedby = self.uploader) + self.setup_maintainers() + self.setup_poolfiles() + self.source = DBSource(source = 'hello', version = '2.2-2', \ + maintainer = self.maintainer['maintainer'], \ + changedby = self.maintainer['uploader'], \ + poolfile = self.file['hello'], install_date = self.now()) def test_maintainers(self): ''' @@ -162,21 +187,34 @@ class PackageTestCase(DBDakTestCase): TODO: add relations to changes_pending_source ''' - self.setup_maintainers() - self.assertEqual('Mr. Maintainer', - self.session.query(Maintainer)[0].name) - self.assertEqual('Mrs. Uploader', - self.session.query(Maintainer)[1].name) - self.assertEqual('Lazy Guy', - self.session.query(Maintainer)[2].name) self.setup_sources() - #TODO: needs File and Location - #self.assertEqual(self.maintainer.maintains_sources, [self.source]) - #self.assertEqual(self.maintainer.changed_sources, []) - #self.assertEqual(self.uploader.maintains_sources, []) - #self.assertEqual(self.uploader.changed_sources, [self.source]) - #self.assertEqual(self.lazyguy.maintains_sources, []) - #self.assertEqual(self.lazyguy.changed_sources, []) + self.session.flush() + maintainer = self.maintainer['maintainer'] + self.assertEqual(maintainer, + self.session.query(Maintainer).get(maintainer.maintainer_id)) + uploader = self.maintainer['uploader'] + self.assertEqual(uploader, + self.session.query(Maintainer).get(uploader.maintainer_id)) + lazyguy = self.maintainer['lazyguy'] + self.assertEqual(lazyguy, + self.session.query(Maintainer).get(lazyguy.maintainer_id)) + self.assertEqual(maintainer.maintains_sources, [self.source]) + self.assertEqual(maintainer.changed_sources, []) + self.assertEqual(uploader.maintains_sources, []) + self.assertEqual(uploader.changed_sources, [self.source]) + self.assertEqual(lazyguy.maintains_sources, []) + self.assertEqual(lazyguy.changed_sources, []) + + def test_sources(self): + 'test relation between DBSource and PoolFile' + + self.setup_sources() + poolfile_hello = self.session.query(DBSource)[0].poolfile + self.assertEqual(self.file['hello'], poolfile_hello) + self.assertEqual(self.source, poolfile_hello.source) + poolfile_sl = self.session.query(PoolFile). \ + filter(PoolFile.filename.like('%/sl/%'))[0] + self.assertEqual(None, poolfile_sl.source) if __name__ == '__main__':