def connect(ns):
password = ns.password
if password is None:
password = getpass.getpass()
connect_args = dict(host=ns.host,
port=ns.port,
username=ns.username,
password=password,
key_filename=ns.privKeyFile,
hostkey_verify=False,
look_for_keys=False,
allow_agent=False)
device_handler = manager.make_device_handler(None)
if not ns.tcp and not ns.raw:
# use the public API only
session = transport.SSHSession(device_handler)
elif ns.tcp:
session = nctransport.TCPSession(device_handler, ns.raw)
else:
session = nctransport.SSHSession(device_handler, ns.raw)
if not ns.tcp and ("hostkey_verify" not in connect_args
or connect_args["hostkey_verify"]):
session.load_known_hosts()
try:
session.connect(**connect_args)
except Exception:
if session.transport:
session.close()
raise
return manager.Manager(session, device_handler, **connect_args)
def test_manager_locked(self, mock_lock):
conn = manager.Manager(None, None, timeout=20)
conn.locked(None)
mock_lock.assert_called_once_with(None, None, None)
