fix: Typo when setting the state for the pickle deserializer. (#1479)

* fix: Typo when setting the state for the pickle deserializer.
googleapis · Feb 21, 2024 · 08b5cc3 · 08b5cc3
1 parent 32ed601
commit 08b5cc3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/google/auth/_refresh_worker.py b/google/auth/_refresh_worker.py
@@ -75,7 +75,7 @@ def __getstate__(self):
 
     def __setstate__(self, state):
         """Pickle helper that deserializes the _lock attribute."""
-        state["_key"] = threading.Lock()
+        state["_lock"] = threading.Lock()
         self.__dict__.update(state)
 
 

diff --git a/system_tests/secrets.tar.enc b/system_tests/secrets.tar.enc
diff --git a/tests/test__refresh_worker.py b/tests/test__refresh_worker.py
@@ -150,7 +150,10 @@ def test_refresh_dead_worker():
 
 def test_pickle():
     w = _refresh_worker.RefreshThreadManager()
+    # For some reason isinstance cannot interpret threading.Lock as a type.
+    assert w._lock is not None
 
     pickled_manager = pickle.dumps(w)
     manager = pickle.loads(pickled_manager)
     assert isinstance(manager, _refresh_worker.RefreshThreadManager)
+    assert manager._lock is not None