31 lines
964 B
Python
31 lines
964 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# The LLVM Compiler Infrastructure
|
||
|
|
#
|
||
|
|
# This file is distributed under the University of Illinois Open Source
|
||
|
|
# License. See LICENSE.TXT for details.
|
||
|
|
|
||
|
|
import libear as sut
|
||
|
|
import unittest
|
||
|
|
import os.path
|
||
|
|
|
||
|
|
|
||
|
|
class TemporaryDirectoryTest(unittest.TestCase):
|
||
|
|
def test_creates_directory(self):
|
||
|
|
dirname = None
|
||
|
|
with sut.TemporaryDirectory() as tmpdir:
|
||
|
|
self.assertTrue(os.path.isdir(tmpdir))
|
||
|
|
dirname = tmpdir
|
||
|
|
self.assertIsNotNone(dirname)
|
||
|
|
self.assertFalse(os.path.exists(dirname))
|
||
|
|
|
||
|
|
def test_removes_directory_when_exception(self):
|
||
|
|
dirname = None
|
||
|
|
try:
|
||
|
|
with sut.TemporaryDirectory() as tmpdir:
|
||
|
|
self.assertTrue(os.path.isdir(tmpdir))
|
||
|
|
dirname = tmpdir
|
||
|
|
raise RuntimeError('message')
|
||
|
|
except:
|
||
|
|
self.assertIsNotNone(dirname)
|
||
|
|
self.assertFalse(os.path.exists(dirname))
|