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 test_main():
    import gc
    gc.disable()
    gc.set_debug(gc.DEBUG_LEAK)
    if not os.path.isdir(TEMPDIR):
        os.mkdir(TEMPDIR)
    try:
        unittest.main()
    finally:
        print("===========================")
        gc.collect()
        print("===========================")
        shutil.rmtree(TEMPDIR)

if __name__ == '__main__':
    test_main()