apply the immutable init wrapper to __setstate__ too, if present

1 files changed, 7 insertions, 0 deletions

diff --git a/dns/_immutable_attr.py b/dns/_immutable_attr.py
index 9d76cdf..61aa1b1 100644
--- a/dns/_immutable_attr.py
+++ b/dns/_immutable_attr.py
@@ -60,9 +60,16 @@ def immutable(cls):
     else:
         # Mixin the Immutable class and follow the __init__ protocol.
         class ncls(_Immutable, cls):
+
             @_immutable_init
             def __init__(self, *args, **kwargs):
                 super().__init__(*args, **kwargs)
+
+            if hasattr(cls, '__setstate__'):
+                @_immutable_init
+                def __setstate__(self, *args, **kwargs):
+                    super().__setstate__(*args, **kwargs)
+
         # make ncls have the same name and module as cls
         ncls.__name__ = cls.__name__
         ncls.__qualname__ = cls.__qualname__