80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
"""
|
|
Create and delete FILES_PER_THREAD temp files (via tempfile.TemporaryFile)
|
|
in each of NUM_THREADS threads, recording the number of successes and
|
|
failures. A failure is a bug in tempfile, and may be due to:
|
|
|
|
+ Trying to create more than one tempfile with the same name.
|
|
+ Trying to delete a tempfile that doesn't still exist.
|
|
+ Something we've never seen before.
|
|
|
|
By default, NUM_THREADS == 20 and FILES_PER_THREAD == 50. This is enough to
|
|
create about 150 failures per run under Win98SE in 2.0, and runs pretty
|
|
quickly. Guido reports needing to boost FILES_PER_THREAD to 500 before
|
|
provoking a 2.0 failure under Linux.
|
|
"""
|
|
|
|
NUM_THREADS = 20
|
|
FILES_PER_THREAD = 50
|
|
|
|
import thread # If this fails, we can't test this module
|
|
import threading
|
|
import tempfile
|
|
|
|
from test.test_support import threading_setup, threading_cleanup, run_unittest
|
|
import unittest
|
|
import StringIO
|
|
from traceback import print_exc
|
|
|
|
startEvent = threading.Event()
|
|
|
|
class TempFileGreedy(threading.Thread):
|
|
error_count = 0
|
|
ok_count = 0
|
|
|
|
def run(self):
|
|
self.errors = StringIO.StringIO()
|
|
startEvent.wait()
|
|
for i in range(FILES_PER_THREAD):
|
|
try:
|
|
f = tempfile.TemporaryFile("w+b")
|
|
f.close()
|
|
except:
|
|
self.error_count += 1
|
|
print_exc(file=self.errors)
|
|
else:
|
|
self.ok_count += 1
|
|
|
|
|
|
class ThreadedTempFileTest(unittest.TestCase):
|
|
def test_main(self):
|
|
threads = []
|
|
thread_info = threading_setup()
|
|
|
|
for i in range(NUM_THREADS):
|
|
t = TempFileGreedy()
|
|
threads.append(t)
|
|
t.start()
|
|
|
|
startEvent.set()
|
|
|
|
ok = 0
|
|
errors = []
|
|
for t in threads:
|
|
t.join()
|
|
ok += t.ok_count
|
|
if t.error_count:
|
|
errors.append(str(t.getName()) + str(t.errors.getvalue()))
|
|
|
|
threading_cleanup(*thread_info)
|
|
|
|
msg = "Errors: errors %d ok %d\n%s" % (len(errors), ok,
|
|
'\n'.join(errors))
|
|
self.assertEquals(errors, [], msg)
|
|
self.assertEquals(ok, NUM_THREADS * FILES_PER_THREAD)
|
|
|
|
def test_main():
|
|
run_unittest(ThreadedTempFileTest)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|