Rev 212: Extracted assertQueueLength in tests. in http://bzr.daniel-watkins.co.uk/pqm/xmlrpc

[Daniel Watkins]

At http://bzr.daniel-watkins.co.uk/pqm/xmlrpc

------------------------------------------------------------
revno: 212
revision-id: daniel at daniel-watkins.co.uk-20080807021639-gzmjwj24mjomrvhm
parent: daniel at daniel-watkins.co.uk-20080807021129-0tft5ozta0y3qhpi
committer: Daniel Watkins <daniel at daniel-watkins.co.uk>
branch nick: xmlrpc
timestamp: Thu 2008-08-07 03:16:39 +0100
message:
  Extracted assertQueueLength in tests.
-------------- next part --------------
=== modified file 'pqm/ui/tests/test_xmlrpc.py'
--- a/pqm/ui/tests/test_xmlrpc.py	2008-08-07 02:11:29 +0000
+++ b/pqm/ui/tests/test_xmlrpc.py	2008-08-07 02:16:39 +0000
@@ -24,6 +24,11 @@
 
 class TestPQM_XMLRPC(TestCaseWithQueue):
 
+    def assertQueueLength(self, queuedir, length):
+        patches = [f for f in os.listdir(queuedir) if f.startswith('patch.')]
+        self.assertEqual(1, len(patches))
+        return patches
+
     def get_xmlrpc(self, empty=True, verify_sigs=False):
         queue = self.getQueue(empty=empty, verify_sigs=verify_sigs)
         pqminfo = PQMInfo([queue.configFileName])
@@ -41,7 +46,6 @@
         out = xmlrpc.xmlrpc_submit(sample_message)
         self.assertEqual('Success!', out)
         queuedir = xmlrpc.pqminfo.queuedir
-        patches = [f for f in os.listdir(queuedir) if f.startswith('patch.')]
-        self.assertEqual(1, len(patches)) # We should have only one patch
+        patches = self.assertQueueLength(queuedir, 1)
         self.assertFileEqual(sample_message,
                              os.path.join(queuedir, patches[0]))