<filename>lemur/tests/test_dns_providers.py
import json
import unittest
from lemur.dns_providers import util as dnsutil
from lemur.dns_providers.schemas import dns_provider_output_schema


class TestDNSProvider(unittest.TestCase):
    def test_is_valid_domain(self):
        self.assertTrue(dnsutil.is_valid_domain('example.com'))
        self.assertTrue(dnsutil.is_valid_domain('foo.bar.org'))
        self.assertTrue(dnsutil.is_valid_domain('exam--ple.io'))
        self.assertTrue(dnsutil.is_valid_domain('a.example.com'))
        self.assertTrue(dnsutil.is_valid_domain('example.io'))
        self.assertTrue(dnsutil.is_valid_domain('example-of-under-63-character-domain-label-length-limit-1234567.com'))
        self.assertFalse(dnsutil.is_valid_domain('example-of-over-63-character-domain-label-length-limit-123456789.com'))
        self.assertTrue(dnsutil.is_valid_domain('_acme-chall.example.com'))
        self.assertFalse(dnsutil.is_valid_domain('e/xample.com'))
        self.assertFalse(dnsutil.is_valid_domain('exam\\ple.com'))
        self.assertFalse(dnsutil.is_valid_domain('<example.com'))
        self.assertFalse(dnsutil.is_valid_domain('*.example.com'))
        self.assertFalse(dnsutil.is_valid_domain('-example.io'))
        self.assertFalse(dnsutil.is_valid_domain('example-.io'))
        self.assertFalse(dnsutil.is_valid_domain('example..io'))
        self.assertFalse(dnsutil.is_valid_domain('exa mple.io'))
        self.assertFalse(dnsutil.is_valid_domain('-'))


def test_output_schema(dns_provider):
    # no credentials using the output schema dump
    assert dns_provider.credentials
    assert json.loads(dns_provider.credentials)["account_id"]
    dump = dns_provider_output_schema.dump(dns_provider).data
    assert 'name' in dump
    assert 'credentials' not in dump


def test_json(dns_provider):
    # we can still get credentials using json.load
    assert 'account_id' in json.loads(dns_provider.credentials)

<filename>lemur/tests/test_dns_providers.py
import json
import unittest
from lemur.dns_providers import util as dnsutil
from lemur.dns_providers.schemas import dns_provider_output_schema


class TestDNSProvider(unittest.TestCase):
    def test_is_valid_domain(self):
        self.assertTrue(dnsutil.is_valid_domain('example.com'))
        self.assertTrue(dnsutil.is_valid_domain('foo.bar.org'))
        self.assertTrue(dnsutil.is_valid_domain('exam--ple.io'))
        self.assertTrue(dnsutil.is_valid_domain('a.example.com'))
        self.assertTrue(dnsutil.is_valid_domain('example.io'))
        self.assertTrue(dnsutil.is_valid_domain('example-of-under-63-character-domain-label-length-limit-1234567.com'))
        self.assertFalse(dnsutil.is_valid_domain('example-of-over-63-character-domain-label-length-limit-123456789.com'))
        self.assertTrue(dnsutil.is_valid_domain('_acme-chall.example.com'))
        self.assertFalse(dnsutil.is_valid_domain('e/xample.com'))
        self.assertFalse(dnsutil.is_valid_domain('exam\\ple.com'))
        self.assertFalse(dnsutil.is_valid_domain('<example.com'))
        self.assertFalse(dnsutil.is_valid_domain('*.example.com'))
        self.assertFalse(dnsutil.is_valid_domain('-example.io'))
        self.assertFalse(dnsutil.is_valid_domain('example-.io'))
        self.assertFalse(dnsutil.is_valid_domain('example..io'))
        self.assertFalse(dnsutil.is_valid_domain('exa mple.io'))
        self.assertFalse(dnsutil.is_valid_domain('-'))


def test_output_schema(dns_provider):
    # no credentials using the output schema dump
    assert dns_provider.credentials
    assert json.loads(dns_provider.credentials)["account_id"]
    dump = dns_provider_output_schema.dump(dns_provider).data
    assert 'name' in dump
    assert 'credentials' not in dump


def test_json(dns_provider):
    # we can still get credentials using json.load
    assert 'account_id' in json.loads(dns_provider.credentials)

# no credentials using the output schema dump # we can still get credentials using json.load