def test_context_manager(self):
api = Symantec("user", "p@ssw0rd")
api.close = pretend.call_recorder(lambda: None)
with api as ctx_api:
assert ctx_api is api
assert api.close.calls == [pretend.call()]
def test_session_close(self):
api = Symantec("user", "p@ssw0rd")
api.session = pretend.stub(close=pretend.call_recorder(lambda: None))
api.close()
assert api.session.close.calls == [pretend.call()]
