[dak/master] Refactor PackageTestCase.test_sources().

[Torsten Werner <twerner@debian.org>]

Signed-off-by: Torsten Werner <twerner@debian.org>
---
 tests/dbtest_packages.py |    9 +++------
 1 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py
index e972532..8764bc7 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__':
-- 
1.5.6.5