Lib/test/test_threadedtempfile.py

   1 """
   2 Create and delete FILES_PER_THREAD temp files (via tempfile.TemporaryFile)
   3 in each of NUM_THREADS threads, recording the number of successes and
   4 failures.  A failure is a bug in tempfile, and may be due to:
   5
   6 + Trying to create more than one tempfile with the same name.
   7 + Trying to delete a tempfile that doesn't still exist.
   8 + Something we've never seen before.
   9
  10 By default, NUM_THREADS == 20 and FILES_PER_THREAD == 50.  This is enough to
  11 create about 150 failures per run under Win98SE in 2.0, and runs pretty
  12 quickly. Guido reports needing to boost FILES_PER_THREAD to 500 before
  13 provoking a 2.0 failure under Linux.
  14 """
  15
  16 NUM_THREADS = 20
  17 FILES_PER_THREAD = 50
  18
  19 import tempfile
  20
  21 from test.test_support import threading_setup, threading_cleanup, run_unittest, import_module
  22 threading = import_module('threading')
  23 import unittest
  24 import StringIO
  25 from traceback import print_exc
  26
  27 startEvent = threading.Event()
  28
  29 class TempFileGreedy(threading.Thread):
  30     error_count = 0
  31     ok_count = 0
  32
  33     def run(self):
  34         self.errors = StringIO.StringIO()
  35         startEvent.wait()
  36         for i in range(FILES_PER_THREAD):
  37             try:
  38                 f = tempfile.TemporaryFile("w+b")
  39                 f.close()
  40             except:
  41                 self.error_count += 1
  42                 print_exc(file=self.errors)
  43             else:
  44                 self.ok_count += 1
  45
  46
  47 class ThreadedTempFileTest(unittest.TestCase):
  48     def test_main(self):
  49         threads = []
  50         thread_info = threading_setup()
  51
  52         for i in range(NUM_THREADS):
  53             t = TempFileGreedy()
  54             threads.append(t)
  55             t.start()
  56
  57         startEvent.set()
  58
  59         ok = 0
  60         errors = []
  61         for t in threads:
  62             t.join()
  63             ok += t.ok_count
  64             if t.error_count:
  65                 errors.append(str(t.getName()) + str(t.errors.getvalue()))
  66
  67         threading_cleanup(*thread_info)
  68
  69         msg = "Errors: errors %d ok %d\n%s" % (len(errors), ok,
  70             '\n'.join(errors))
  71         self.assertEquals(errors, [], msg)
  72         self.assertEquals(ok, NUM_THREADS * FILES_PER_THREAD)
  73
  74 def test_main():
  75     run_unittest(ThreadedTempFileTest)
  76
  77 if __name__ == "__main__":
  78     test_main()