xref: /third_party/python/Lib/test/test_tempfile.py

98     def nameCheck(self, name, dir, pre, suf):
103         if dir is not None:
107                 if type(dir) is str or isinstance(dir, os.PathLike) else
117         if (dir, pre, suf) == (None, None, None):
121         self.assertEqual(os.path.abspath(ndir), os.path.abspath(dir),
122                          "file %r not in directory %r" % (name, dir))
320     dir = tempfile.mkdtemp()
322         with support.swap_attr(tempfile, 'tempdir', dir):
325         os_helper.rmtree(dir)
378         def __init__(self, dir, pre, suf, bin):
382             output_type = tempfile._infer_return_type(dir, pre, suf)
383             (self.fd, self.name) = tempfile._mkstemp_inner(dir, pre, suf, flags, output_type)
392     def do_create(self, dir=None, pre=None, suf=None, bin=1):
393         output_type = tempfile._infer_return_type(dir, pre, suf)
394         if dir is None:
396                 dir = tempfile.gettempdir()
398                 dir = tempfile.gettempdirb()
403         file = self.mkstemped(dir, pre, suf, bin)
405         self.nameCheck(file.name, dir, pre, suf)
420         self.do_create(dir=dir_b, suf=b"").write(b"blat")
421         self.do_create(dir=dir_b, pre=b"a").write(b"blat")
422         self.do_create(dir=dir_b, suf=b"b").write(b"blat")
423         self.do_create(dir=dir_b, pre=b"a", suf=b"b").write(b"blat")
424         self.do_create(dir=dir_b, pre=b"aa", suf=b".txt").write(b"blat")
427             self.do_create(dir="", suf=b"").write(b"blat")
429             self.do_create(dir=dir_b, pre="").write(b"blat")
431             self.do_create(dir=dir_b, pre=b"", suf="").write(b"blat")
441         dir = tempfile.mkdtemp()
443             self.do_create(dir=dir).write(b"blat")
444             self.do_create(dir=pathlib.Path(dir)).write(b"blat")
447             os.rmdir(dir)
538             dir = tempfile.mkdtemp()
539             self.assertTrue(dir.endswith('aaa'))
627     def do_create(self, dir=None, pre=None, suf=None):
628         output_type = tempfile._infer_return_type(dir, pre, suf)
629         if dir is None:
631                 dir = tempfile.gettempdir()
633                 dir = tempfile.gettempdirb()
638         (fd, name) = tempfile.mkstemp(dir=dir, prefix=pre, suffix=suf)
640         adir = os.path.abspath(dir)
645             self.nameCheck(name, dir, pre, suf)
657         self.do_create(dir=".")
663         self.do_create(dir=d, suf=b"")
664         self.do_create(dir=d, pre=b"a")
665         self.do_create(dir=d, suf=b"b")
666         self.do_create(dir=d, pre=b"a", suf=b"b")
667         self.do_create(dir=d, pre=b"aa", suf=b".txt")
668         self.do_create(dir=b".")
670             self.do_create(dir=".", pre=b"aa", suf=b".txt")
672             self.do_create(dir=b".", pre="aa", suf=b".txt")
674             self.do_create(dir=b".", pre=b"aa", suf=".txt")
679         dir = tempfile.mkdtemp()
681             self.do_create(dir=dir)
682             self.do_create(dir=pathlib.Path(dir))
684             os.rmdir(dir)
710             fd, path = tempfile.mkstemp(dir=tempfile.gettempdir())
724     def do_create(self, dir=None, pre=None, suf=None):
725         output_type = tempfile._infer_return_type(dir, pre, suf)
726         if dir is None:
728                 dir = tempfile.gettempdir()
730                 dir = tempfile.gettempdirb()
735         name = tempfile.mkdtemp(dir=dir, prefix=pre, suffix=suf)
738             self.nameCheck(name, dir, pre, suf)
755         os.rmdir(self.do_create(dir=d))
756         os.rmdir(self.do_create(dir=d, pre=b"a"))
757         os.rmdir(self.do_create(dir=d, suf=b"b"))
758         os.rmdir(self.do_create(dir=d, pre=b"a", suf=b"b"))
759         os.rmdir(self.do_create(dir=d, pre=b"aa", suf=b".txt"))
761             os.rmdir(self.do_create(dir=d, pre="aa", suf=b".txt"))
763             os.rmdir(self.do_create(dir=d, pre=b"aa", suf=".txt"))
765             os.rmdir(self.do_create(dir="", pre=b"aa", suf=b".txt"))
780         dir = tempfile.mkdtemp()
782             os.rmdir(self.do_create(dir=dir))
783             os.rmdir(self.do_create(dir=pathlib.Path(dir)))
785             os.rmdir(dir)
791         dir = self.do_create()
793             mode = stat.S_IMODE(os.stat(dir).st_mode)
803             os.rmdir(dir)
811         dir = self.do_create()
813             out = subprocess.check_output(["icacls.exe", dir], encoding="oem").casefold()
815             os.rmdir(dir)
817         dir = dir.casefold()
823             if line.startswith(dir):
824                 acl = line.removeprefix(dir).strip()
840             dir = tempfile.mkdtemp()
841             self.assertTrue(dir.endswith('bbb'))
867             path = tempfile.mkdtemp(suffix='-dir')
873             path = tempfile.mkdtemp(dir=tempfile.gettempdir())
886         self.dir = tempfile.mkdtemp()
890         if self.dir:
891             os.rmdir(self.dir)
892             self.dir = None
899         def __init__(self, dir, pre, suf):
900             self.name = tempfile.mktemp(dir=dir, prefix=pre, suffix=suf)
909         file = self.mktemped(self.dir, pre, suf)
911         self.nameCheck(file.name, self.dir, pre, suf)
936 ##                           tempfile.mktemp, dir=self.dir)
945     def do_create(self, dir=None, pre="", suf="", delete=True):
946         if dir is None:
947             dir = tempfile.gettempdir()
948         file = tempfile.NamedTemporaryFile(dir=dir, prefix=pre, suffix=suf,
951         self.nameCheck(file.name, dir, pre, suf)
1000         dir = tempfile.mkdtemp()
1002             with tempfile.NamedTemporaryFile(dir=dir) as f:
1004             self.assertEqual(os.listdir(dir), [])
1008             os.rmdir(dir)
1012         dir = tempfile.mkdtemp()
1015             f = tempfile.NamedTemporaryFile(dir=dir, delete=False)
1024             os.rmdir(dir)
1045         dir = tempfile.mkdtemp()
1046         self.addCleanup(os_helper.rmtree, dir)
1048             tempfile.NamedTemporaryFile(mode='wr', dir=dir)
1050             tempfile.NamedTemporaryFile(mode=2, dir=dir)
1051         self.assertEqual(os.listdir(dir), [])
1054         dir = tempfile.mkdtemp()
1055         self.addCleanup(os_helper.rmtree, dir)
1057             tempfile.NamedTemporaryFile('w', encoding='bad-encoding', dir=dir)
1058         self.assertEqual(os.listdir(dir), [])
1061         dir = tempfile.mkdtemp()
1062         self.addCleanup(os_helper.rmtree, dir)
1067                 tempfile.NamedTemporaryFile(dir=dir)
1069         self.assertEqual(os.listdir(dir), [])
1076     def do_create(self, max_size=0, dir=None, pre="", suf=""):
1077         if dir is None:
1078             dir = tempfile.gettempdir()
1079         file = tempfile.SpooledTemporaryFile(max_size=max_size, dir=dir, prefix=pre, suffix=suf)
1108         spooledtempfile_attrs = set(dir(tempfile.SpooledTemporaryFile))
1117         dir = tempfile.mkdtemp()
1119             f = tempfile.SpooledTemporaryFile(max_size=10, dir=dir)
1125             self.assertEqual(os.listdir(dir), [])
1130             os.rmdir(dir)
1409             dir = tempfile.mkdtemp()
1410             f = tempfile.TemporaryFile(dir=dir)
1416                 os.rmdir(dir)
1420                 os.rmdir(dir)
1446             dir = tempfile.mkdtemp()
1447             self.addCleanup(os_helper.rmtree, dir)
1449                 tempfile.TemporaryFile(mode='wr', dir=dir)
1451                 tempfile.TemporaryFile(mode=2, dir=dir)
1452             self.assertEqual(os.listdir(dir), [])
1455             dir = tempfile.mkdtemp()
1456             self.addCleanup(os_helper.rmtree, dir)
1458                 tempfile.TemporaryFile('w', encoding='bad-encoding', dir=dir)
1459             self.assertEqual(os.listdir(dir), [])
1462             dir = tempfile.mkdtemp()
1463             self.addCleanup(os_helper.rmtree, dir)
1470                     tempfile.TemporaryFile(dir=dir)
1472             self.assertEqual(os.listdir(dir), [])
1495     def do_create(self, dir=None, pre="", suf="", recurse=1, dirs=1, files=1,
1497         if dir is None:
1498             dir = tempfile.gettempdir()
1500             dir=dir, prefix=pre, suffix=suf,
1502         self.nameCheck(tmp.name, dir, pre, suf)
1510                 name = os.path.join(path, "dir%d" % i)
1524             tempfile.TemporaryDirectory(dir=nonexistent)
1529         dir = tempfile.mkdtemp()
1531             d = self.do_create(dir=dir)
1538             os.rmdir(dir)
1544                 dir=working_dir, ignore_cleanup_errors=True)
1615                     with self.subTest('nonexisting dir'):
1626                     with self.subTest('existing dir'):
1665             with self.subTest('nonexisting dir'):
1675             with self.subTest('existing dir'):
1685         dir = tempfile.mkdtemp()
1687             d = self.do_create(dir=dir)
1693             os.rmdir(dir)
1700                 dir=working_dir, ignore_cleanup_errors=True)
1718         with self.do_create() as dir:
1728                     tmp = tempfile.TemporaryDirectory(dir={dir!r})
1739                     """.format(dir=dir, mod=mod)
1758                     dir={working_dir!r}, ignore_cleanup_errors=True)
1787         with self.do_create() as dir:
1794                     with tempfile.TemporaryDirectory(dir={dir!r}) as tmp:
1800                 """.format(dir=dir)
1811         with self.do_create() as dir:
1812             d = self.do_create(dir=dir, recurse=3)

Indexes created Thu Nov 07 10:32:03 CST 2024