Last changes for release 0.1.6.

ralph089 · May 6, 2016 · 26577f6 · 26577f6
1 parent ab814e1
commit 26577f6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/check_pa/check_paloalto.py b/check_pa/check_paloalto.py
@@ -78,7 +78,7 @@ def parse_args(args):
 
     info = parser.add_argument_group('Info')
     info.add_argument('--version', action='version',
-                      version='%(prog)s ' + pkg_resources.require("check_paloalto")[0].version)
+                      version='%(prog)s 0.1.6')
 
     subparsers = parser.add_subparsers(dest='command')
     subparsers.required = True

diff --git a/tests/test_check_paloalto.py b/tests/test_check_paloalto.py
@@ -7,11 +7,10 @@
 
 Tests for `check_paloalto` module.
 """
-import pytest
 import responses
 
 from check_pa.check_paloalto import parse_args, _diskspace, _certificates, \
-    _environmental, _load, _sessinfo, _thermal, _throughput
+    _environmental, _load, _sessinfo, _thermal, _throughput, _useragent
 from tests.conftest import read_xml
 
 
@@ -35,6 +34,12 @@ def test_arg_diskspace(self):
         assert args.token == self.token
         assert args.func.__name__ == _diskspace.__name__
 
+    def test_arg_useragent(self):
+        args = parse_args(['-H', self.host, '-T', self.token, 'useragent'])
+        assert args.host == self.host
+        assert args.token == self.token
+        assert args.func.__name__ == _useragent.__name__
+
     def test_arg_certificates(self):
         args = parse_args(
             ['-H', self.host, '-T', self.token, 'certificates', '-ex',