Extend the testcase for check_poolfile().

[dak.git] / tests / dbtest_packages.py

diff --git a/tests/dbtest_packages.py b/tests/dbtest_packages.py
index f4749aca0c8f7a4396ff985568961d4ca97323f8..54aa8476c16ecf14d2dc193a2be8fda23743a339 100755 (executable)
--- a/tests/dbtest_packages.py
+++ b/tests/dbtest_packages.py
@@ -279,6 +279,10 @@ class PackageTestCase(DBDakTestCase):
         self.assertEqual((True, self.file['sl_3.03-16.dsc']), \
             check_poolfile('main/s/sl/sl_3.03-16.dsc', 0, '', \
                 contrib.location_id, self.session))
+        # test string value of 2nd argument
+        self.assertEqual((True, self.file['sl_3.03-16.dsc']), \
+            check_poolfile('main/s/sl/sl_3.03-16.dsc', '0', '', \
+                contrib.location_id, self.session))
         self.assertEqual((False, None), \
             check_poolfile('foobar', 0, '', contrib.location_id, self.session))
         self.assertEqual((False, self.file['sl_3.03-16.dsc']), \
@@ -544,17 +548,29 @@ class PackageTestCase(DBDakTestCase):
     def test_get_component_by_package_suite(self):
         'test get_component_by_package_suite()'
 
-        result = get_component_by_package_suite('hello', ['sid'], self.session)
+        result = get_component_by_package_suite('hello', ['sid'], \
+            session = self.session)
         self.assertEqual('main', result)
-        result = get_component_by_package_suite('hello', ['hamm'], self.session)
+        result = get_component_by_package_suite('hello', ['hamm'], \
+            session = self.session)
         self.assertEqual(None, result)
-        result = get_component_by_package_suite('foobar', ['sid'], self.session)
+        result = get_component_by_package_suite('foobar', ['sid'], \
+            session = self.session)
         self.assertEqual(None, result)
         # test that the newest version is returend
-        result = get_component_by_package_suite('gnome-hello', ['squeeze'], self.session)
+        result = get_component_by_package_suite('gnome-hello', ['squeeze'], \
+            session = self.session)
         self.assertEqual('main', result)
-        result = get_component_by_package_suite('gnome-hello', ['sid'], self.session)
+        result = get_component_by_package_suite('gnome-hello', ['sid'], \
+            session = self.session)
         self.assertEqual('contrib', result)
+        # test arch_list
+        result = get_component_by_package_suite('hello', ['sid'], \
+            arch_list = ['i386'], session = self.session)
+        self.assertEqual('main', result)
+        result = get_component_by_package_suite('hello', ['sid'], \
+            arch_list = ['amd64'], session = self.session)
+        self.assertEqual(None, result)
 
 if __name__ == '__main__':
     unittest.main()