Added a verbosity option and banner to osmotestvty

2 files changed, 10 insertions, 1 deletions

diff --git a/osmopy/.gitignore b/osmopy/.gitignore
new file mode 100644
index 0000000..c9b568f
--- /dev/null
+++ b/osmopy/.gitignore
@@ -0,0 +1,2 @@
+*.pyc
+*.swp
diff --git a/osmopy/osmotestvty.py b/osmopy/osmotestvty.py
index 2ee877a..0f685d6 100755
--- a/osmopy/osmotestvty.py
+++ b/osmopy/osmotestvty.py
@@ -65,12 +65,18 @@ if __name__ == '__main__':
     confpath = "."
 
     parser = argparse.ArgumentParser()
+    parser.add_argument("-v", "--verbose", dest="verbose",
+                        action="store_true", help="verbose mode")
     parser.add_argument("-p", "--pythonconfpath", dest="p",
                         help="searchpath for config")
     parser.add_argument("-w", "--workdir", dest="w",
                         help="Working directory to run in")
     args = parser.parse_args()
 
+    verbose_level = 1
+    if args.verbose:
+        verbose_level = 2
+
     if args.w:
         workdir = args.w
 
@@ -84,6 +90,7 @@ if __name__ == '__main__':
         sys.exit(1)
 
     os.chdir(workdir)
+    print "Running tests for specific VTY commands"
     suite = unittest.TestLoader().loadTestsFromTestCase(TestVTY)
-    res = unittest.TextTestRunner(verbosity=1).run(suite)
+    res = unittest.TextTestRunner(verbosity=verbose_level).run(suite)
     sys.exit(len(res.errors) + len(res.failures))