allow zonefile directives to be specified explicitly

1 files changed, 63 insertions, 0 deletions

diff --git a/tests/test_zone.py b/tests/test_zone.py
index 2d10274..5dfa6ff 100644
--- a/tests/test_zone.py
+++ b/tests/test_zone.py
@@ -854,12 +854,75 @@ class ZoneTestCase(unittest.TestCase):
         z2 = dns.zone.from_file(here("example"), "example.", relativize=True)
         self.assertEqual(z1, z2)
 
+    def testNoInclude(self):
+        def bad():
+            dns.zone.from_text(
+                include_text, "example.", relativize=True, allow_include=False
+            )
+
+        self.assertRaises(dns.exception.SyntaxError, bad)
+
+    def testExplicitInclude(self):
+        z1 = dns.zone.from_text(
+            include_text,
+            "example.",
+            relativize=True,
+            allow_directives={"$INCLUDE", "$ORIGIN", "$TTL"},
+        )
+        z2 = dns.zone.from_file(here("example"), "example.", relativize=True)
+        self.assertEqual(z1, z2)
+
+    def testExplicitLowerCase(self):
+        z1 = dns.zone.from_text(
+            include_text,
+            "example.",
+            relativize=True,
+            allow_directives={"$include", "$origin", "$ttl"},
+        )
+        z2 = dns.zone.from_file(here("example"), "example.", relativize=True)
+        self.assertEqual(z1, z2)
+
+    def testExplicitWithoutInclude1(self):
+        def bad():
+            dns.zone.from_text(
+                include_text,
+                "example.",
+                relativize=True,
+                allow_include=False,
+                allow_directives={"$ORIGIN", "$TTL"},
+            )
+
+        self.assertRaises(dns.exception.SyntaxError, bad)
+
+    def testExplicitWithoutInclude2(self):
+        def bad():
+            dns.zone.from_text(
+                include_text,
+                "example.",
+                relativize=True,
+                allow_include=True,
+                allow_directives={"$ORIGIN", "$TTL"},
+            )
+
+        self.assertRaises(dns.exception.SyntaxError, bad)
+
     def testBadDirective(self):
         def bad():
             dns.zone.from_text(bad_directive_text, "example.", relativize=True)
 
         self.assertRaises(dns.exception.SyntaxError, bad)
 
+    def testAllowedButNotImplementedDirective(self):
+        def bad():
+            dns.zone.from_text(
+                bad_directive_text,
+                "example.",
+                relativize=True,
+                allow_directives={"$FOO", "$ORIGIN"},
+            )
+
+        self.assertRaises(dns.exception.SyntaxError, bad)
+
     def testFirstRRStartsWithWhitespace(self):
         # no name is specified, so default to the initial origin
         z = dns.zone.from_text(