1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
# fixtures: Fixtures with cleanups for testing and convenience.
#
# Copyright (c) 2010, Robert Collins <robertc@robertcollins.net>
#
# Licensed under either the Apache License, Version 2.0 or the BSD 3-clause
# license at the users choice. A copy of both licenses are available in the
# project source as Apache-2.0 and BSD. You may not use this file except in
# compliance with one of these two licences.
#
# Unless required by applicable law or agreed to in writing, software
# distributed under these licenses is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# license you chose for the specific language governing permissions and
# limitations under that license.
import os
import tempfile
import testtools
from testtools.matchers import StartsWith
from fixtures import (
NestedTempfile,
TempDir,
)
class TestTempDir(testtools.TestCase):
def test_basic(self):
fixture = TempDir()
sentinel = object()
self.assertEqual(sentinel, getattr(fixture, 'path', sentinel))
fixture.setUp()
try:
path = fixture.path
self.assertTrue(os.path.isdir(path))
finally:
fixture.cleanUp()
self.assertFalse(os.path.isdir(path))
def test_under_dir(self):
root = self.useFixture(TempDir()).path
fixture = TempDir(root)
fixture.setUp()
with fixture:
self.assertThat(fixture.path, StartsWith(root))
def test_join(self):
temp_dir = self.useFixture(TempDir())
root = temp_dir.path
relpath = 'foo/bar/baz'
self.assertEqual(
os.path.join(root, relpath), temp_dir.join(relpath))
def test_join_multiple_children(self):
temp_dir = self.useFixture(TempDir())
root = temp_dir.path
self.assertEqual(
os.path.join(root, 'foo', 'bar', 'baz'),
temp_dir.join('foo', 'bar', 'baz'))
def test_join_naughty_children(self):
temp_dir = self.useFixture(TempDir())
root = temp_dir.path
self.assertEqual(
os.path.abspath(os.path.join(root, '..', 'bar', 'baz')),
temp_dir.join('..', 'bar', 'baz'))
class NestedTempfileTest(testtools.TestCase):
"""Tests for `NestedTempfile`."""
def test_normal(self):
# The temp directory is removed when the context is exited.
starting_tempdir = tempfile.gettempdir()
with NestedTempfile():
self.assertEqual(tempfile.tempdir, tempfile.gettempdir())
self.assertNotEqual(starting_tempdir, tempfile.tempdir)
self.assertTrue(os.path.isdir(tempfile.tempdir))
nested_tempdir = tempfile.tempdir
self.assertEqual(tempfile.tempdir, tempfile.gettempdir())
self.assertEqual(starting_tempdir, tempfile.tempdir)
self.assertFalse(os.path.isdir(nested_tempdir))
def test_exception(self):
# The temp directory is removed when the context is exited, even if
# the code running in context raises an exception.
class ContrivedException(Exception):
pass
try:
with NestedTempfile():
nested_tempdir = tempfile.tempdir
raise ContrivedException
except ContrivedException:
self.assertFalse(os.path.isdir(nested_tempdir))
|
