X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_packages.py;h=5f5b89c6a536c34dfbcb1d8ca14947d8b8227256;hb=2918ca9e2fa7db73192a6ad22c1dd405204e2386;hp=fabb4152ff1672680e0e4d6a399cf70d5fa3cc1c;hpb=9701189232ab74d2057abe9e59afab71530d7820;p=dak.git

diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py
index fabb4152..5f5b89c6 100755
--- a/tests/dbtest_packages.py
+++ b/tests/dbtest_packages.py
@@ -7,6 +7,7 @@ from daklib.dbconn import Architecture, Suite, get_suite_architectures, \
     check_poolfile, get_poolfile_like_name, get_source_in_suite, \
     get_suites_source_in, add_dsc_to_db, source_exists
 from daklib.queue_install import package_to_suite
+from daklib.queue import get_newest_source
 
 from sqlalchemy.orm.exc import MultipleResultsFound
 import unittest
@@ -348,7 +349,6 @@ class PackageTestCase(DBDakTestCase):
         'test function package_to_suite()'
 
         self.setup_sources()
-        self.session.flush()
         pkg = Pkg()
         pkg.changes = { 'distribution': {} }
         upload = Upload(pkg)
@@ -368,6 +368,16 @@ class PackageTestCase(DBDakTestCase):
         pkg.changes['distribution'] = { 'lenny': '' }
         self.assertTrue(package_to_suite(upload, 'lenny', self.session))
 
+    def test_get_newest_source(self):
+        'test function get_newest_source'
+
+        self.setup_sources()
+        self.session.flush()
+        import daklib.queue
+        daklib.queue.dm_suites = ['sid']
+        self.assertEqual(self.source['hello'], get_newest_source('hello', self.session))
+        self.assertEqual(None, get_newest_source('foobar', self.session))
+
 
 if __name__ == '__main__':
     unittest.main()