Jan Dittberner
585085250e
- Fix deprecation warnings - Fix missing API in tests - Fix relative imports in tests - use proper logger
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from io import StringIO
|
|
from .util import TestApp, test_root
|
|
import unittest
|
|
|
|
|
|
IP4_ADDRESSES = ['127.0.0.1', '192.168.0.1']
|
|
IP6_ADDRESSES = ['::1', '2001:dead:beef::1']
|
|
IP4_RANGES = ['172.16.0.0/24', '192.168.0.0/24']
|
|
IP6_RANGES = ['2001:dead:beef::/64', '2001:dada:b001::/64']
|
|
|
|
|
|
class TestIPExtension(unittest.TestCase):
|
|
def setUp(self):
|
|
if not (test_root / '_static').exists():
|
|
(test_root / '_static').mkdir()
|
|
self.feed_warnfile = StringIO()
|
|
self.app = TestApp(
|
|
buildername='html', warning=self.feed_warnfile, cleanenv=True)
|
|
self.app.build(force_all=True, filenames=[])
|
|
|
|
def tearDown(self):
|
|
self.app.cleanup()
|
|
(test_root / '_build').rmtree(True)
|
|
|
|
def test_ip_domaindata(self):
|
|
self.assertIn('ip', self.app.env.domaindata)
|
|
ipdomdata = self.app.env.domaindata['ip']
|
|
self.assertIn('v4', ipdomdata)
|
|
self.assertIn('v6', ipdomdata)
|
|
self.assertIn('v4range', ipdomdata)
|
|
self.assertIn('v6range', ipdomdata)
|
|
self.assertIn('ips', ipdomdata)
|
|
|
|
def find_in_index(self, entry):
|
|
indexentries = self.app.env.indexentries
|
|
for index in indexentries:
|
|
for value in indexentries[index]:
|
|
if value[1] == entry:
|
|
return
|
|
self.fail("%s not found in index" % entry)
|
|
|
|
def test_ip4_addresses(self):
|
|
ipv4 = self.app.env.domaindata['ip']['v4']
|
|
ips = self.app.env.domaindata['ip']['ips']
|
|
for ip in IP4_ADDRESSES:
|
|
self.assertIn(ip, ipv4)
|
|
self.assertIn(ip, [item['ip'] for item in ips])
|
|
self.find_in_index("IPv4 address; %s" % ip)
|
|
self.find_in_index("%s; Test page 2" % ip)
|
|
|
|
def test_ip6_addresses(self):
|
|
ipv6 = self.app.env.domaindata['ip']['v6']
|
|
ips = self.app.env.domaindata['ip']['ips']
|
|
for ip in IP6_ADDRESSES:
|
|
self.assertIn(ip, ipv6)
|
|
self.assertIn(ip, [item['ip'] for item in ips])
|
|
self.find_in_index("IPv6 address; %s" % ip)
|
|
self.find_in_index("%s; Test page 2" % ip)
|