rec: Fix the EDNS padding python implementation for dnspython 2.0.0

PowerDNS · Mar 8, 2021 · c8567be · c8567be
1 parent 56f1a30
commit c8567be
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/regression-tests.recursor-dnssec/paddingoption.py b/regression-tests.recursor-dnssec/paddingoption.py
@@ -14,10 +14,13 @@ def __init__(self, numberOfBytes):
         super(PaddingOption, self).__init__(12)
         self.numberOfBytes = numberOfBytes
 
-    def to_wire(self, file):
+    def to_wire(self, file=None):
         """Create EDNS packet as defined in rfc7830."""
 
-        file.write(bytes(self.numberOfBytes))
+        if file:
+            file.write(bytes(self.numberOfBytes))
+        else:
+            return bytes(self.numberOfBytes)
 
     def from_wire(cls, otype, wire, current, olen):
         """Read EDNS packet as defined in rfc7830.
@@ -32,6 +35,12 @@ def from_wire(cls, otype, wire, current, olen):
 
     from_wire = classmethod(from_wire)
 
+    # needed in 2.0.0
+    @classmethod
+    def from_wire_parser(cls, otype, parser):
+        data = parser.get_remaining()
+        return cls(len(data))
+
     def __repr__(self):
         return '%s(%d)' % (
             self.__class__.__name__,