You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.5 KiB
58 lines
1.5 KiB
#!/usr/bin/python2
|
|
|
|
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()
|