X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_packages.py;h=97cc8b10cacd497c9968b4dff70a15c70c8b9a3e;hb=99626b2f8ea34520cb7fe1aa20f665a8305224fc;hp=f1ac75b89b1c35c495648c50be03adf991168767;hpb=073da36122b23d7bc96b8d81a8da52d00d034da7;p=dak.git

diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py
index f1ac75b8..97cc8b10 100755
--- a/tests/dbtest_packages.py
+++ b/tests/dbtest_packages.py
@@ -2,7 +2,8 @@
 
 from db_test import DBDakTestCase
 
-from daklib.dbconn import Architecture, Suite, get_suite_architectures
+from daklib.dbconn import Architecture, Suite, get_suite_architectures, \
+    get_architecture_suites
 
 import unittest
 
@@ -81,6 +82,13 @@ class PackageTestCase(DBDakTestCase):
         architectures = get_suite_architectures('lenny', skipall = True, session = self.session)
         self.assertEqual(3, len(architectures))
         self.assertTrue(self.arch['all'] not in architectures)
+        # check the function get_architecture_suites()
+        suites = get_architecture_suites('i386', self.session)
+        self.assertEqual(3, len(suites))
+        self.assertTrue(self.suite['lenny'] in suites)
+        suites = get_architecture_suites('kfreebsd-i386', self.session)
+        self.assertEqual(2, len(suites))
+        self.assertTrue(self.suite['lenny'] not in suites)
 
 if __name__ == '__main__':
     unittest.main()