improved test coverage for command.bdist

1 files changed, 13 insertions, 2 deletions

diff --git a/distutils2/tests/test_command_bdist.py b/distutils2/tests/test_command_bdist.py
index 1b447ae..c38a76b 100644
--- a/distutils2/tests/test_command_bdist.py
+++ b/distutils2/tests/test_command_bdist.py
@@ -2,8 +2,9 @@
 
 from distutils2.tests import run_unittest
 
-from distutils2.command.bdist import bdist
-from distutils2.tests import unittest, support
+from distutils2.command.bdist import bdist, show_formats
+from distutils2.tests import unittest, support, captured_stdout
+
 
 class BuildTestCase(support.TempdirManager,
                     unittest.TestCase):
@@ -28,6 +29,16 @@ class BuildTestCase(support.TempdirManager,
         found.sort()
         self.assertEqual(found, formats)
 
+    def test_show_formats(self):
+        __, stdout = captured_stdout(show_formats)
+
+        # the output should be a header line + one line per format
+        num_formats = len(bdist.format_commands)
+        output = [line for line in stdout.split('\n')
+                  if line.strip().startswith('--formats=')]
+        self.assertEqual(len(output), num_formats)
+
+
 def test_suite():
     return unittest.makeSuite(BuildTestCase)