forked from ioerror/blockfinder
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathblockfindertest.py
93 lines (79 loc) · 4.11 KB
/
blockfindertest.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
#!/usr/bin/python
import blockfinder
import unittest
import os
import shutil
import tempfile
try:
import ipaddr
except ImportError:
from embedded_ipaddr import ipaddr
class BaseBlockfinderTest(unittest.TestCase):
def setUp(self):
self.base_test_dir = tempfile.mkdtemp()
self.test_dir = self.base_test_dir + "/test/"
self.database_cache = blockfinder.DatabaseCache(self.test_dir)
self.downloader_parser = blockfinder.DownloaderParser(
self.test_dir, self.database_cache, "Mozilla")
self.lookup = blockfinder.Lookup(self.test_dir, self.database_cache)
self.database_cache.connect_to_database()
self.database_cache.set_db_version()
shutil.copy('test_rir_data', self.test_dir + 'test_rir_data')
shutil.copy('test_lir_data.gz', self.test_dir + 'test_lir_data.gz')
self.downloader_parser.parse_rir_files(['test_rir_data'])
self.downloader_parser.parse_lir_files(['test_lir_data.gz'])
def tearDown(self):
shutil.rmtree(self.base_test_dir, True)
class CheckReverseLookup(BaseBlockfinderTest):
def test_rir_ipv4_lookup(self):
self.assertEqual(self.database_cache.fetch_country_code('ipv4',
'rir', int(ipaddr.IPv4Address('175.45.176.100'))), 'KP')
self.assertEqual(self.database_cache.fetch_country_code('ipv4',
'rir', int(ipaddr.IPv4Address('193.9.26.0'))), 'HU')
self.assertEqual(self.database_cache.fetch_country_code('ipv4',
'rir', int(ipaddr.IPv4Address('193.9.25.1'))), 'PL')
self.assertEqual(self.database_cache.fetch_country_code('ipv4',
'rir', int(ipaddr.IPv4Address('193.9.25.255'))), 'PL')
def test_rir_asn_lookup(self):
self.assertEqual(self.database_cache.fetch_country_code('asn',
'rir', 681), 'NZ')
self.assertEqual(self.database_cache.fetch_country_code('asn',
'rir', 173), 'JP')
def test_lir_ipv4_lookup(self):
self.assertEqual(self.database_cache.fetch_country_code('ipv4',
'lir', int(ipaddr.IPv4Address('80.16.151.184'))), 'IT')
self.assertEqual(self.database_cache.fetch_country_code('ipv4',
'lir', int(ipaddr.IPv4Address('80.16.151.180'))), 'IT')
self.assertEqual(self.database_cache.fetch_country_code('ipv4',
'lir', int(ipaddr.IPv4Address('213.95.6.32'))), 'DE')
def test_lir_ipv6_lookup(self):
self.assertEqual(self.database_cache.fetch_country_code('ipv6',
'lir', int(ipaddr.IPv6Address('2001:0658:021A::'))), 'DE')
self.assertEqual(self.database_cache.fetch_country_code('ipv6',
'lir', int(ipaddr.IPv6Address('2001:67c:320::'))), 'DE')
self.assertEqual(self.database_cache.fetch_country_code('ipv6',
'lir', int(ipaddr.IPv6Address('2001:670:0085::'))), 'FI')
class CheckBlockFinder(BaseBlockfinderTest):
def test_ipv4_bf(self):
known_ipv4_assignments = (
('MM', ['203.81.64.0/19', '203.81.160.0/20']),
('KP', ['175.45.176.0/22']))
for cc, values in known_ipv4_assignments:
expected = [(int(ipaddr.IPv4Network(network_str).network),
int(ipaddr.IPv4Network(network_str).broadcast))
for network_str in values]
result = self.database_cache.fetch_assignments('ipv4', cc)
self.assertEqual(result, expected)
def test_ipv6_bf(self):
known_ipv6_assignments = ['2001:200::/35', '2001:200:2000::/35',
'2001:200:4000::/34', '2001:200:8000::/33']
expected = [(int(ipaddr.IPv6Network(network_str).network),
int(ipaddr.IPv6Network(network_str).broadcast))
for network_str in known_ipv6_assignments]
result = self.database_cache.fetch_assignments('ipv6', 'JP')
self.assertEqual(result, expected)
if __name__ == '__main__':
for test_class in [CheckReverseLookup, CheckBlockFinder]:
test_suite = unittest.makeSuite(test_class)
test_runner = unittest.TextTestRunner(verbosity=2)
test_runner.run(test_suite)