def calls_agent_handler_close_if_enabled(self, Handler, client):
c = Connection("host", forward_agent=True)
c.create_session()
c.close()
# NOTE: this will need to change if, for w/e reason, we ever want
# to run multiple handlers at once
Handler.return_value.close.assert_called_once_with()
def activates_paramiko_agent_forwarding_if_configured(
self, Handler, client):
c = Connection("host", forward_agent=True)
chan = c.create_session()
Handler.assert_called_once_with(chan)
def calls_open_for_you(self, client):
c = Connection("host")
c.open = Mock()
c.transport = Mock() # so create_session no asplode
c.create_session()
assert c.open.called
