X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_packages.py;h=8764bc78a4139152d790baf867cf1835711f9107;hb=1d3791dfe0910312c0e192f1cd81c2f55690b8f6;hp=e97253213c38ad8ea4366967a0f1b74329e72659;hpb=fa9204359367665156ba36756acda4095864016a;p=dak.git

diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py
index e9725321..8764bc78 100755
--- a/tests/dbtest_packages.py
+++ b/tests/dbtest_packages.py
@@ -209,12 +209,9 @@ class PackageTestCase(DBDakTestCase):
         '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)
+        self.assertEqual(self.file['hello'], self.source.poolfile)
+        self.assertEqual(self.source, self.file['hello'].source)
+        self.assertEqual(None, self.file['sl'].source)
 
 
 if __name__ == '__main__':