fix: fix failing pickling tests by restoring _rab_manager and adding __eq__

Author: macastelaz

packages/google-auth/google/auth/_regional_access_boundary_utils.py

@@ -110,6 +110,16 @@ def __setstate__(self, state):
         self.__dict__.update(state)
         self._update_lock = threading.Lock()
 
+    def __eq__(self, other):
+        """Checks if two managers are equal."""
+        if not isinstance(other, _RegionalAccessBoundaryManager):
+            return NotImplemented
+        return (
+            self._data == other._data
+            and self._use_blocking_regional_access_boundary_lookup
+            == other._use_blocking_regional_access_boundary_lookup
+        )
+
     def use_blocking_regional_access_boundary_lookup(self):
         """Enables blocking regional access boundary lookup to true"""
         self._use_blocking_regional_access_boundary_lookup = True

packages/google-auth/google/oauth2/credentials.py

@@ -207,7 +207,7 @@ def __setstate__(self, d):
         self._refresh_worker = None
         self._use_non_blocking_refresh = d.get("_use_non_blocking_refresh", False)
         self._account = d.get("_account", "")
-        self._rab_manager = (
+        self._rab_manager = d.get("_rab_manager") or (
             _regional_access_boundary_utils._RegionalAccessBoundaryManager()
         )
         self._use_blocking_regional_access_boundary_lookup = False