#!/usr/bin/env python3

import re

from updater import (
    diff_zones,
    RECORD_LINE_PATTERN,
)


def test_record_pattern():
    assert re.match(RECORD_LINE_PATTERN, "foo 300 IN A 1.1.1.1")
    assert re.match(RECORD_LINE_PATTERN, "foo\t 300\t   IN   \tA\t   1.1.1.1")


AT_RECORD = {
    "comment": None,
    "rrset_name": "@",
    "rrset_ttl": 300,
    "rrset_type": "A",
    "rrset_values": ["67.166.27.157"],
}
A_RECORD = {
    "comment": None,
    "rrset_name": "www",
    "rrset_ttl": 300,
    "rrset_type": "A",
    "rrset_values": ["67.166.27.157"],
}
REGISTRY_RECORD = {
    "comment": None,
    "rrset_name": "registry",
    "rrset_ttl": 300,
    "rrset_type": "A",
    "rrset_values": ["67.166.27.157"],
}
MIRROR_RECORD = {
    "comment": None,
    "rrset_name": "mirror",
    "rrset_ttl": 300,
    "rrset_type": "A",
    "rrset_values": ["67.166.27.157"],
}


def test_diff_zones():
    z1 = [AT_RECORD, A_RECORD]
    z2 = []
    assert diff_zones(z1, z2) == (z1, [])

    z1 = [AT_RECORD, A_RECORD]
    z2 = [AT_RECORD]
    assert diff_zones(z1, z2) == ([A_RECORD], [])

    z1 = [AT_RECORD, A_RECORD]
    z2 = [A_RECORD]
    assert diff_zones(z1, z2) == ([AT_RECORD], [])

    z2 = [AT_RECORD, A_RECORD]
    z1 = [A_RECORD]
    assert diff_zones(z1, z2) == ([], [AT_RECORD])