"""
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)