/usr/share/pyshared/schooltool/testing/mock.py is in python-schooltool 1:2.1.0-0ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 | #
# SchoolTool - common information systems platform for school administration
# Copyright (c) 2010 Shuttleworth Foundation
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
import sys, imp
from zope.testing import cleanup
class ModulesSnapshot(object):
def __init__(self):
self.originals = dict(sys.modules)
self.faked_attrs = {}
self.imported = set()
def mock(self, values):
"""Usage:
mock({'my.module.Foo': FooClass,
'my.module.another.GLOBAL_VAL': 1,
})
"""
for full_name in sorted(values):
names = full_name.split('.')
attr_name = names[-1]
module_name = '.'.join(names[:-1])
assert module_name
self.mock_attr(module_name, attr_name, fake=values[full_name])
def mock_attr(self, modulename, name, fake=None):
module = self.get_module(modulename)
if modulename not in self.faked_attrs:
self.faked_attrs[modulename] = {
'injected': [],
'replaced': {},
}
faked = self.faked_attrs[modulename]
if hasattr(module, name):
if (name not in faked['injected'] and
name not in faked['replaced']):
faked['replaced'][name] = getattr(module, name)
else:
faked['injected'].append(name)
setattr(module, name, fake)
def mock_module(self, modulename, fake=None):
if fake is None:
fake = imp.new_module(modulename)
names = modulename.split('.')
this_name = names[-1]
parent_name = '.'.join(names[:-1])
if parent_name:
self.mock_attr(parent_name, this_name, fake)
sys.modules[modulename] = fake
self.imported.add(modulename)
def get_module(self, modulename):
try:
sys.modules[modulename]
except:
self.mock_module(modulename)
return sys.modules[modulename]
def restore(self):
for module_name, faked in self.faked_attrs.items():
try:
module = sys.modules[module_name]
except:
continue
for name in faked['injected']:
del module.__dict__[name]
for name, old_value in faked['replaced'].items():
setattr(module, name, old_value)
for name in list(sys.modules):
if name in self.imported:
del sys.modules[name]
elif name in self.originals:
sys.modules[name] = self.originals[name]
_snapshot = None
def getSnapshot():
global _snapshot
if _snapshot is None:
_snapshot = ModulesSnapshot()
return _snapshot
def restoreModules():
global _snapshot
if _snapshot is not None:
_snapshot.restore()
_snapshot = None
cleanup.addCleanUp(restoreModules)
def module(module_name):
"""A decorator to put method or class to the given module."""
snapshot = getSnapshot()
if type(module_name) == type(sys):
module_name = module_name.__name__
def mock_something(something):
name = getattr(something, '__name__', None)
snapshot.mock_attr(module_name, name, fake=something)
return something
return mock_something
def fake_global(module_name, name, value):
"""Set a global variable to a module."""
snapshot = getSnapshot()
if type(module_name) == type(sys):
module_name = module_name.__name__
snapshot.mock_attr(module_name, name, fake=value)
def fake_module(module_name, other=None):
"""Replace a module with a fake one."""
snapshot = getSnapshot()
if type(module_name) == type(sys):
module_name = module_name.__name__
snapshot.mock_module(module_name, fake=other)
|