Fix os.environb

python · Jun 5, 2024 · df97d03 · df97d03
1 parent df06534
commit df97d03
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/Lib/os.py b/Lib/os.py
@@ -790,7 +790,10 @@ def refresh(self):
                     data[encodekey(key)] = value
             else:
                 data = environ
-            self._data = data
+
+            # modify in-place to keep os.environb in sync
+            self._data.clear()
+            self._data.update(data)
 
 def _create_environ_mapping():
     if name == 'nt':

diff --git a/Lib/test/test_os.py b/Lib/test/test_os.py
@@ -1310,9 +1310,13 @@ def test_refresh(self):
         os.environ['test_env'] = 'python_value'
         putenv("test_env", "new_value")
         self.assertEqual(os.environ['test_env'], 'python_value')
+        if hasattr(os, 'environb'):
+            self.assertEqual(os.environb[b'test_env'], b'python_value')
 
         os.environ.refresh()
         self.assertEqual(os.environ['test_env'], 'new_value')
+        if hasattr(os, 'environb'):
+            self.assertEqual(os.environb[b'test_env'], b'new_value')
 
 
 class WalkTests(unittest.TestCase):