def test_tty_telnet_baud(self, mock_telnet):
tel_conn = Telnet(host='1.1.1.1', user='******',
password='******', port=23,
timeout=30, baud=0)
tel_conn._tty_open()
tel_conn.rawwrite('<rpc>')
tel_conn._tn.write.assert_called_with('<rpc>')
class TestTTYTelnet(unittest.TestCase):
@patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet')
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host='1.1.1.1',
user='******',
password='******',
port=23,
timeout=30)
def test_open(self):
self.tel_conn._tty_open()
self.tel_conn._tn.open.assert_called()
def test_open_exception(self):
self.tel_conn._tn.open.side_effect = Exception
Telnet.RETRY_OPEN = 1
Telnet.RETRY_BACKOFF = 0.1
self.assertRaises(RuntimeError, self.tel_conn._tty_open)
# reset
Telnet.RETRY_OPEN = 3
Telnet.RETRY_BACKOFF = 2
def test_close(self):
self.tel_conn._tty_close()
self.tel_conn._tn.close.assert_called()
def test_read(self):
self.tel_conn.read()
self.tel_conn._tn.read_until.assert_called()
@patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet')
def test_tty_telnet_baud(self, mock_telnet):
tel_conn = Telnet(host='1.1.1.1',
user='******',
password='******',
port=23,
timeout=30,
baud=0)
tel_conn._tty_open()
tel_conn.rawwrite('<rpc>')
tel_conn._tn.write.assert_called_with('<rpc>')
def test_read_prompt_RuntimeError(self):
self.tel_conn.expect = MagicMock()
self.tel_conn.expect = (None, None, 'port already in use')
self.assertRaises(RuntimeError, self.tel_conn._login_state_machine)
def test_read_prompt_in_use_RuntimeError(self):
self.tel_conn.expect = MagicMock()
self.tel_conn._tn.expect.return_value = (None, None,
six.b('port already in use'))
self.assertRaises(RuntimeError, self.tel_conn._login_state_machine)
def test_tty_telnet_rawwrite_sys_py3(self):
with patch.object(sys.modules['sys'], 'version', '3.x') \
as mock_sys:
self.tel_conn._tty_open()
content = MagicMock()
self.tel_conn.rawwrite(content)
content.decode.assert_called_with('utf-8')
class TestTTYTelnet(unittest.TestCase):
@patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet')
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host='1.1.1.1', user='******',
password='******', port=23,
timeout=30)
def test_open(self):
self.tel_conn._tty_open()
self.tel_conn._tn.open.assert_called()
def test_open_exception(self):
self.tel_conn._tn.open.side_effect = Exception
Telnet.RETRY_OPEN = 1
Telnet.RETRY_BACKOFF = 0.1
self.assertRaises(RuntimeError, self.tel_conn._tty_open)
# reset
Telnet.RETRY_OPEN = 3
Telnet.RETRY_BACKOFF = 2
def test_close(self):
self.tel_conn._tty_close()
self.tel_conn._tn.close.assert_called()
def test_read(self):
self.tel_conn.read()
self.tel_conn._tn.read_until.assert_called()
@patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet')
def test_tty_telnet_baud(self, mock_telnet):
tel_conn = Telnet(host='1.1.1.1', user='******',
password='******', port=23,
timeout=30, baud=0)
tel_conn._tty_open()
tel_conn.rawwrite('<rpc>')
tel_conn._tn.write.assert_called_with('<rpc>')
def test_read_prompt_RuntimeError(self):
self.tel_conn.expect = MagicMock()
self.tel_conn.expect = (None, None, 'port already in use')
self.assertRaises(RuntimeError, self.tel_conn._login_state_machine)
def test_read_prompt_in_use_RuntimeError(self):
self.tel_conn.expect = MagicMock()
self.tel_conn._tn.expect.return_value = (
None,
None,
six.b('port already in use'))
self.assertRaises(RuntimeError, self.tel_conn._login_state_machine)
def test_tty_telnet_rawwrite_sys_py3(self):
with patch.object(sys.modules['sys'], 'version', '3.x') \
as mock_sys:
self.tel_conn._tty_open()
content = MagicMock()
self.tel_conn.rawwrite(content)
content.decode.assert_called_with('utf-8')
def _tty_login(self):
tty_args = dict()
tty_args['user'] = self._auth_user
tty_args['passwd'] = self._auth_password
tty_args['timeout'] = float(self._timeout)
tty_args['attempts'] = int(self._attempts)
tty_args['baud'] = self._baud
tty_args['huge_tree'] = self._huge_tree
if self._mode and self._mode.upper() == 'TELNET':
tty_args['host'] = self._hostname
tty_args['port'] = self._port
tty_args['console_has_banner'] = self.console_has_banner
self.console = ('telnet', self._hostname, self.port)
self._tty = Telnet(**tty_args)
elif self.cs_user is not None:
tty_args['cs_user'] = self.cs_user
tty_args['cs_passwd'] = self.cs_passwd
tty_args['host'] = self._hostname
tty_args['port'] = self._port
tty_args['console_has_banner'] = self.console_has_banner
tty_args['ssh_private_key_file'] = self._ssh_private_key_file
self.console = ('ssh', self._hostname, self.port)
self._tty = SSH(**tty_args)
elif self._mode.upper() == 'SERIAL':
tty_args['port'] = self._port
self.console = ('serial', self._port)
self._tty = Serial(**tty_args)
else:
logger.error('Mode should be either telnet or serial')
raise AttributeError('Mode to be telnet/serial')
self._tty.login()
def _tty_login(self):
tty_args = dict()
tty_args["user"] = self._auth_user
tty_args["passwd"] = self._auth_password
tty_args["timeout"] = float(self._timeout)
tty_args["attempts"] = int(self._attempts)
tty_args["baud"] = self._baud
tty_args["huge_tree"] = self._huge_tree
if self._mode and self._mode.upper() == "TELNET":
tty_args["host"] = self._hostname
tty_args["port"] = self._port
tty_args["console_has_banner"] = self.console_has_banner
self.console = ("telnet", self._hostname, self.port)
self._tty = Telnet(**tty_args)
elif self.cs_user is not None:
tty_args["cs_user"] = self.cs_user
tty_args["cs_passwd"] = self.cs_passwd
tty_args["host"] = self._hostname
tty_args["port"] = self._port
tty_args["console_has_banner"] = self.console_has_banner
tty_args["ssh_private_key_file"] = self._ssh_private_key_file
self.console = ("ssh", self._hostname, self.port)
self._tty = SSH(**tty_args)
elif self._mode.upper() == "SERIAL":
tty_args["port"] = self._port
self.console = ("serial", self._port)
self._tty = Serial(**tty_args)
else:
logger.error("Mode should be either telnet or serial")
raise AttributeError("Mode to be telnet/serial")
self._tty.login()
class TestTTYTelnet(unittest.TestCase):
@patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet')
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host='1.1.1.1',
user='******',
password='******',
port=23,
timeout=30)
def test_open(self):
self.tel_conn._tty_open()
self.tel_conn._tn.open.assert_called()
def test_open_exception(self):
self.tel_conn._tn.open.side_effect = Exception
Telnet.RETRY_OPEN = 1
Telnet.RETRY_BACKOFF = 0.1
self.assertRaises(RuntimeError, self.tel_conn._tty_open)
# reset
Telnet.RETRY_OPEN = 3
Telnet.RETRY_BACKOFF = 2
def test_close(self):
self.tel_conn._tty_close()
self.tel_conn._tn.close.assert_called()
def test_read(self):
self.tel_conn.read()
self.tel_conn._tn.read_until.assert_called()
def test_read_prompt_RuntimeError(self):
self.tel_conn.expect = MagicMock()
self.tel_conn.expect = (None, None, 'port already in use')
self.assertRaises(RuntimeError, self.tel_conn._login_state_machine)
class TestTTYTelnet(unittest.TestCase):
@patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet')
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host='1.1.1.1', user='******',
password='******', port=23,
timeout=30)
def test_open(self):
self.tel_conn._tty_open()
self.tel_conn._tn.open.assert_called()
def test_open_exception(self):
self.tel_conn._tn.open.side_effect = Exception
Telnet.RETRY_OPEN = 1
Telnet.RETRY_BACKOFF = 0.1
self.assertRaises(RuntimeError, self.tel_conn._tty_open)
# reset
Telnet.RETRY_OPEN = 3
Telnet.RETRY_BACKOFF = 2
def test_close(self):
self.tel_conn._tty_close()
self.tel_conn._tn.close.assert_called()
def test_read(self):
self.tel_conn.read()
self.tel_conn._tn.read_until.assert_called()
def test_tty_telnet_baud(self, mock_telnet):
tel_conn = Telnet(host='1.1.1.1',
user='******',
password='******',
port=23,
timeout=30,
baud=0)
tel_conn._tty_open()
tel_conn.rawwrite('<rpc>')
tel_conn._tn.write.assert_called_with('<rpc>')
def test_tty_telnet_baud(self, mock_telnet):
tel_conn = Telnet(
host="1.1.1.1",
user="******",
password="******",
port=23,
timeout=30,
baud=0,
)
tel_conn._tty_open()
tel_conn.rawwrite("<rpc>")
tel_conn._tn.write.assert_called_with("<rpc>")
def _tty_login(self):
tty_args = dict()
tty_args['user'] = self._auth_user
tty_args['passwd'] = self._auth_password
tty_args['timeout'] = float(self._timeout)
tty_args['attempts'] = int(self._attempts)
tty_args['baud'] = self._baud
if self._mode.upper() == 'TELNET':
tty_args['host'] = self._hostname
tty_args['port'] = self._port
self.console = ('telnet', self._hostname, self.port)
self._tty = Telnet(**tty_args)
elif self._mode.upper() == 'SERIAL':
tty_args['port'] = self._port
self.console = ('serial', self._port)
self._tty = Serial(**tty_args)
else:
logger.error('Mode should be either telnet or serial')
raise AttributeError('Mode to be telnet/serial')
self._tty.login()
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host='1.1.1.1',
user='******',
password='******',
port=23,
timeout=30)
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host='1.1.1.1', user='******',
password='******', port=23,
timeout=30)
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host="1.1.1.1",
user="******",
password="******",
port=23,
timeout=30)
class TestTTYTelnet(unittest.TestCase):
@patch("jnpr.junos.transport.tty_telnet.telnetlib.Telnet")
def setUp(self, mpock_telnet):
self.tel_conn = Telnet(host="1.1.1.1",
user="******",
password="******",
port=23,
timeout=30)
def test_open(self):
self.tel_conn._tty_open()
self.tel_conn._tn.open.assert_called()
def test_open_exception(self):
self.tel_conn._tn.open.side_effect = Exception
Telnet.RETRY_OPEN = 1
Telnet.RETRY_BACKOFF = 0.1
self.assertRaises(RuntimeError, self.tel_conn._tty_open)
# reset
Telnet.RETRY_OPEN = 3
Telnet.RETRY_BACKOFF = 2
def test_close(self):
self.tel_conn._tty_close()
self.tel_conn._tn.close.assert_called()
def test_read(self):
self.tel_conn.read()
self.tel_conn._tn.read_until.assert_called()
@patch("jnpr.junos.transport.tty_telnet.telnetlib.Telnet")
def test_tty_telnet_baud(self, mock_telnet):
tel_conn = Telnet(
host="1.1.1.1",
user="******",
password="******",
port=23,
timeout=30,
baud=0,
)
tel_conn._tty_open()
tel_conn.rawwrite("<rpc>")
tel_conn._tn.write.assert_called_with("<rpc>")
def test_read_prompt_RuntimeError(self):
self.tel_conn.expect = MagicMock()
self.tel_conn.expect = (None, None, "port already in use")
self.assertRaises(RuntimeError, self.tel_conn._login_state_machine)
def test_read_prompt_in_use_RuntimeError(self):
self.tel_conn.expect = MagicMock()
self.tel_conn._tn.expect.return_value = (
None,
None,
six.b("port already in use"),
)
self.assertRaises(RuntimeError, self.tel_conn._login_state_machine)
def test_tty_telnet_rawwrite_sys_py3(self):
with patch.object(sys.modules["sys"], "version", "3.x") as mock_sys:
self.tel_conn._tty_open()
content = MagicMock()
self.tel_conn.rawwrite(content)
content.decode.assert_called_with("utf-8")
