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
|
import types
import paver.deps.six as six
from paver import setuputils, tasks
class FakeModule(object):
def __init__(self, **kw):
for name, value in kw.items():
setattr(self, name, value)
def patched_print(self, output):
self.patch_captured.append(output)
class FakeExitException(Exception):
""" Fake out tasks.Environment._exit to avoid interupting tests """
def patched_exit(self, code):
self.exit_code = 1
raise FakeExitException(code)
def _set_environment(patch_print=False, **kw):
pavement = FakeModule(**kw)
env = tasks.Environment(pavement)
tasks.environment = env
if six.PY3:
method_args = (env,)
else:
method_args = (env, tasks.Environment)
env._exit = types.MethodType(patched_exit, *method_args)
if patch_print:
env._print = types.MethodType(patched_print, *method_args)
env.patch_captured = []
return env
|
