added test and test_hashlib to list of tsan tests

python · gpshead · Feb 8, 2025 · Jan 15, 2025 · Jan 15, 2025 · Jan 15, 2025
commit 35fd4ba04bc8c0d2d69bdbf32eedf5bfb3c5b402
diff --git a/Lib/test/libregrtest/tsan.py b/Lib/test/libregrtest/tsan.py
@@ -8,6 +8,7 @@
     'test_code',
     'test_enum',
     'test_functools',
+    'test_hashlib',
     'test_httpservers',
     'test_imaplib',
     'test_importlib',

diff --git a/Lib/test/test_hashlib.py b/Lib/test/test_hashlib.py
@@ -1196,6 +1196,29 @@ def test_file_digest(self):
             with open(os_helper.TESTFN, "wb") as f:
                 hashlib.file_digest(f, "sha256")
 
+    @unittest.skipUnless(support.check_sanitizer(thread=True), "only meaningful on free-threading")
+    def test_gh_128657(self):
+        def test_hashlib_sha256():
+            hash_obj = hashlib.sha256()
+
+        def closure(barrier):
+            barrier.wait()
+            test_hashlib_sha256()
+
+        num_workers = 40
+        num_runs = 20
+
+        for i in range(num_runs):
+            barrier = threading.Barrier(num_workers)
+            thrds = []
+
+            for i in range(num_workers):
+                thrds.append(thrd := threading.Thread(target=closure, args=(barrier,)))
+                thrd.start()
+
+            for thrd in thrds:
+                thrd.join()
+
 
 if __name__ == "__main__":
     unittest.main()