import dnscode
import pytest

def test_A(tmp_path):
	zone = dnscode.Zone(origin='minecraftchest1.us')

	# 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 = """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 == f.read()

##############################################

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()