import pytest

import dns.rrset
from dns.zonefile import read_rrsets

expected_mx_1= dns.rrset.from_text('name.', 300, 'in', 'mx', '10 a.', '20 b.')
expected_mx_2 = dns.rrset.from_text('name.', 10, 'in', 'mx', '10 a.', '20 b.')
expected_mx_3 = dns.rrset.from_text('foo.', 10, 'in', 'mx', '10 a.')
expected_mx_4 = dns.rrset.from_text('bar.', 10, 'in', 'mx', '20 b.')
expected_mx_5 = dns.rrset.from_text('foo.example.', 10, 'in', 'mx',
                                    '10 a.example.')
expected_mx_6 = dns.rrset.from_text('bar.example.', 10, 'in', 'mx', '20 b.')
expected_mx_7 = dns.rrset.from_text('foo', 10, 'in', 'mx', '10 a')
expected_mx_8 = dns.rrset.from_text('bar', 10, 'in', 'mx', '20 b.')
expected_ns_1 = dns.rrset.from_text('name.', 300, 'in', 'ns', 'hi.')
expected_ns_2 = dns.rrset.from_text('name.', 300, 'ch', 'ns', 'hi.')

def equal_rrsets(a, b):
    # return True iff. a and b have the same rrsets regardless of order
    if len(a) != len(b):
        return False
    for rrset in a:
        if not rrset in b:
            return False
    return True

def test_name_ttl_rdclass_forced():
    input=''';
mx 10 a
mx 20 b.
ns hi'''
    rrsets = read_rrsets(input, name='name', ttl=300)
    assert equal_rrsets(rrsets, [expected_mx_1, expected_ns_1])
    assert rrsets[0].ttl == 300
    assert rrsets[1].ttl == 300

def test_name_ttl_rdclass_forced_rdata_split():
    input=''';
mx 10 a
ns hi
mx 20 b.'''
    rrsets = read_rrsets(input, name='name', ttl=300)
    assert equal_rrsets(rrsets, [expected_mx_1, expected_ns_1])

def test_name_ttl_rdclass_rdtype_forced():
    input=''';
10 a
20 b.'''
    rrsets = read_rrsets(input, name='name', ttl=300, rdtype='mx')
    assert equal_rrsets(rrsets, [expected_mx_1])

def test_name_rdclass_forced():
    input = '''30 mx 10 a
10 mx 20 b.
'''
    rrsets = read_rrsets(input, name='name')
    assert equal_rrsets(rrsets, [expected_mx_2])
    assert rrsets[0].ttl == 10

def test_rdclass_forced():
    input = ''';
foo 20 mx 10 a
bar 30 mx 20 b.
'''
    rrsets = read_rrsets(input)
    assert equal_rrsets(rrsets, [expected_mx_3, expected_mx_4])

def test_rdclass_forced_with_origin():
    input = ''';
foo 20 mx 10 a
bar.example. 30 mx 20 b.
'''
    rrsets = read_rrsets(input, origin='example')
    assert equal_rrsets(rrsets, [expected_mx_5, expected_mx_6])


def test_rdclass_forced_with_origin_relativized():
    input = ''';
foo 20 mx 10 a.example.
bar.example. 30 mx 20 b.
'''
    rrsets = read_rrsets(input, origin='example', relativize=True)
    assert equal_rrsets(rrsets, [expected_mx_7, expected_mx_8])

def test_rdclass_matching_default_tolerated():
    input = ''';
foo 20 mx 10 a.example.
bar.example. 30 in mx 20 b.
'''
    rrsets = read_rrsets(input, origin='example', relativize=True,
                         rdclass=None)
    assert equal_rrsets(rrsets, [expected_mx_7, expected_mx_8])

def test_rdclass_not_matching_default_rejected():
    input = ''';
foo 20 mx 10 a.example.
bar.example. 30 ch mx 20 b.
'''
    with pytest.raises(dns.exception.SyntaxError):
        rrsets = read_rrsets(input, origin='example', relativize=True,
                             rdclass=None)

def test_default_rdclass_is_none():
    input = ''
    with pytest.raises(TypeError):
        rrsets = read_rrsets(input, default_rdclass=None, origin='example',
                             relativize=True)

def test_name_rdclass_rdtype_force():
    # No real-world usage should do this, but it can be specified so we test it.
    input = ''';
30 10 a
10 20 b.
'''
    rrsets = read_rrsets(input, name='name', rdtype='mx')
    assert equal_rrsets(rrsets, [expected_mx_1])
    assert rrsets[0].ttl == 10

def test_rdclass_rdtype_force():
    # No real-world usage should do this, but it can be specified so we test it.
    input = ''';
foo 30 10 a
bar 30 20 b.
'''
    rrsets = read_rrsets(input, rdtype='mx')
    assert equal_rrsets(rrsets, [expected_mx_3, expected_mx_4])

# also weird but legal
#input5 = '''foo 30 10 a
#bar 10 20 foo.
#'''