add a test case for when user specifies only stem as arg

tests/test_driver.py

@@ -157,6 +157,7 @@ class TestDriverProcessSkips(TestInventoryBase):
         self.assertEquals(excluded.stem, 'Docbook4XML-HOWTO')
         self.assertEquals(len(inc) + 1, len(inv.all.keys()))
 
+
 @unittest.skip("Except when you want to spend time....")
 class TestDriverBuild(TestInventoryBase):
 
@@ -196,7 +197,6 @@ class TestDriverBuild(TestInventoryBase):
         doc = docs.pop(0)
         self.assertTrue(doc.output.iscomplete)
 
-
     def test_build_one_broken(self):
         c = self.config
         self.add_new('Frobnitz-DocBook-XML-4-HOWTO', example.ex_docbook4xml)
@@ -213,5 +213,17 @@ class TestDriverBuild(TestInventoryBase):
         result = tldp.driver.build(c, docs)
         self.assertEquals(1, result)
 
+    def test_build_only_requested_stem(self):
+        c = self.config
+        ex = example.ex_linuxdoc
+        self.add_published('Published-HOWTO', ex)
+        self.add_new('New-HOWTO', ex)
+        argv = ['--pubdir', c.pubdir, '--sourcedir', c.sourcedir[0]]
+        argv.extend(['--build', 'Published-HOWTO'])
+        tldp.driver.run(argv)
+        inv = tldp.inventory.Inventory(c.pubdir, c.sourcedir)
+        self.assertEquals(1, len(inv.published.keys()))
+        self.assertEquals(1, len(inv.work.keys()))
+
 #
 # -- end of file

tldp/driver.py

@@ -261,7 +261,7 @@ def run(argv):
 
     unknownargs = None
     if remainder:
-        docs, unknownargs = getDocumentsByStems(inv.work.values(), remainder)
+        docs, unknownargs = getDocumentsByStems(inv.all.values(), remainder)
         workset.update(docs)
         logger.info("Added %d docs, found by stem name.", len(docs))