23 lines
585 B
Python
23 lines
585 B
Python
|
"""
|
||
|
|
Tests for common protocol functions
|
||
|
|
"""
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
class TestProtocol(unittest.TestCase):
|
||
|
|
"""Main protocol test case"""
|
||
|
|
|
||
|
|
def test_check_local(self):
|
||
|
|
"""Check the logic of TCPConnection.local"""
|
||
|
|
from pybitmessage import protocol, state
|
||
|
|
|
||
|
|
self.assertTrue(
|
||
|
|
protocol.checkIPAddress(protocol.encodeHost('127.0.0.1'), True))
|
||
|
|
self.assertTrue(
|
||
|
|
protocol.checkIPAddress(protocol.encodeHost('192.168.0.1'), True))
|
||
|
|
|
||
|
|
self.assertTrue(
|
||
|
|
not protocol.checkSocksIP('127.0.0.1')
|
||
|
|
or state.socksIP)
|