1 #!/usr/bin/python 2 3 import unittest, os 4 import common 5 from autotest_lib.client.common_lib import autotemp 6 7 8 class tempfile_test(unittest.TestCase): 9 10 def test_create_file(self): 11 temp = autotemp.tempfile(unique_id='file') 12 self.assertTrue(os.path.exists(temp.name)) 13 14 15 def test_clean(self): 16 temp = autotemp.tempfile(unique_id='clean') 17 # clean up sets name to None so we preserve it this way 18 name = temp.name 19 self.assertTrue(os.path.exists(name)) 20 temp.clean() 21 self.assertFalse(os.path.exists(name)) 22 23 24 def test_del(self): 25 tmp_file = autotemp.tempfile(unique_id='del') 26 name = tmp_file.name 27 self.assertTrue(os.path.exists(name)) 28 tmp_file.__del__() 29 self.assertFalse(os.path.exists(name)) 30 31 32 class tempdir(unittest.TestCase): 33 34 def test_create_dir(self): 35 temp_dir = autotemp.tempdir(unique_id='dir') 36 self.assertTrue(os.path.exists(temp_dir.name)) 37 self.assertTrue(os.path.isdir(temp_dir.name)) 38 39 40 def test_clean(self): 41 temp_dir = autotemp.tempdir(unique_id='clean') 42 name = temp_dir.name 43 self.assertTrue(os.path.exists(name)) 44 temp_dir.clean() 45 self.assertFalse(os.path.exists(name)) 46 47 48 def test_del(self): 49 temp_dir = autotemp.tempdir(unique_id='del') 50 name = temp_dir.name 51 self.assertTrue(os.path.exists(name)) 52 temp_dir.__del__() 53 self.assertFalse(os.path.exists(name)) 54 55 56 if __name__ == '__main__': 57 unittest.main() 58
