blob: 9a84c488cf4ebce668176c37b8b007883b943d50 [file] [log] [blame] [edit]
#!/usr/bin/python
import unittest, os
import common
from autotest_lib.client.common_lib import autotemp
class tempfile_test(unittest.TestCase):
def test_create_file(self):
temp = autotemp.tempfile(unique_id='file')
self.assertTrue(os.path.exists(temp.name))
def test_clean(self):
temp = autotemp.tempfile(unique_id='clean')
# clean up sets name to None so we preserve it this way
name = temp.name
self.assertTrue(os.path.exists(name))
temp.clean()
self.assertFalse(os.path.exists(name))
def test_del(self):
tmp_file = autotemp.tempfile(unique_id='del')
name = tmp_file.name
self.assertTrue(os.path.exists(name))
tmp_file.__del__()
self.assertFalse(os.path.exists(name))
class tempdir(unittest.TestCase):
def test_create_dir(self):
temp_dir = autotemp.tempdir(unique_id='dir')
self.assertTrue(os.path.exists(temp_dir.name))
self.assertTrue(os.path.isdir(temp_dir.name))
def test_clean(self):
temp_dir = autotemp.tempdir(unique_id='clean')
name = temp_dir.name
self.assertTrue(os.path.exists(name))
temp_dir.clean()
self.assertFalse(os.path.exists(name))
def test_del(self):
temp_dir = autotemp.tempdir(unique_id='del')
name = temp_dir.name
self.assertTrue(os.path.exists(name))
temp_dir.__del__()
self.assertFalse(os.path.exists(name))
if __name__ == '__main__':
unittest.main()