def test_it_can_stub_an_attr_from_an_obj(self):
o = Foobar(1, 2, 3)
s = stub_attr(o, 'a')
s.replace()
self.assertIs(o.a, s.value)
self.assertTrue(isinstance(s.value, Mock))
s.restore()
self.assertIsNot(o.a, s)
def test_it_can_stub_an_attr_from_an_obj_using_with(self):
o = Foobar(1, 2, 3)
with stub_attr(o, 'a') as s:
self.assertIs(o.a, s)
self.assertTrue(isinstance(s, Mock))
self.assertIsNot(o.a, s)
