# -*- Mode:Python; indent-tabs-mode:nil; tab-width:4; encoding:utf-8 -*-
#
# Copyright 2002 Ben Escoto <ben@emerose.org>
# Copyright 2007 Kenneth Loafman <kenneth@loafman.com>
#
# This file is part of duplicity.
#
# Duplicity is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2 of the License, or (at your
# option) any later version.
#
# Duplicity is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with duplicity; if not, write to the Free Software Foundation,
# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
from __future__ import print_function
from future import standard_library
standard_library.install_aliases()
import os
import tempfile
import unittest
from duplicity import tempdir
from . import UnitTestCase
[docs]class TempDirTest(UnitTestCase):
[docs] def test_all(self):
td = tempdir.default()
# are generated temp files unique?
self.assertTrue(td.mktemp() != td.mktemp())
# create and remove a temp dir
dir = td.mktemp() # pylint: disable=redefined-builtin
os.mkdir(dir)
os.rmdir(dir)
# test mkstemp()
fd, fname = td.mkstemp()
os.close(fd)
os.unlink(fname)
td.forget(fname)
# test mkstemp_file()
fo, fname = td.mkstemp_file()
fo.close() # don't forget, leave to cleanup()
# cleanup
td.cleanup()
[docs] def test_dirname(self):
u"""
test if we generated a dirname
"""
td = tempdir.default()
dirname = td.dir()
self.assertTrue(dirname is not None)
u"""
test if duplicity's temp files are created in our temp dir
"""
f1d, f1_name = tempdir.default().mkstemp()
f1_dirname = os.path.dirname(f1_name)
self.assertTrue(dirname == f1_dirname)
u"""
test if tempfile creates in our temp dir now as well by default
"""
f2 = tempfile.NamedTemporaryFile()
f2_dirname = os.path.dirname(f2.name)
self.assertTrue(dirname == f2_dirname)
# cleanup
os.close(f1d)
os.unlink(f1_name)
td.forget(f1_name)
f2.close()
td.cleanup()
if __name__ == u"__main__":
unittest.main()