Alter generate_reject_messages to return a generator.

[dak.git] / tests / test_lintian.py

diff --git a/tests/test_lintian.py b/tests/test_lintian.py
index 27d13680bf8855bae44b0e9a11e09ddafb334b71..4a574a9d1a5c5fdba63d5ff3136b5a3481e19f61 100755 (executable)
--- a/tests/test_lintian.py
+++ b/tests/test_lintian.py
@@ -43,7 +43,8 @@ class ParseLintianTestCase(DakTestCase):
 
 class GenerateRejectMessages(DakTestCase):
     def assertNumReject(self, input, defs, num):
-        self.assertEqual(len(generate_reject_messages(input, defs)), num)
+        msgs = list(generate_reject_messages(input, defs))
+        self.assertEqual(len(msgs), num)
 
     def testUnknownTag(self):
         self.assertNumReject(