jocay/webssh
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactored handler.py

Browse Source
This commit is contained in:
Sheng
2018-10-19 18:18:55 +08:00
parent fbb3e466b2
commit 1f835f5a70
2 changed files with 15 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_handler.py
+8 -8
View File
@@ -18,33 +18,33 @@ class TestMixinHandler(unittest.TestCase):
handler = MixinHandler()
options.fbidhttp = True
handler.context = Mock(
context = Mock(
address=('8.8.8.8', 8888),
trusted_downstream=['127.0.0.1'],
_orig_protocol='http'
)
self.assertTrue(handler.is_forbidden())
self.assertTrue(handler.is_forbidden(context))
handler.context = Mock(
context = Mock(
address=('8.8.8.8', 8888),
trusted_downstream=[],
_orig_protocol='http'
)
self.assertTrue(handler.is_forbidden())
self.assertTrue(handler.is_forbidden(context))
handler.context = Mock(
context = Mock(
address=('192.168.1.1', 8888),
trusted_downstream=[],
_orig_protocol='http'
)
self.assertIsNone(handler.is_forbidden())
self.assertIsNone(handler.is_forbidden(context))
handler.context = Mock(
context = Mock(
address=('8.8.8.8', 8888),
trusted_downstream=[],
_orig_protocol='https'
)
self.assertIsNone(handler.is_forbidden())
self.assertIsNone(handler.is_forbidden(context))
def test_get_client_addr(self):
handler = MixinHandler()