Added to_int to utils

This commit is contained in:
Sheng
2018-10-10 08:49:59 +08:00
parent de0fda1ae1
commit 88405eddac
3 changed files with 24 additions and 18 deletions
+10 -2
View File
@@ -1,7 +1,9 @@
import unittest
from webssh.utils import (is_valid_ipv4_address, is_valid_ipv6_address,
is_valid_port, is_valid_hostname, to_str, to_bytes)
from webssh.utils import (
is_valid_ipv4_address, is_valid_ipv6_address, is_valid_port,
is_valid_hostname, to_str, to_bytes, to_int
)
class TestUitls(unittest.TestCase):
@@ -18,6 +20,12 @@ class TestUitls(unittest.TestCase):
self.assertEqual(to_bytes(b), b)
self.assertEqual(to_bytes(u), b)
def test_to_int(self):
self.assertEqual(to_int(''), None)
self.assertEqual(to_int(None), None)
self.assertEqual(to_int('22'), 22)
self.assertEqual(to_int(' 22 '), 22)
def test_is_valid_ipv4_address(self):
self.assertFalse(is_valid_ipv4_address('127.0.0'))
self.assertFalse(is_valid_ipv4_address(b'127.0.0'))