test ArgumentTypeError

mushorg · Mar 13, 2019 · 08d12f7 · 08d12f7
1 parent a51448f
commit 08d12f7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/snare/tests/test_parse_timeout.py b/snare/tests/test_parse_timeout.py
@@ -1,4 +1,6 @@
 import unittest
+from argparse import ArgumentTypeError
+
 from snare.utils.snare_helpers import parse_timeout, str_to_bool
 
 
@@ -21,3 +23,6 @@ def test_str_to_bool(self):
         assert str_to_bool(self.v) is True
         self.v = 'false'
         assert str_to_bool(self.v) is False
+
+        with self.assertRaises(ArgumentTypeError):
+            str_to_bool('twz')