From: Bob Halley <halley@dnspython.org>
Date: Tue, 1 Sep 2020 13:51:25 +0000 (-0700)
Subject: add some versioned zone test cases
X-Git-Tag: v2.1.0rc1~25
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=97e35f2b0ec57a38b473744f1cd7783044bc1129;p=thirdparty%2Fdnspython.git

add some versioned zone test cases
---

diff --git a/tests/test_zone.py b/tests/test_zone.py
index 78fbb28c..3f938c6d 100644
--- a/tests/test_zone.py
+++ b/tests/test_zone.py
@@ -32,6 +32,7 @@ import dns.rdataset
 import dns.rdataclass
 import dns.rdatatype
 import dns.rrset
+import dns.versioned
 import dns.zone
 import dns.node
 
@@ -801,5 +802,29 @@ class ZoneTestCase(unittest.TestCase):
         self.assertTrue(rds is not rrs)
         self.assertFalse(isinstance(rds, dns.rrset.RRset))
 
+
+class VersionedZoneTestCase(unittest.TestCase):
+    def testUseTransaction(self):
+        z = dns.zone.from_text(example_text, 'example.', relativize=True,
+                               zone_factory=dns.versioned.Zone)
+        with self.assertRaises(dns.versioned.UseTransaction):
+            z.find_node('not_there', True)
+        with self.assertRaises(dns.versioned.UseTransaction):
+            z.delete_node('not_there')
+        with self.assertRaises(dns.versioned.UseTransaction):
+            z.find_rdataset('not_there', 'a', create=True)
+        with self.assertRaises(dns.versioned.UseTransaction):
+            z.get_rdataset('not_there', 'a', create=True)
+        with self.assertRaises(dns.versioned.UseTransaction):
+            z.delete_rdataset('not_there', 'a')
+        with self.assertRaises(dns.versioned.UseTransaction):
+            z.replace_rdataset('not_there', None)
+
+    def testSelectDefaultPruningPolicy(self):
+        z = dns.zone.from_text(example_text, 'example.', relativize=True,
+                               zone_factory=dns.versioned.Zone)
+        z.set_pruning_policy(None)
+        self.assertEqual(z._pruning_policy, z._default_pruning_policy)
+
 if __name__ == '__main__':
     unittest.main()