def test_use_as_cleanup(self):
def test_z():
return 'replacement z'
compare(sample1.z(), 'original z')
replace = Replacer()
compare(sample1.z(), 'original z')
replace('testfixtures.tests.sample1.z', test_z)
cleanup = replace.restore
try:
compare(sample1.z(), 'replacement z')
finally:
cleanup()
compare(sample1.z(), 'original z')
def test_use_as_cleanup(self):
def test_z():
return 'replacement z'
compare(sample1.z(), 'original z')
replace = Replacer()
compare(sample1.z(), 'original z')
replace('testfixtures.tests.sample1.z', test_z)
cleanup = replace.restore
try:
compare(sample1.z(), 'replacement z')
finally:
cleanup()
compare(sample1.z(), 'original z')
def test_multiple_context_managers(self):
def test_y(self):
return 'test y'
def test_z():
return 'test z'
compare(sample1.z(), 'original z')
compare(sample1.X().y(), 'original y')
with Replacer() as replace:
z = replace('testfixtures.tests.sample1.z', test_z)
y = replace('testfixtures.tests.sample1.X.y', test_y)
compare(z(), 'test z')
if PY3:
compare(y, sample1.X.y)
compare(sample1.X().y(), 'test y')
compare(sample1.z(), 'test z')
compare(sample1.X().y(), 'test y')
compare(sample1.z(), 'original z')
compare(sample1.X().y(), 'original y')
def test_multiple_context_managers(self):
def test_y(self):
return 'test y'
def test_z():
return 'test z'
compare(sample1.z(), 'original z')
compare(sample1.X().y(), 'original y')
with Replacer() as replace:
z = replace('testfixtures.tests.sample1.z', test_z)
y = replace('testfixtures.tests.sample1.X.y', test_y)
compare(z(), 'test z')
if PY3:
compare(y, sample1.X.y)
compare(sample1.X().y(), 'test y')
compare(sample1.z(), 'test z')
compare(sample1.X().y(), 'test y')
compare(sample1.z(), 'original z')
compare(sample1.X().y(), 'original y')
