import unittest
import tempfile
import tarfile
import shutil
import os

TEMPDIR = os.path.join(tempfile.gettempdir(), "__foobarbaz__")
tmpname = os.path.join(TEMPDIR, "tmp.tar")

class Bz2StreamReadTest(unittest.TestCase):
    mode = "r|bz2"
    def test_null_tarfile(self):
        with open(tmpname, "wb") as f:
            f.write(b"invalid data")
        self.assertRaises(tarfile.ReadError, tarfile.open, tmpname, self.mode)

def exc_clear():
    # clear exception stack if exist
    try:
        raise RuntimeError
    except RuntimeError:
        pass

def test_main():
    import gc
    gc.disable()
    if not os.path.isdir(TEMPDIR):
        os.mkdir(TEMPDIR)
    try:
        unittest.main()
    finally:
#        gc.collect()
#        exc_clear()
        shutil.rmtree(TEMPDIR)

if __name__ == '__main__':
    test_main()