Merge "Handle ints passed to `boolean_from_string`"

1 files changed, 12 insertions, 0 deletions

diff --git a/tests/unit/test_strutils.py b/tests/unit/test_strutils.py
index 1863c40..b0af958 100644
--- a/tests/unit/test_strutils.py
+++ b/tests/unit/test_strutils.py
@@ -48,6 +48,10 @@ class StrUtilsTest(utils.BaseTestCase):
         self.assertFalse(strutils.bool_from_string(c(
                          'This should not be True')))
 
+        # Whitespace should be stripped
+        self.assertTrue(strutils.bool_from_string(c(' true ')))
+        self.assertFalse(strutils.bool_from_string(c(' false ')))
+
     def test_bool_from_string(self):
         self._test_bool_from_string(lambda s: s)
 
@@ -55,8 +59,16 @@ class StrUtilsTest(utils.BaseTestCase):
         self._test_bool_from_string(six.text_type)
 
     def test_other_bool_from_string(self):
+        self.assertFalse(strutils.bool_from_string(None))
         self.assertFalse(strutils.bool_from_string(mock.Mock()))
 
+    def test_int_bool_from_string(self):
+        self.assertTrue(strutils.bool_from_string(1))
+
+        self.assertFalse(strutils.bool_from_string(-1))
+        self.assertFalse(strutils.bool_from_string(0))
+        self.assertFalse(strutils.bool_from_string(2))
+
     def test_int_from_bool_as_string(self):
         self.assertEqual(1, strutils.int_from_bool_as_string(True))
         self.assertEqual(0, strutils.int_from_bool_as_string(False))