"""

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()

