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
|
from __future__ import unicode_literals
import posixpath
import unittest
from fs import memoryfs
from fs.test import UNICODE_TEXT, FSTestCases
try:
# Only supported on Python 3.4+
import tracemalloc
except ImportError:
tracemalloc = None
class TestMemoryFS(FSTestCases, unittest.TestCase):
"""Test OSFS implementation."""
def make_fs(self):
return memoryfs.MemoryFS()
def _create_many_files(self):
for parent_dir in {"/", "/one", "/one/two", "/one/other-two/three"}:
self.fs.makedirs(parent_dir, recreate=True)
for file_id in range(50):
self.fs.writetext(
posixpath.join(parent_dir, str(file_id)), UNICODE_TEXT
)
@unittest.skipUnless(
tracemalloc, reason="`tracemalloc` isn't supported on this Python version."
)
def test_close_mem_free(self):
"""Ensure all file memory is freed when calling close().
Prevents regression against issue #308.
"""
trace_filters = [tracemalloc.Filter(True, "*/memoryfs.py")]
tracemalloc.start()
before = tracemalloc.take_snapshot().filter_traces(trace_filters)
self._create_many_files()
after_create = tracemalloc.take_snapshot().filter_traces(trace_filters)
self.fs.close()
after_close = tracemalloc.take_snapshot().filter_traces(trace_filters)
tracemalloc.stop()
[diff_create] = after_create.compare_to(
before, key_type="filename", cumulative=True
)
self.assertGreater(
diff_create.size_diff,
0,
"Memory usage didn't increase after creating files; diff is %0.2f KiB."
% (diff_create.size_diff / 1024.0),
)
[diff_close] = after_close.compare_to(
after_create, key_type="filename", cumulative=True
)
self.assertLess(
diff_close.size_diff,
0,
"Memory usage increased after closing the file system; diff is %0.2f KiB."
% (diff_close.size_diff / 1024.0),
)
def test_copy_preserve_time(self):
self.fs.makedir("foo")
self.fs.makedir("bar")
self.fs.touch("foo/file.txt")
src_datetime = self.fs.getmodified("foo/file.txt")
self.fs.copy("foo/file.txt", "bar/file.txt", preserve_time=True)
self.assertTrue(self.fs.exists("bar/file.txt"))
dst_datetime = self.fs.getmodified("bar/file.txt")
self.assertEqual(dst_datetime, src_datetime)
class TestMemoryFile(unittest.TestCase):
def setUp(self):
self.fs = memoryfs.MemoryFS()
def tearDown(self):
self.fs.close()
def test_readline_writing(self):
with self.fs.openbin("test.txt", "w") as f:
self.assertRaises(IOError, f.readline)
def test_readinto_writing(self):
with self.fs.openbin("test.txt", "w") as f:
self.assertRaises(IOError, f.readinto, bytearray(10))
|
