From a8ecfab93096cd3201efcb06a65bc86a2444611a Mon Sep 17 00:00:00 2001 From: olofvndrhr Date: Thu, 29 Feb 2024 16:23:00 +0100 Subject: [PATCH] add new tests --- tests/test_escaple_semicolon.py | 75 +++++++++++++++++++++++++++++++++ 1 file changed, 75 insertions(+) create mode 100644 tests/test_escaple_semicolon.py diff --git a/tests/test_escaple_semicolon.py b/tests/test_escaple_semicolon.py new file mode 100644 index 0000000..04a9b65 --- /dev/null +++ b/tests/test_escaple_semicolon.py @@ -0,0 +1,75 @@ +from octodns_netbox_dns import NetBoxDNSProvider + + +DEFAULT_CONFIG = { + "id": 1, + "url": "https://localhost:8000", + "token": "", + "view": False, + "replace_duplicates": False, + "make_absolute": True, +} + + +def test_escape1(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"v=TLSRPTv1\\; rua=mailto:tlsrpt@example.com" + value = nbdns._fix_semicolon(rcd_value, escape=True) + + assert value == r"v=TLSRPTv1\; rua=mailto:tlsrpt@example.com" + + +def test_escape2(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"v=TLSRPTv1; rua=mailto:tlsrpt@example.com" + value = nbdns._fix_semicolon(rcd_value, escape=True) + + assert value == r"v=TLSRPTv1\; rua=mailto:tlsrpt@example.com" + + +def test_escape3(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"t=y\;o=~\;" + value = nbdns._fix_semicolon(rcd_value, escape=True) + + assert value == r"t=y\;o=~\;" + + +def test_escape4(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"t=y;o=~;" + value = nbdns._fix_semicolon(rcd_value, escape=True) + + assert value == r"t=y\;o=~\;" + + +def test_unescape1(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"v=TLSRPTv1\; rua=mailto:tlsrpt@example.com" + value = nbdns._fix_semicolon(rcd_value, escape=False) + + assert value == r"v=TLSRPTv1; rua=mailto:tlsrpt@example.com" + + +def test_unescape2(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"v=TLSRPTv1\\; rua=mailto:tlsrpt@example.com" + value = nbdns._fix_semicolon(rcd_value, escape=False) + + assert value == r"v=TLSRPTv1; rua=mailto:tlsrpt@example.com" + + +def test_unescape3(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"t=y\\;o=~\;" + value = nbdns._fix_semicolon(rcd_value, escape=False) + + assert value == r"t=y;o=~;" + + +def test_unescape4(): + nbdns = NetBoxDNSProvider(**DEFAULT_CONFIG) + rcd_value = r"t=y;o=~;" + value = nbdns._fix_semicolon(rcd_value, escape=False) + + assert value == r"t=y;o=~;"