add a test for the new --version CLI

Author: martin-a-brown

tests/test_driver.py

@@ -12,6 +12,7 @@
 import errno
 import codecs
 import random
+import unittest
 from tempfile import NamedTemporaryFile as ntf
 from argparse import Namespace
 
@@ -20,6 +21,7 @@
 from tldp.inventory import stypes, status_types
 from tldp.sources import SourceDocument
 from tldp.outputs import OutputDirectory
+from tldp import VERSION
 
 # -- Test Data
 import example
@@ -127,6 +129,22 @@ def test_run_statustypes(self):
         self.assertEqual(exitcode, os.EX_OK)
 
 
+class TestDriverShowVersion(unittest.TestCase):
+
+    def test_show_version(self):
+        stdout = io.StringIO()
+        result = tldp.driver.show_version(Namespace(), file=stdout)
+        self.assertEqual(result, os.EX_OK)
+        stdout.seek(0)
+        data = stdout.read().strip()
+        for status in status_types:
+            self.assertEqual(VERSION, data)
+
+    def test_run_statustypes(self):
+        exitcode = tldp.driver.run(['--version'])
+        self.assertEqual(exitcode, os.EX_OK)
+
+
 class TestDriverSummary(TestInventoryBase):
 
     def test_run_summary(self):