pubs/tests/fake_env.py

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