X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_packages.py;h=249ad81eabf4c96d7e21df93966e863ac2791c66;hb=2f142b2bf324aba8fcc74cd1aef5de6fa5b9e49c;hp=deca4812d3bc08ac6a158f30ad1f100432103ba4;hpb=d6fb77d2760192195245e87d6f60a418c465d0b1;p=dak.git diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py index deca4812..249ad81e 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 + get_architecture_suites, Maintainer, DBSource, Location, PoolFile, \ + check_poolfile, get_poolfile_like_name import unittest @@ -90,19 +91,100 @@ class PackageTestCase(DBDakTestCase): self.assertEqual(2, len(suites)) self.assertTrue(self.suite['lenny'] not in suites) + def setup_locations(self): + 'create some Location objects, TODO: add component' + + self.loc = {} + self.loc['main'] = Location(path = \ + '/srv/ftp-master.debian.org/ftp/pool/') + self.session.add(self.loc['main']) + + 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 = '') + self.file['sl'] = PoolFile(filename = 'main/s/sl/sl_3.03-16.dsc', \ + location = self.loc['main'], filesize = 0, md5sum = '') + self.session.add_all(self.file.values()) + + def test_poolfiles(self): + ''' + Test the relation of the classes PoolFile and Location. + + The code needs some explaination. The property Location.files is not a + list as in other relations because such a list would become rather + huge. It is a query object that can be queried, filtered, and iterated + as usual. But list like methods like append() and remove() are + supported as well which allows code like: + + somelocation.files.append(somefile) + ''' + + self.setup_poolfiles() + location = self.session.query(Location)[0] + self.assertEqual('/srv/ftp-master.debian.org/ftp/pool/', location.path) + self.assertEqual(2, location.files.count()) + poolfile = location.files. \ + 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']) + self.session.refresh(location) + self.assertEqual(1, location.files.count()) + # please note that we intentionally do not specify 'location' here + self.file['sl'] = PoolFile(filename = 'main/s/sl/sl_3.03-16.dsc', \ + filesize = 0, md5sum = '') + 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()) + self.session.flush() 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_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): ''' @@ -112,20 +194,22 @@ class PackageTestCase(DBDakTestCase): ''' 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) + 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.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.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, []) if __name__ == '__main__':