• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• pyfilesystem2-master
  • .github
    • FUNDING.yml
  • CONTRIBUTORS.md
  • Makefile
  • fs
    • permissions.py
    • memoryfs.py
    • appfs.py
    • filesize.py
    • copy.py
    • move.py
    • iotools.py
    • path.py
    • wrap.py
    • py.typed
    • _repr.py
    • zipfs.py
    • subfs.py
    • _ftp_parse.py
    • tarfs.py
    • _bulk.py
    • tools.py
    • error_tools.py
    • info.py
    • _url_tools.py
    • multifs.py
    • constants.py
    • mirror.py
    • ftpfs.py
    • lrucache.py
    • osfs.py
    • wildcard.py
    • glob.py
    • __init__.py
    • compress.py
    • opener
      • memoryfs.py
      • appfs.py
      • parse.py
      • py.typed
      • zipfs.py
      • tarfs.py
      • ftpfs.py
      • registry.py
      • osfs.py
      • __init__.py
      • errors.py
      • tempfs.py
      • base.py
    • _version.py
    • errors.py
    • tree.py
    • test.py
    • mode.py
    • enums.py
    • _fscompat.py
    • walk.py
    • mountfs.py
    • tempfs.py
    • _typing.py
    • time.py
    • wrapfs.py
    • base.py
  • examples
    • find_dups.py
    • upload.py
    • README.txt
    • count_py.py
    • rm_pyc.py
  • pull_request_template.md
  • LICENSE
  • appveyor.yml
  • CONTRIBUTING.md
  • CHANGELOG.md
  • setup.py
  • requirements-readthedocs.txt
  • setup.cfg
  • .travis.yml
  • README.md
  • tests
    • test_encoding.py
    • test_bulk.py
    • test_enums.py
    • test_base.py
    • test_errors.py
    • test_permissions.py
    • test_iotools.py
    • test_walk.py
    • test_move.py
    • test_ftp_parse.py
    • test_osfs.py
    • test_filesize.py
    • test_tarfs.py
    • test_appfs.py
    • test_wildcard.py
    • test_tempfs.py
    • test_memoryfs.py
    • test_url_tools.py
    • test_copy.py
    • test_error_tools.py
    • test_archives.py
    • test_mirror.py
    • test_imports.py
    • test_zipfs.py
    • test_lrucache.py
    • test_fscompat.py
    • test_ftpfs.py
    • test_time.py
    • test_new_name.py
    • test_mode.py
    • test_path.py
    • test_wrap.py
    • test_multifs.py
    • test_info.py
    • __init__.py
    • test_tools.py
    • test_subfs.py
    • test_wrapfs.py
    • test_tree.py
    • test_mountfs.py
    • test_opener.py
    • conftest.py
    • test_glob.py
  • testrequirements.txt
  • .gitignore
  • docs
    • Makefile
    • source
      • interface.rst
      • info.rst
      • globbing.rst
      • reference.rst
      • introduction.rst
      • openers.rst
      • external.rst
      • builtin.rst
      • reference
        • wildcard.rst
        • path.rst
        • glob.rst
        • mountfs.rst
        • ftpfs.rst
        • errors.rst
        • opener.rst
        • multifs.rst
        • memoryfs.rst
        • wrapfs.rst
        • compress.rst
        • osfs.rst
        • zipfs.rst
        • permissions.rst
        • tree.rst
        • mirror.rst
        • wrap.rst
        • copy.rst
        • walk.rst
        • enums.rst
        • base.rst
        • appfs.rst
        • filesize.rst
        • tools.rst
        • tempfs.rst
        • tarfs.rst
        • mode.rst
        • move.rst
        • subfs.rst
        • info_objects.rst
      • guide.rst
      • extension.rst
      • implementers.rst
      • index.rst
      • conf.py
      • walking.rst
      • concepts.rst
    • make.bat
    • editdocs.sh
    • tree.html
  • MANIFEST.in
  • tox.ini

from __future__ import unicode_literals

import posixpath
import unittest

import pytest

from fs import memoryfs
from fs.test import FSTestCases
from fs.test import UNICODE_TEXT

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
                )

    @pytest.mark.skipif(
        not 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),
        )