28 lines
874 B
Python
28 lines
874 B
Python
|
"""
|
||
|
|
Tests for dnstools module.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from nsupdate.main.dnstools import (query_ns, NONEXISTING_HOST, WWW_HOST, WWW_IPV4_HOST, WWW_IPV4_IP,
|
||
|
|
WWW_IPV6_HOST, WWW_IPV6_IP, )
|
||
|
|
|
||
|
|
from dns.resolver import NXDOMAIN
|
||
|
|
|
||
|
|
|
||
|
|
class Tests(object):
|
||
|
|
def test_queries_ok(self):
|
||
|
|
"""
|
||
|
|
check some simple dns lookups
|
||
|
|
"""
|
||
|
|
assert query_ns(WWW_IPV4_HOST, 'A') == WWW_IPV4_IP # v4 ONLY
|
||
|
|
assert query_ns(WWW_IPV6_HOST, 'AAAA') == WWW_IPV6_IP # v6 ONLY
|
||
|
|
assert query_ns(WWW_HOST, 'A') == WWW_IPV4_IP # v4 and v6, query v4
|
||
|
|
assert query_ns(WWW_HOST, 'AAAA') == WWW_IPV6_IP # v4 and v6, query v6
|
||
|
|
|
||
|
|
def test_queries_failing(self):
|
||
|
|
with pytest.raises(NXDOMAIN):
|
||
|
|
query_ns(NONEXISTING_HOST, 'A')
|
||
|
|
with pytest.raises(NXDOMAIN):
|
||
|
|
query_ns(NONEXISTING_HOST, 'AAAA')
|