gh-136394: Fix race condition in test_zstd (GH-136432)

This commit is contained in:
Rogdham 2025-07-10 14:47:27 +02:00 committed by GitHub
parent d754f75f42
commit f519918ec6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2673,8 +2673,12 @@ def test_compress_locking(self):
input = b'a'* (16*_1K)
num_threads = 8
# gh-136394: the first output of .compress() includes the frame header
# we run the first .compress() call outside of the threaded portion
# to make the test order-independent
comp = ZstdCompressor()
parts = []
parts = [comp.compress(input, ZstdCompressor.FLUSH_BLOCK)]
for _ in range(num_threads):
res = comp.compress(input, ZstdCompressor.FLUSH_BLOCK)
if res:
@ -2683,7 +2687,7 @@ def test_compress_locking(self):
expected = b''.join(parts) + rest1
comp = ZstdCompressor()
output = []
output = [comp.compress(input, ZstdCompressor.FLUSH_BLOCK)]
def run_method(method, input_data, output_data):
res = method(input_data, ZstdCompressor.FLUSH_BLOCK)
if res: