fix: Guard delete statements. Add default fallback for _use_non_blocking_refresh. (#1445)

Author: clundin25

google/oauth2/credentials.py

@@ -160,9 +160,11 @@ def __getstate__(self):
         # unpickling certain callables (lambda, functools.partial instances)
         # because they need to be importable.
         # Instead, the refresh_handler setter should be used to repopulate this.
-        del state_dict["_refresh_handler"]
-        # Remove worker as it contains multiproccessing queue objects.
-        del state_dict["_refresh_worker"]
+        if "_refresh_handler" in state_dict:
+            del state_dict["_refresh_handler"]
+
+        if "_refresh_worker" in state_dict:
+            del state_dict["_refresh_worker"]
         return state_dict
 
     def __setstate__(self, d):
@@ -186,7 +188,7 @@ def __setstate__(self, d):
         # The refresh_handler setter should be used to repopulate this.
         self._refresh_handler = None
         self._refresh_worker = None
-        self._use_non_blocking_refresh = d.get("_use_non_blocking_refresh")
+        self._use_non_blocking_refresh = d.get("_use_non_blocking_refresh", False)
 
     @property
     def refresh_token(self):

system_tests/secrets.tar.enc

@@ -963,6 +963,8 @@ def test_pickle_with_missing_attribute(self):
         # this mimics a pickle created with a previous class definition with
         # fewer attributes
         del creds.__dict__["_quota_project_id"]
+        del creds.__dict__["_refresh_handler"]
+        del creds.__dict__["_refresh_worker"]
 
         unpickled = pickle.loads(pickle.dumps(creds))