github/webssh
1
0
Fork 0
mirror of https://github.com/huashengdun/webssh.git synced 2026-05-20 08:13:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added an option for blocking public non-https requests

Browse Source
This commit is contained in:
Sheng
2018-10-17 22:39:53 +08:00
parent 746982b001
commit c06bf5311a
7 changed files with 115 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tests/test_utils.py
View File

@@ -1,8 +1,9 @@
import unittest
from webssh.utils import (
is_valid_ip_address, is_valid_port, is_valid_hostname,
to_str, to_bytes, to_int
is_valid_ip_address, is_valid_port, is_valid_hostname, to_str, to_bytes,
to_int, on_public_network_interface, on_public_network_interfaces,
get_ips_by_name
)
@@ -51,3 +52,25 @@ class TestUitls(unittest.TestCase):
self.assertFalse(is_valid_hostname('https://www.google.com'))
self.assertFalse(is_valid_hostname('127.0.0.1'))
self.assertFalse(is_valid_hostname('::1'))
def test_get_ips_by_name(self):
self.assertTrue(get_ips_by_name(''), {'0.0.0.0', '::'})
self.assertTrue(get_ips_by_name('localhost'), {'127.0.0.1'})
self.assertTrue(get_ips_by_name('192.68.1.1'), {'192.168.1.1'})
self.assertTrue(get_ips_by_name('2.2.2.2'), {'2.2.2.2'})
def test_on_public_network_interface(self):
self.assertTrue(on_public_network_interface('0.0.0.0'))
self.assertTrue(on_public_network_interface('::'))
self.assertTrue(on_public_network_interface('0:0:0:0:0:0:0:0'))
self.assertTrue(on_public_network_interface('2.2.2.2'))
self.assertTrue(on_public_network_interface('2:2:2:2:2:2:2:2'))
self.assertIsNone(on_public_network_interface('127.0.0.1'))
def test_on_public_network_interfaces(self):
self.assertTrue(
on_public_network_interfaces(['0.0.0.0', '127.0.0.1'])
)
self.assertIsNone(
on_public_network_interfaces(['192.168.1.1', '127.0.0.1'])
)