fake_env module for fake fs, fake input

tests/fake_env.py

@@ -0,0 +1,162 @@
+import sys
+import os
+import shutil
+import glob
+import unittest
+import pkgutil
+import re
+
+import testenv
+import fake_filesystem
+import fake_filesystem_shutil
+import fake_filesystem_glob
+
+from papers import color
+from papers.p3 import io, input
+
+    # code for fake fs
+
+real_os = os
+real_open = open
+real_file = file
+real_shutil = shutil
+real_glob = glob
+
+
+
+# def _mod_list():
+#     ml = []
+#     import papers
+#     for importer, modname, ispkg in pkgutil.walk_packages(
+#                                         path=papers.__path__,
+#                                         prefix=papers.__name__ + '.',
+#                                         onerror=lambda x: None):
+#         # HACK to not load textnote
+#         if not modname.startswith('papers.plugs.texnote'):
+#             ml.append((modname, __import__(modname, fromlist='dummy')))
+#     return ml
+
+
+
+def create_fake_fs(module_list):
+
+    fake_fs = fake_filesystem.FakeFilesystem()
+    fake_os = fake_filesystem.FakeOsModule(fake_fs)
+    fake_open = fake_filesystem.FakeFileOpen(fake_fs)
+    fake_shutil = fake_filesystem_shutil.FakeShutilModule(fake_fs)
+    fake_glob = fake_filesystem_glob.FakeGlobModule(fake_fs)
+
+    fake_fs.CreateDirectory(fake_os.path.expanduser('~'))
+
+    try:
+        __builtins__.open = fake_open
+        __builtins__.file = fake_open
+    except AttributeError:
+        __builtins__['open'] = fake_open
+        __builtins__['file'] = fake_open
+
+    sys.modules['os']     = fake_os
+    sys.modules['shutil'] = fake_shutil
+    sys.modules['glob']   = fake_glob
+
+    for md in module_list:
+        md.os = fake_os
+        md.shutil = fake_shutil
+        md.open = fake_open
+        md.file = fake_open
+
+    return {'fs': fake_fs,
+            'os': fake_os,
+            'open': fake_open,
+            'shutil': fake_shutil,
+            'glob': fake_glob}
+
+def unset_fake_fs(module_list):
+    try:
+        __builtins__.open = real_open
+        __builtins__.file = real_file
+    except AttributeError:
+        __builtins__['open'] = real_open
+        __builtins__['file'] = real_file
+
+    sys.modules['os']     = real_os
+    sys.modules['shutil'] = real_shutil
+    sys.modules['glob']   = real_glob
+
+    for md in module_list:
+        md.os = real_os
+        md.shutil = real_shutil
+        md.open = real_open
+        md.file = real_file
+
+
+def copy_dir(fs, real_dir, fake_dir = None):
+    """Copy all the data directory into the fake fs"""
+    if fake_dir is None:
+        fake_dir = real_dir
+    for filename in real_os.listdir(real_dir):
+        real_path = real_os.path.join(real_dir, filename)
+        fake_path = fs['os'].path.join(fake_dir, filename)
+        if real_os.path.isfile(real_path):
+            with real_open(real_path, 'r') as f:
+                fs['fs'].CreateFile(fake_path, contents=f.read())
+        if real_os.path.isdir(real_path):
+            fs['fs'].CreateDirectory(fake_path)
+            copy_dir(fs, real_path, fake_path)
+
+
+    # redirecting output
+
+def redirect(f):
+    def newf(*args, **kwargs):
+        old_stderr, old_stdout = sys.stderr, sys.stdout
+        stdout = io.StringIO()
+        stderr = io.StringIO()
+        sys.stdout, sys.stderr = stdout, stderr
+        try:
+            return f(*args, **kwargs), stdout, stderr
+        finally:
+            sys.stderr, sys.stdout = old_stderr, old_stdout
+    return newf
+
+
+# Test helpers
+
+# automating input
+
+real_input = input
+
+
+class FakeInput():
+    """ Replace the input() command, and mock user input during tests
+
+        Instanciate as :
+        input = FakeInput(['yes', 'no'])
+        then replace the input command in every module of the package :
+        input.as_global()
+        Then :
+        input() returns 'yes'
+        input() returns 'no'
+        input() raise IndexError
+     """
+
+    def __init__(self, module_list, inputs=None):
+        self.inputs = list(inputs) or []
+        self.module_list = module_list
+        self._cursor = 0
+
+    def as_global(self):
+        for md in module_list:
+            md.input = self
+            md.editor_input = self
+            # if mdname.endswith('files'):
+            #     md.editor_input = self
+
+    def add_input(self, inp):
+        self.inputs.append(inp)
+
+    def __call__(self, *args, **kwargs):
+        inp = self.inputs[self._cursor]
+        self._cursor += 1
+        return inp
+