def test_wrap_me_init(self):
wrap = wrappers.wrap_me(before='foo', after=['bar'])
assert wrap.before == ['foo']
assert wrap.after == ['bar']
wrap = wrappers.wrap_me(after=['bar'])
assert wrap.before == []
assert wrap.after == ['bar']
wrap = wrappers.wrap_me()
assert wrap.before == []
assert wrap.after == []
def test_wrap_me_init(self):
wrap = wrappers.wrap_me(before='foo', after=['bar'])
assert wrap.before == ['foo']
assert wrap.after == ['bar']
wrap = wrappers.wrap_me(after=['bar'])
assert wrap.before == []
assert wrap.after == ['bar']
wrap = wrappers.wrap_me()
assert wrap.before == []
assert wrap.after == []
def test_wrap_me_init(self):
wrap = wrappers.wrap_me(before="foo", after=["bar"])
assert wrap.before == ["foo"]
assert wrap.after == ["bar"]
wrap = wrappers.wrap_me(after=["bar"])
assert wrap.before == []
assert wrap.after == ["bar"]
wrap = wrappers.wrap_me()
assert wrap.before == []
assert wrap.after == []
def test_wrap_me_call(self):
meth = lambda x: x
wrap = wrappers.wrap_me(before=['foo'], after=['bar'])
assert not hasattr(meth, '_before_calls')
assert not hasattr(meth, '_after_calls')
wrap(meth)
assert meth._before_calls == ['foo']
assert meth._after_calls == ['bar']
wrap(meth)
assert meth._before_calls == ['foo', 'foo']
assert meth._after_calls == ['bar', 'bar']
def test_wrap_me_call(self):
meth = lambda x: x
wrap = wrappers.wrap_me(before=['foo'], after=['bar'])
assert not hasattr(meth, '_before_calls')
assert not hasattr(meth, '_after_calls')
wrap(meth)
assert meth._before_calls == ['foo']
assert meth._after_calls == ['bar']
wrap(meth)
assert meth._before_calls == ['foo', 'foo']
assert meth._after_calls == ['bar', 'bar']
def test_wrap_me_call(self):
meth = lambda x: x
wrap = wrappers.wrap_me(before=["foo"], after=["bar"])
assert not hasattr(meth, "_before_calls")
assert not hasattr(meth, "_after_calls")
wrap(meth)
assert meth._before_calls == ["foo"]
assert meth._after_calls == ["bar"]
wrap(meth)
assert meth._before_calls == ["foo", "foo"]
assert meth._after_calls == ["bar", "bar"]
