""" Test for network group """ import unittest from .common import skip_python3 skip_python3() class TestNetworkGroup(unittest.TestCase): """ Test case for network group """ def test_network_group(self): """Test various types of network groups""" from pybitmessage.protocol import network_group test_ip = '1.2.3.4' self.assertEqual('\x01\x02', network_group(test_ip)) test_ip = '127.0.0.1' self.assertEqual('IPv4', network_group(test_ip)) test_ip = '0102:0304:0506:0708:090A:0B0C:0D0E:0F10' self.assertEqual( '\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C', network_group(test_ip)) test_ip = 'bootstrap8444.bitmessage.org' self.assertEqual( 'bootstrap8444.bitmessage.org', network_group(test_ip)) test_ip = 'quzwelsuziwqgpt2.onion' self.assertEqual( test_ip, network_group(test_ip)) test_ip = None self.assertEqual( None, network_group(test_ip))