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() 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_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()
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 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>")
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")