Do no raise an assertion getting a nonexistent rdataset without a txn in a versioned zone.2.2

2 files changed, 16 insertions, 1 deletions

diff --git a/dns/versioned.py b/dns/versioned.py
index 8b6c275..9e26759 100644
--- a/dns/versioned.py
+++ b/dns/versioned.py
@@ -265,7 +265,10 @@ class Zone(dns.zone.Zone):
         if create:
             raise UseTransaction
         rdataset = super().get_rdataset(name, rdtype, covers)
-        return dns.rdataset.ImmutableRdataset(rdataset)
+        if rdataset is not None:
+            return dns.rdataset.ImmutableRdataset(rdataset)
+        else:
+            return None
 
     def delete_rdataset(self, name, rdtype, covers=dns.rdatatype.NONE):
         raise UseTransaction
diff --git a/tests/test_zone.py b/tests/test_zone.py
index 45fa272..9059388 100644
--- a/tests/test_zone.py
+++ b/tests/test_zone.py
@@ -1076,6 +1076,18 @@ class VersionedZoneTestCase(unittest.TestCase):
                                             dns.rdatatype.RRSIG,
                                             dns.rdatatype.NSEC))
 
+    def testGetRdataset1(self):
+        z = dns.zone.from_text(example_text, 'example.', relativize=True,
+                               zone_factory=dns.versioned.Zone)
+        rds = z.get_rdataset('@', 'soa')
+        exrds = dns.rdataset.from_text('IN', 'SOA', 300, 'foo bar 1 2 3 4 5')
+        self.assertEqual(rds, exrds)
+
+    def testGetRdataset2(self):
+        z = dns.zone.from_text(example_text, 'example.', relativize=True,
+                               zone_factory=dns.versioned.Zone)
+        rds = z.get_rdataset('@', 'loc')
+        self.assertTrue(rds is None)
 
 if __name__ == '__main__':
     unittest.main()