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