-
Notifications
You must be signed in to change notification settings - Fork 89
/
test.py
238 lines (208 loc) · 11.5 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
import unittest
from ipcalc import IP, Network
class TestSuite(unittest.TestCase):
"""Tests."""
def test_ipv4_1(self):
net = Network('192.168.114.42', 23)
self.assertTrue(str(net) == '192.168.114.42/23')
self.assertTrue(str(net.to_ipv6().to_compressed()) == '2002:c0a8:722a::')
self.assertTrue(net.info() == 'PRIVATE')
self.assertTrue(net.subnet() == 23)
self.assertTrue(net.size() == 1 << (32 - 23))
self.assertTrue(int(net) == 0xc0a8722a)
self.assertTrue(net.hex().lower() == 'c0a8722a')
self.assertTrue(str(net.netmask()) == '255.255.254.0')
self.assertTrue(net.version() == 4)
self.assertTrue(str(net.network()) == '192.168.114.0')
self.assertTrue(str(net.broadcast()) == '192.168.115.255')
self.assertFalse('192.168.0.1' in net)
self.assertTrue('192.168.114.128' in net)
self.assertFalse('10.0.0.1' in net)
self.assertTrue(str(net + 6) == '192.168.114.48/23')
self.assertTrue((net + 6) in net)
self.assertTrue(str(net - 6) == '192.168.114.36/23')
self.assertTrue((net - 6) in net)
def test_ipv4_2(self):
net = Network('10.10.0.0', '255.255.255.0')
self.assertTrue(str(net) == '10.10.0.0/24')
self.assertTrue(str(net.to_ipv6().to_compressed()) == '2002:a0a::')
self.assertTrue(net.info() == 'PRIVATE')
self.assertTrue(net.subnet() == 24)
self.assertTrue(net.size() == 1 << (32 - 24))
self.assertTrue(int(net) == 0x0a0a0000)
self.assertTrue(net.hex().lower() == '0a0a0000')
self.assertTrue(str(net.netmask()) == '255.255.255.0')
self.assertTrue(net.version() == 4)
self.assertTrue(str(net.network()) == '10.10.0.0')
self.assertTrue(str(net.broadcast()) == '10.10.0.255')
self.assertFalse('192.168.0.1' in net)
self.assertFalse('192.168.114.128' in net)
self.assertFalse('10.0.0.1' in net)
self.assertTrue('10.10.0.254' in net)
self.assertTrue('10.10.0.100' in net)
self.assertTrue(str(net + 6) == '10.10.0.6/24')
self.assertTrue(str(net + 6) in net)
self.assertTrue(str(net - 6) == '10.9.255.250/24') # note, result is not in subnet
self.assertFalse(str(net -6) in net)
def test_ipv4_3(self):
net = Network('10.10.0.0/255.255.255.0')
self.assertTrue(str(net) == '10.10.0.0/24')
self.assertTrue(str(net.to_ipv6().to_compressed()) == '2002:a0a::')
self.assertTrue(net.info() == 'PRIVATE')
self.assertTrue(net.subnet() == 24)
self.assertTrue(net.size() == 1 << (32 - 24))
self.assertTrue(int(net) == 0x0a0a0000)
self.assertTrue(net.hex().lower() == '0a0a0000')
self.assertTrue(str(net.netmask()) == '255.255.255.0')
self.assertTrue(net.version() == 4)
self.assertTrue(str(net.network()) == '10.10.0.0')
self.assertTrue(str(net.broadcast()) == '10.10.0.255')
self.assertFalse('192.168.0.1' in net)
self.assertFalse('192.168.114.128' in net)
self.assertFalse('10.0.0.1' in net)
self.assertTrue('10.10.0.254' in net)
self.assertTrue('10.10.0.100' in net)
def test_ipv6_1(self):
net = Network('123::', 128)
self.assertTrue(str(net) == '0123:0000:0000:0000:0000:0000:0000:0000/128')
self.assertTrue(str(net.to_compressed()) == '123::')
self.assertTrue(str(net.to_ipv6().to_compressed()) == '123::')
self.assertTrue(net.info() == 'UNKNOWN')
self.assertTrue(net.subnet() == 128)
self.assertTrue(net.size() == 1 << (128 - 128))
self.assertTrue(int(net) == (0x123<<112))
self.assertTrue(net.hex().lower() == '01230000000000000000000000000000')
self.assertTrue(str(net.netmask()) == 'ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff')
self.assertTrue(net.version() == 6)
self.assertTrue(str(net.network()) == '0123:0000:0000:0000:0000:0000:0000:0000')
self.assertTrue(str(net.broadcast()) == '0123:0000:0000:0000:0000:0000:0000:0000')
self.assertFalse('123:456::' in net)
self.assertTrue('123::' in net)
self.assertFalse('::1' in net)
self.assertFalse('123::456' in net)
self.assertTrue(str((net + 6).to_compressed()).lower() == '123::6')
self.assertFalse((net + 6) in net)
self.assertTrue(str((net - 6).to_compressed()).lower() == '122:ffff:ffff:ffff:ffff:ffff:ffff:fffa')
self.assertFalse((net - 6) in net)
def test_ipv6_2(self):
net = Network('::42', 64)
self.assertTrue(str(net) == '0000:0000:0000:0000:0000:0000:0000:0042/64')
self.assertTrue(str(net.to_compressed()) == '::42')
self.assertTrue(str(net.to_ipv6().to_compressed()) == '::42')
self.assertTrue(net.info() == 'IPV4COMP')
self.assertTrue(net.subnet() == 64)
self.assertTrue(net.size() == 1 << (128 - 64))
self.assertTrue(int(net) == 0x42)
self.assertTrue(net.hex().lower() == '00000000000000000000000000000042')
self.assertTrue(str(net.netmask()) == 'ffff:ffff:ffff:ffff:0000:0000:0000:0000')
self.assertTrue(net.version() == 6)
self.assertTrue(str(net.network()) == '0000:0000:0000:0000:0000:0000:0000:0000')
self.assertTrue(str(net.broadcast()) == '0000:0000:0000:0000:ffff:ffff:ffff:ffff')
self.assertFalse('123:456::' in net)
self.assertTrue('::aaaa:bbbb:cccc:dddd' in net)
self.assertTrue('::dddd' in net)
self.assertTrue('::1' in net)
self.assertFalse('123::456' in net)
self.assertTrue(str((net + 6).to_compressed()).lower() == '::48')
self.assertTrue((net + 6) in net)
self.assertTrue(str((net - 6).to_compressed()).lower() == '::3c')
self.assertTrue((net - 6) in net)
def test_ipv6_3(self):
net = Network('2001:dead:beef:1:c01d:c01a::', 48)
self.assertTrue(str(net) == '2001:dead:beef:0001:c01d:c01a:0000:0000/48')
self.assertTrue(str(net.to_compressed()) == '2001:dead:beef:1:c01d:c01a::')
self.assertTrue(str(net.to_ipv6().to_compressed()) == '2001:dead:beef:1:c01d:c01a::')
self.assertTrue(net.info() == 'UNKNOWN')
self.assertTrue(net.subnet() == 48)
self.assertTrue(net.size() == 1 << (128 - 48))
self.assertTrue(int(net) == 0x2001deadbeef0001c01dc01a00000000)
self.assertTrue(net.hex().lower() == '2001deadbeef0001c01dc01a00000000')
self.assertTrue(str(net.netmask()) == 'ffff:ffff:ffff:0000:0000:0000:0000:0000')
self.assertTrue(net.version() == 6)
self.assertTrue(str(net.network()) == '2001:dead:beef:0000:0000:0000:0000:0000')
self.assertTrue(str(net.broadcast()) == '2001:dead:beef:ffff:ffff:ffff:ffff:ffff')
self.assertFalse('123:456::' in net)
self.assertFalse('::aaaa:bbbb:cccc:dddd' in net)
self.assertFalse('::dddd' in net)
self.assertFalse('::1' in net)
self.assertFalse('123::456' in net)
self.assertTrue('2001:dead:beef:babe::1234' in net)
def test_ipv6_4(self):
net = Network('2001:dead:beef:1:c01d:c01a::', 'ffff:ffff:ffff::')
self.assertTrue(str(net) == '2001:dead:beef:0001:c01d:c01a:0000:0000/48')
self.assertTrue(str(net.to_compressed()) == '2001:dead:beef:1:c01d:c01a::')
self.assertTrue(str(net.to_ipv6().to_compressed()) == '2001:dead:beef:1:c01d:c01a::')
self.assertTrue(net.info() == 'UNKNOWN')
self.assertTrue(net.subnet() == 48)
self.assertTrue(net.size() == 1 << (128 - 48))
self.assertTrue(int(net) == 0x2001deadbeef0001c01dc01a00000000)
self.assertTrue(net.hex().lower() == '2001deadbeef0001c01dc01a00000000')
self.assertTrue(str(net.netmask()) == 'ffff:ffff:ffff:0000:0000:0000:0000:0000')
self.assertTrue(net.version() == 6)
self.assertTrue(str(net.network()) == '2001:dead:beef:0000:0000:0000:0000:0000')
self.assertTrue(str(net.broadcast()) == '2001:dead:beef:ffff:ffff:ffff:ffff:ffff')
self.assertFalse('123:456::' in net)
self.assertFalse('::aaaa:bbbb:cccc:dddd' in net)
self.assertFalse('::dddd' in net)
self.assertFalse('::1' in net)
self.assertFalse('123::456' in net)
self.assertTrue('2001:dead:beef:babe::1234' in net)
def test_ipv6_5(self):
# test parsing of 4-in-6 IPv6 address
ip = IP('8000::0.0.0.1')
self.assertTrue(ip.ip == ((2**127) + 1))
ip = IP('8000:8000::0.0.0.1')
self.assertTrue(ip.ip == ((2**127) + (2**111) + 1))
class TestIP(unittest.TestCase):
"""Tests for IP."""
def test_eq_le_gt(self):
self.assertEqual(IP('192.168.11.0'), IP('192.168.11.0'))
self.assertNotEqual(IP('192.168.1.0'), IP('192.168.11.0'))
def test_guesstimation(self):
self.assertEqual(IP('192.168.0.1', mask=28).guess_network(), Network('192.168.0.0/28'))
self.assertEqual(IP('192.168.0.1/24').guess_network(), Network('192.168.0.0/24'))
self.assertEqual(IP('192.168.0.1/255.255.255.0', mask=28).guess_network(), Network('192.168.0.0/24'))
self.assertEqual(IP('192.168.0.56', mask=26).guess_network(), Network('192.168.0.0/26'))
self.assertEqual(IP('192.168.0.1').guess_network(), Network('192.168.0.1/32'))
class TestNetwork(unittest.TestCase):
"""Tests for Network."""
def setUp(self):
self.network = Network('192.168.11.0/255.255.255.0')
def test_calculation(self):
self.assertEqual(self.network[1].subnet(), 24)
a = Network('192.168.0.100/28')
self.assertEqual(str(a), '192.168.0.100/28')
self.assertEqual(a.size(), 16)
self.assertEqual(a.size(), len(a))
self.assertEqual(int(a), 0xC0A80064)
for i in range(a.size()):
self.assertEqual(int(a[i]), i + 0xC0A80064)
self.assertRaises(IndexError, lambda: a[a.size()])
def test_indexers(self):
expected = range(int(0xC0A80B00), int(0xC0A80C00))
self.assertEqual(self.network.size(), len(expected))
for i in range(self.network.size()):
self.assertEqual(int(self.network[i]), expected[i])
self.assertEqual(int(self.network[-1]), expected[-1])
def test_contains(self):
self.assertTrue(IP('192.168.11.0') in self.network)
self.assertTrue(IP('192.168.11.1') in self.network)
self.assertTrue(IP('192.168.11.255') in self.network)
def test_eq_le_gt(self):
self.assertEqual(Network('192.168.11.0'), Network('192.168.11.0'))
self.assertEqual(Network('192.168.11.0/32'), Network('192.168.11.0'))
self.assertEqual(Network('192.168.11.0'), IP('192.168.11.0'))
self.assertEqual(Network('192.168.11.0/32'), IP('192.168.11.0'))
self.assertNotEqual(Network('192.168.11.0/28'), Network('192.168.11.0/24'))
self.assertNotEqual(Network('192.168.11.0'), Network('192.168.11.1'))
self.assertNotEqual(Network('192.168.11.0'), Network('192.168.2.1'))
self.assertNotEqual(Network('192.168.11.0/30'), IP('192.168.11.0'))
self.assertNotEqual(Network('192.168.1.0'), IP('192.168.11.0'))
self.assertTrue(Network('192.168.1.0/30') < Network('192.168.1.0/29'))
self.assertTrue(Network('192.168.1.0/30') <= Network('192.168.1.0/29'))
self.assertTrue(Network('192.168.1.0/30') <= Network('192.168.1.0/30'))
self.assertTrue(Network('192.168.1.0/28') > Network('192.168.1.0/29'))
self.assertTrue(Network('192.168.1.0/28') >= Network('192.168.1.0/29'))
self.assertTrue(Network('192.168.1.0/28') >= Network('192.168.1.0/28'))
if __name__ == '__main__':
unittest.main()