88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
import dnscode
|
|
import pytest
|
|
|
|
def test_A(tmp_path):
|
|
zone = dnscode.Zone(origin='minecraftchest1.us.')
|
|
|
|
zone.new_SOA(mname='ns1.minecraftchest1.us.', rname='admin.minecraftchest1.us.', serial=1, refresh=65535, retry=3600, ttl=3600)
|
|
|
|
# Test named and positional arguments. Ensure defaults work.
|
|
zone.new_A("1")
|
|
zone.new_A("2", 60)
|
|
zone.new_A("3", 60, "0.0.0.0")
|
|
zone.new_A(host="0.0.0.0")
|
|
zone.new_A(name="4")
|
|
zone.new_A(ttl=120)
|
|
zone.new_A(name=5, host="0.0.0.0", ttl=120)
|
|
|
|
# Test improper arguments
|
|
zone.new_A(name=6)
|
|
zone.new_A(ttl="60")
|
|
with pytest.raises(dnscode.InvalidDataException):
|
|
zone.new_A(host="fe80::42:2cff:fe29:8db1")
|
|
with pytest.raises(ValueError):
|
|
zone.new_A(host="fe80::42:2cff:fe29:8db1/64")
|
|
with pytest.raises(ValueError):
|
|
zone.new_A(host="0.0.0.0/32")
|
|
|
|
zone_file_1 = tmp_path / "test-A.zone"
|
|
zone.save_file(zone_file_1)
|
|
|
|
expected = """@ 3600 IN SOA ns1.minecraftchest1.us. admin.minecraftchest1.us. 1 65535 3600 15552000 65535
|
|
1.minecraftchest1.us. 3600 IN A 0.0.0.0
|
|
2.minecraftchest1.us. 60 IN A 0.0.0.0
|
|
3.minecraftchest1.us. 60 IN A 0.0.0.0
|
|
@.minecraftchest1.us. 3600 IN A 0.0.0.0
|
|
4.minecraftchest1.us. 3600 IN A 0.0.0.0
|
|
@.minecraftchest1.us. 120 IN A 0.0.0.0
|
|
5.minecraftchest1.us. 120 IN A 0.0.0.0
|
|
6.minecraftchest1.us. 3600 IN A 0.0.0.0
|
|
@.minecraftchest1.us. 60 IN A 0.0.0.0
|
|
|
|
"""
|
|
|
|
f = open(zone_file_1, "rt")
|
|
assert expected.strip() == f.read().strip()
|
|
|
|
##############################################
|
|
|
|
def test_AAAA(tmp_path):
|
|
zoneAAAA = dnscode.Zone(origin='minecraftchest1.us')
|
|
|
|
# Test named and positional arguments. Ensure defaults work.
|
|
zoneAAAA.new_AAAA("0")
|
|
zoneAAAA.new_AAAA("1", 60)
|
|
zoneAAAA.new_AAAA("2", 60, "fe80::42:2cff:fe29:8db1")
|
|
zoneAAAA.new_AAAA(host="fe79::42:2cff:fe29:8db1")
|
|
zoneAAAA.new_AAAA(name="3")
|
|
zoneAAAA.new_AAAA(ttl=119)
|
|
zoneAAAA.new_AAAA(name=4, host="fe80::42:2cff:fe29:8db1", ttl=120)
|
|
|
|
# Test improper arguments
|
|
zoneAAAA.new_AAAA(name=5)
|
|
zoneAAAA.new_AAAA(ttl="59")
|
|
with pytest.raises(dnscode.InvalidDataException):
|
|
zoneAAAA.new_AAAA(host="1.0.0.0")
|
|
with pytest.raises(ValueError):
|
|
zoneAAAA.new_AAAA(host="fe79::42:2cff:fe29:8db1/64")
|
|
with pytest.raises(ValueError):
|
|
zoneAAAA.new_AAAA(host="-1.0.0.0/32")
|
|
|
|
zone_file_2 = tmp_path / "test-AAAA.zone"
|
|
zoneAAAA.save_file(zone_file_2)
|
|
|
|
expected = """0.minecraftchest1.us. 3600 IN AAAA fe80::42:2cff:fe29:8db1
|
|
1.minecraftchest1.us. 60 IN AAAA fe80::42:2cff:fe29:8db1
|
|
2.minecraftchest1.us. 60 IN AAAA fe80::42:2cff:fe29:8db1
|
|
@.minecraftchest1.us. 3600 IN AAAA fe79::42:2cff:fe29:8db1
|
|
3.minecraftchest1.us. 3600 IN AAAA fe80::42:2cff:fe29:8db1
|
|
@.minecraftchest1.us. 119 IN AAAA fe80::42:2cff:fe29:8db1
|
|
4.minecraftchest1.us. 120 IN AAAA fe80::42:2cff:fe29:8db1
|
|
5.minecraftchest1.us. 3600 IN AAAA fe80::42:2cff:fe29:8db1
|
|
@.minecraftchest1.us. 59 IN AAAA fe80::42:2cff:fe29:8db1
|
|
"""
|
|
|
|
f = open(zone_file_2, "rt")
|
|
assert expected == f.read()
|
|
|
|
|