--- a/test/test_listpackages.py	Thu Apr 09 18:59:44 2009 +0000
+++ b/test/test_listpackages.py	Tue Apr 14 09:41:59 2009 +0000
@@ -30,19 +30,29 @@
 
     package_dirs = []
     for revision in revisions:
-        rev, rules_rev = revision
         status = RevisionStatus(None)
-        status.binary_package_created()
+        if len(revision) == 2:
+            rev, rules_rev = revision
+            status.binary_package_created()
+        else:
+            rev, rules_rev, status_name = revision
+            getattr(status, status_name)()
+
         status.start = start_time
         status.stop = start_time + build_duration
         start_time += check_interval
 
+        if status.status.finished and not status.status.error:
+            src_files = [(template % (rev,), "") for template in src_templates]
+            bin_files = [(template % (rev,), "") for template in bin_templates]
+        else:
+            src_files = []
+            bin_files = []
+
         package_dirs.append(("%s-%s" % (rev, rules_rev),
                              [("status", "".join(status.serialize())),
-                              ("src", [(template % (rev,), "")
-                                       for template in src_templates]),
-                              ("binary", [(template % (rev,), "")
-                                          for template in bin_templates])]))
+                              ("src", src_files),
+                              ("binary", bin_files)]))
     return (track,
             [("pkg",
               package_dirs)])
@@ -162,6 +172,29 @@
 """)
 
 
+class TestListPackagesTrackNoRevisionWithIncompleteBuilds(ListPackagesTest):
+
+    track_files = [package_track_files("kdepim", "1.0.1",
+                                       ["kdepim", "kleopatra"],
+                                       [(702432, 47),
+                                        (704195, 31, "error")])]
+
+    def test_listpackages_source(self):
+        self.check_cmd(["--track=kdepim", "--source"],
+                       """\
+kdepim/pkg/702432-47/src/kdepim_1.0.1.702432-1.diff.gz
+kdepim/pkg/702432-47/src/kdepim_1.0.1.702432-1.dsc
+kdepim/pkg/702432-47/src/kdepim_1.0.1.702432.orig.tar.gz
+""")
+
+    def test_listpackages_binary(self):
+        self.check_cmd(["--track=kdepim", "--binary"],
+                       """\
+kdepim/pkg/702432-47/binary/kdepim_1.0.1.702432-1_i386.deb
+kdepim/pkg/702432-47/binary/kleopatra_1.0.1.702432-1_i386.deb
+""")
+
+
 class TestListPackagesNoTrackNoRevision(ListPackagesTest):
 
     track_files = [package_track_files("kdepim", "1.0.1",