Python console.Console类代码示例

 def test_telnet_host_none(self):
     self.dev = Console(
         host=None,
         user='lab',
         password='lab123',
         mode='Telnet')
     self.assertTrue(self.dev.open()['failed'])

 def setUp(self, mock_write, mock_expect, mock_open):
     tty_netconf.open = MagicMock()
     mock_expect.side_effect=[(1, re.search('(?P<login>ogin:\s*$)', "login: "), six.b('\r\r\n ogin:')),
                             (2, re.search('(?P<passwd>assword:\s*$)', "password: "), six.b('\r\r\n password:')),
                             (3, re.search('(?P<shell>%|#\s*$)', "junos % "), six.b('\r\r\[email protected]:~ # '))]
     self.dev = Console(host='1.1.1.1', user='lab', password='lab123', mode='Telnet')
     self.dev.open()

 def test_console_serial(self, mock_write, mock_expect, mock_open):
     tty_netconf.open = MagicMock()
     mock_expect.side_effect = [six.b('\r\r\n Login:'),
                                six.b('\r\r\n password:'),
                                six.b('\r\r\[email protected]:~ # ')]
     self.dev = Console(host='1.1.1.1', user='lab', password='lab123', mode='serial')
     self.dev.open()
     self.assertTrue(self.dev.connected)
     self.assertFalse(self.dev.gather_facts)

 def test_telnet_old_fact_warning(self, mock_warn):
     self.dev = Console(
         host='1.1.1.1',
         user='lab',
         password='lab123',
         mode='Telnet',
         fact_style='old')
     mock_warn.assert_called_once('fact-style old will be removed in a '
                                  'future release.',
                                  RuntimeWarning)

class TestSerial(unittest.TestCase):

    @patch('jnpr.junos.transport.tty_serial.serial.Serial.open')
    @patch('jnpr.junos.transport.tty_serial.serial.Serial.write')
    @patch('jnpr.junos.transport.tty_serial.serial.Serial.flush')
    @patch('jnpr.junos.transport.tty_serial.Serial.read_prompt')
    @patch('jnpr.junos.transport.tty.tty_netconf.open')
    def setUp(self, mock_nc_open, mock_read,
              mock_flush, mock_write, mock_open):
        self.dev = Console(port='USB/ttyUSB0', baud=9600, mode='Serial')
        mock_read.side_effect = [
            ('login', 'login'), ('passwd', 'passwd'), ('shell', 'shell')]
        self.dev.open()

    @patch('jnpr.junos.transport.tty.tty_netconf.close')
    @patch('jnpr.junos.transport.tty_serial.Serial.read_prompt')
    @patch('jnpr.junos.transport.tty_serial.Serial.write')
    @patch('jnpr.junos.transport.tty_serial.Serial._tty_close')
    def tearDown(self, mock_serial_close, mock_write, mock_read, mock_close):
        mock_read.side_effect = [
            ('cli', 'cli'), ('shell', 'shell'), ('login', 'login')]
        self.dev.close()

    def test_console_connected(self):
        self.assertTrue(self.dev.connected)

    def test_close_connection(self):
        self.dev._tty._ser = MagicMock()
        self.dev.close(skip_logout=True)
        self.assertTrue(self.dev._tty._ser.close.called)

 def test_login_bad_password(self, mock_write, mock_expect, mock_open):
     tty_netconf.open = MagicMock()
     mock_expect.side_effect = [(1, re.search('(?P<login>ogin:\s*$)', "login: "), six.b('\r\r\n ogin:')),
                                (2,
                                 re.search('(?P<passwd>assword:\s*$)',
                                           "password: "),
                                    six.b('\r\r\n password:')),
                                (3, re.search('(?P<badpasswd>ogin incorrect)', "login incorrect"),
                                 six.b('\r\r\nlogin incorrect'))]
     self.dev = Console(
         host='1.1.1.1',
         user='lab',
         password='lab123',
         mode='Telnet')
     self.assertRaises(StopIteration, self.dev.open)

class TestSerial(unittest.TestCase):

    @patch('jnpr.junos.transport.tty_serial.serial.Serial.open')
    @patch('jnpr.junos.transport.tty_serial.serial.Serial.write')
    @patch('jnpr.junos.transport.tty_serial.serial.Serial.flush')
    @patch('jnpr.junos.transport.tty_serial.Serial.read_prompt')
    @patch('jnpr.junos.transport.tty.tty_netconf.open')
    def setUp(self, mock_nc_open, mock_read,
              mock_flush, mock_write, mock_open):
        self.dev = Console(port='USB/ttyUSB0', baud=9600, mode='Serial')
        mock_read.side_effect = [
            ('login', 'login'), ('passwd', 'passwd'), ('shell', 'shell')]
        self.dev.open()

    @patch('jnpr.junos.transport.tty.sleep')
    @patch('jnpr.junos.transport.tty.tty_netconf.close')
    @patch('jnpr.junos.transport.tty_serial.Serial.read_prompt')
    @patch('jnpr.junos.transport.tty_serial.Serial.write')
    @patch('jnpr.junos.transport.tty_serial.Serial._tty_close')
    def tearDown(self, mock_serial_close, mock_write, mock_read, mock_close,
                 mock_sleep):
        # mock_read.side_effect = [('shell', 'shell'), ('login', 'login'),
        mock_read.side_effect = [('shell', 'shell'), ('login', 'login'),
                                 ('cli', 'cli'), ]
        self.dev.close()

    def test_console_connected(self):
        self.assertTrue(self.dev.connected)

    def test_close_connection(self):
        self.dev._tty._ser = MagicMock()
        self.dev.close(skip_logout=True)
        self.assertTrue(self.dev._tty._ser.close.called)

    @patch('jnpr.junos.transport.tty_serial.serial.Serial.open')
    def test_tty_serial_open_exception(self, mock_open):
        dev = Console(port='USB/ttyUSB0', baud=9600, mode='Serial')
        mock_open.side_effect = OSError
        self.assertRaises(RuntimeError, dev.open)

    def test_tty_serial_rawwrite(self):
        self.dev._tty._ser = MagicMock()
        self.dev._tty.rawwrite('test')
        self.dev._tty._ser.write.assert_called_with('test')

    def test_tty_serial_read(self):
        self.dev._tty._ser = MagicMock()
        self.dev._tty.read()
        self.dev._tty._ser.readline.assert_called()

    def test_tty_serial_read_prompt(self):
        self.dev._tty._ser = MagicMock()
        self.dev._tty.EXPECT_TIMEOUT = 0.1
        self.dev._tty._ser.readline.side_effect = ['', 'test']
        self.assertEqual(self.dev._tty.read_prompt()[0], None)

class TestSerial(unittest.TestCase):
    @patch("jnpr.junos.transport.tty_serial.serial.Serial.open")
    @patch("jnpr.junos.transport.tty_serial.serial.Serial.write")
    @patch("jnpr.junos.transport.tty_serial.serial.Serial.flush")
    @patch("jnpr.junos.transport.tty_serial.Serial.read_prompt")
    @patch("jnpr.junos.transport.tty.tty_netconf.open")
    def setUp(self, mock_nc_open, mock_read, mock_flush, mock_write, mock_open):
        self.dev = Console(port="USB/ttyUSB0", baud=9600, mode="Serial")
        mock_read.side_effect = [("login", "login"), ("passwd", "passwd"), ("shell", "shell")]
        self.dev.open()

    @patch("jnpr.junos.transport.tty.sleep")
    @patch("jnpr.junos.transport.tty.tty_netconf.close")
    @patch("jnpr.junos.transport.tty_serial.Serial.read_prompt")
    @patch("jnpr.junos.transport.tty_serial.Serial.write")
    @patch("jnpr.junos.transport.tty_serial.Serial._tty_close")
    def tearDown(self, mock_serial_close, mock_write, mock_read, mock_close, mock_sleep):
        # mock_read.side_effect = [('shell', 'shell'), ('login', 'login'),
        mock_read.side_effect = [("shell", "shell"), ("login", "login"), ("cli", "cli")]
        self.dev.close()

    def test_console_connected(self):
        self.assertTrue(self.dev.connected)

    def test_close_connection(self):
        self.dev._tty._ser = MagicMock()
        self.dev.close(skip_logout=True)
        self.assertTrue(self.dev._tty._ser.close.called)

    @patch("jnpr.junos.transport.tty_serial.serial.Serial.open")
    def test_tty_serial_open_exception(self, mock_open):
        dev = Console(port="USB/ttyUSB0", baud=9600, mode="Serial")
        mock_open.side_effect = OSError
        self.assertRaises(RuntimeError, dev.open)

    def test_tty_serial_rawwrite(self):
        self.dev._tty._ser = MagicMock()
        self.dev._tty.rawwrite("test")
        self.dev._tty._ser.write.assert_called_with("test")

    def test_tty_serial_read(self):
        self.dev._tty._ser = MagicMock()
        self.dev._tty.read()
        self.dev._tty._ser.readline.assert_is_called()

    def test_tty_serial_read_prompt(self):
        self.dev._tty._ser = MagicMock()
        self.dev._tty.EXPECT_TIMEOUT = 0.1
        self.dev._tty._ser.readline.side_effect = ["", "test"]
        self.assertEqual(self.dev._tty.read_prompt()[0], None)

class TestConsole(unittest.TestCase):

    @patch('jnpr.junos.transport.tty_telnet.Telnet._tty_open')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    def setUp(self, mock_write, mock_expect, mock_open):
        tty_netconf.open = MagicMock()
        mock_expect.side_effect = [(1, re.search('(?P<login>ogin:\s*$)', "login: "), six.b('\r\r\n ogin:')),
                                   (2,
                                    re.search('(?P<passwd>assword:\s*$)',
                                              "password: "),
                                       six.b('\r\r\n password:')),
                                   (3, re.search('(?P<shell>%|#\s*$)', "junos % "), six.b('\r\r\[email protected]:~ # '))]
        self.dev = Console(
            host='1.1.1.1',
            user='lab',
            password='lab123',
            mode='Telnet')
        self.dev.open()

    # @patch('jnpr.junos.transport.tty.tty_netconf.close')
    # @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    # @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    # def tearDown(self, mock_write, mock_expect, mock_nc_close):
    #     mock_expect.side_effect = [(1, re.search('(?P<cli>[^\\-"]>\s*$)', "cli>"), six.b('\r\r\[email protected]>')),
    #                                (2,
    # re.search('(?P<shell>%|#\s*$)',
    #                                           "junos %"),
    # six.b('\r\r\[email protected]:~ # ')),
    #                                (3, re.search('(?P<login>ogin:\s*$)', "login: "), six.b('\r\r\nlogin'))]
    #     self.dev.close()

    @patch('jnpr.junos.console.Console._tty_logout')
    def tearDown(self, mock_tty_logout):
        self.dev.close()

    def test_telnet_host_none(self):
        self.dev = Console(
            host=None,
            user='lab',
            password='lab123',
            mode='Telnet')
        self.assertTrue(self.dev.open()['failed'])

    @patch('jnpr.junos.console.warnings')
    def test_telnet_old_fact_warning(self, mock_warn):
        self.dev = Console(
            host='1.1.1.1',
            user='lab',
            password='lab123',
            mode='Telnet',
            fact_style='old')
        mock_warn.assert_has_calls([call.warn(
            'fact-style old will be removed in a '
                                     'future release.',
                                    RuntimeWarning)])

    @patch('jnpr.junos.transport.tty_telnet.Telnet._tty_open')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    def test_login_bad_password(self, mock_write, mock_expect, mock_open):
        tty_netconf.open = MagicMock()
        mock_expect.side_effect = [(1, re.search('(?P<login>ogin:\s*$)', "login: "), six.b('\r\r\n ogin:')),
                                   (2,
                                    re.search('(?P<passwd>assword:\s*$)',
                                              "password: "),
                                       six.b('\r\r\n password:')),
                                   (3, re.search('(?P<badpasswd>ogin incorrect)', "login incorrect"),
                                    six.b('\r\r\nlogin incorrect'))]
        self.dev = Console(
            host='1.1.1.1',
            user='lab',
            password='lab123',
            mode='Telnet')
        self.assertRaises(StopIteration, self.dev.open)

    @patch('jnpr.junos.console.Console._tty_logout')
    @patch('jnpr.junos.transport.tty_telnet.Telnet._tty_open')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    def test_with_context(
            self, mock_write, mock_expect, mock_open, mock_logout):
        tty_netconf.open = MagicMock()

        mock_expect.side_effect = [(1, re.search('(?P<login>ogin:\s*$)', "login: "),
                                    six.b('\r\r\n ogin:')),
                                   (2,
                                    re.search('(?P<passwd>assword:\s*$)',
                                              "password: "),
                                       six.b('\r\r\n password:')),
                                   (3, re.search('(?P<shell>%|#\s*$)', "junos % "),
                                    six.b('\r\r\[email protected]:~ # '))]
        with Console(host='1.1.1.1', user='lab', password='lab123', mode='Telnet') as dev:
            self.assertTrue(isinstance(self.dev, Console))

    @patch('jnpr.junos.console.Console._tty_login')
    def test_console_open_error(self, mock_tty_login):
        mock_tty_login.side_effect = RuntimeError
        self.assertRaises(RuntimeError, self.dev.open)

#.........这里部分代码省略.........

class TestConsole(unittest.TestCase):

    @patch('jnpr.junos.transport.tty_telnet.Telnet._tty_open')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    def setUp(self, mock_write, mock_expect, mock_open):
        tty_netconf.open = MagicMock()
        mock_expect.side_effect=[(1, re.search('(?P<login>ogin:\s*$)', "login: "), six.b('\r\r\n ogin:')),
                                (2, re.search('(?P<passwd>assword:\s*$)', "password: "), six.b('\r\r\n password:')),
                                (3, re.search('(?P<shell>%|#\s*$)', "junos % "), six.b('\r\r\[email protected]:~ # '))]
        self.dev = Console(host='1.1.1.1', user='lab', password='lab123', mode='Telnet')
        self.dev.open()

    @patch('jnpr.junos.transport.tty.tty_netconf.close')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    def tearDown(self, mock_write, mock_expect, mock_nc_close):
        mock_expect.side_effect = [(1, re.search('(?P<cli>[^\\-"]>\s*$)', "cli>"), six.b('\r\r\[email protected]>')),
                                   (2, re.search('(?P<shell>%|#\s*$)', "junos %"), six.b('\r\r\[email protected]:~ # ')),
                                   (3, re.search('(?P<login>ogin:\s*$)', "login: "), six.b('\r\r\nlogin'))]
        self.dev.close()

    @patch('jnpr.junos.console.Console._tty_logout')
    def tearDown(self, mock_tty_logout):
        self.dev.close()

    @patch('jnpr.junos.console.Console._tty_login')
    def test_console_open_error(self, mock_tty_login):
        mock_tty_login.side_effect = RuntimeError
        self.assertRaises(RuntimeError, self.dev.open)

    def test_console_connected(self):
        self.assertTrue(self.dev.connected)
        self.assertFalse(self.dev.gather_facts)

    @patch('jnpr.junos.console.Console._tty_logout')
    def test_console_close_error(self, mock_logout):
        mock_logout.side_effect = RuntimeError
        self.assertRaises(RuntimeError, self.dev.close)

    @patch('jnpr.junos.transport.tty_telnet.Telnet')
    @patch('jnpr.junos.console.Console._tty_login')
    def test_console_tty_open_err(self, mock_login, mock_telnet):
        with patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.open') as mock_open:
            mock_telnet.RETRY_OPEN = 1
            mock_login.side_effect = ValueError
            self.dev._tty.LOGIN_RETRY = self.dev._tty.RETRY_OPEN = 1
            self.assertRaises(ValueError, self.dev.open)

    @patch('jnpr.junos.transport.tty_serial.Serial._tty_open')
    @patch('jnpr.junos.transport.tty_serial.serial.Serial.readline')
    @patch('jnpr.junos.transport.tty_serial.Serial.write')
    def test_console_serial(self, mock_write, mock_expect, mock_open):
        tty_netconf.open = MagicMock()
        mock_expect.side_effect = [six.b('\r\r\n Login:'),
                                   six.b('\r\r\n password:'),
                                   six.b('\r\r\[email protected]:~ # ')]
        self.dev = Console(host='1.1.1.1', user='lab', password='lab123', mode='serial')
        self.dev.open()
        self.assertTrue(self.dev.connected)
        self.assertFalse(self.dev.gather_facts)

    # @patch('jnpr.junos.transport.tty_telnet.Telnet._tty_close')
    # def test_console_close_error2(self, mock_close):
    #     mock_close.side_effect = RuntimeError
    #     self.assertRaises(RuntimeError, self.dev.close(True))

    @patch('jnpr.junos.transport.tty_netconf.tty_netconf.rpc')
    def test_console_zeroize(self, mock_zeroize):
        self.dev.zeroize()
        self.assertTrue(mock_zeroize.called)

    @patch('jnpr.junos.transport.tty_netconf.tty_netconf.rpc')
    @patch('jnpr.junos.console.FACT_LIST')
    def test_console_gather_facts(self, mock_fact_list, mock_rpc):
        from jnpr.junos.facts.session import facts_session
        mock_fact_list.__iter__.return_value = [facts_session]
        self.dev.facts_refresh()
        self.assertEqual(mock_rpc.call_count, 3)

    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.write')
    @patch('jnpr.junos.transport.tty_netconf.select.select')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.read_until')
    def test_load_console(self, mock_read_until, mock_select, mock_write):
        mock_select.return_value = ([self.dev._tty._rx], [], [])
        xml = """<policy-options>
                  <policy-statement>
                    <name>F5-in</name>
                    <term>
                        <name>testing</name>
                        <then>
                            <accept/>
                        </then>
                    </term>
                    <from>
                        <protocol>mpls</protocol>
                    </from>
                </policy-statement>
                </policy-options>"""

#.........这里部分代码省略.........

 def setUp(self, mock_nc_open, mock_read,
           mock_flush, mock_write, mock_open):
     self.dev = Console(port='USB/ttyUSB0', baud=9600, mode='Serial')
     mock_read.side_effect = [
         ('login', 'login'), ('passwd', 'passwd'), ('shell', 'shell')]
     self.dev.open()

class TestConsole(unittest.TestCase):

    @patch('jnpr.junos.transport.tty_telnet.Telnet._tty_open')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    def setUp(self, mock_write, mock_expect, mock_open):
        tty_netconf.open = MagicMock()
        mock_expect.side_effect=[(1, re.search('(?P<login>ogin:\s*$)', "login: "), '\r\r\n ogin:'),
                                (2, re.search('(?P<passwd>assword:\s*$)', "password: "), '\r\r\n password:'),
                                (3, re.search('(?P<shell>%|#\s*$)', "junos % "), '\r\r\[email protected]:~ # ')]
        self.dev = Console(host='1.1.1.1', user='lab', password='lab123', mode = 'Telnet')
        self.dev.open()

    @patch('jnpr.junos.transport.tty.tty_netconf.close')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.expect')
    @patch('jnpr.junos.transport.tty_telnet.Telnet.write')
    def tearDown(self, mock_write, mock_expect, mock_nc_close):
        mock_expect.side_effect = [(1, re.search('(?P<cli>[^\\-"]>\s*$)', "cli>"), '\r\r\[email protected]>'),
                                   (2, re.search('(?P<shell>%|#\s*$)', "junos %"), '\r\r\[email protected]:~ # '),
                                   (3, re.search('(?P<login>ogin:\s*$)', "login: "), '\r\r\nlogin')]
        self.dev.close()

    @patch('jnpr.junos.console.Console._tty_logout')
    def tearDown(self, mock_tty_logout):
        self.dev.close()

    @patch('jnpr.junos.console.Console._tty_login')
    def test_console_open_error(self, mock_tty_login):
        mock_tty_login.side_effect = RuntimeError
        self.assertRaises(RuntimeError, self.dev.open)

    def test_console_connected(self):
        self.assertTrue(self.dev.connected)
        self.assertFalse(self.dev.gather_facts)

    @patch('jnpr.junos.console.Console._tty_logout')
    def test_console_close_error(self, mock_logout):
        mock_logout.side_effect = RuntimeError
        self.assertRaises(RuntimeError, self.dev.close)

    # @patch('jnpr.junos.transport.tty_telnet.Telnet._tty_close')
    # def test_console_close_error2(self, mock_close):
    #     mock_close.side_effect = RuntimeError
    #     self.assertRaises(RuntimeError, self.dev.close(True))

    @patch('jnpr.junos.transport.tty_netconf.tty_netconf.rpc')
    def test_console_zeroize(self, mock_zeroize):
        self.dev.zeroize()
        self.assertTrue(mock_zeroize.called)

    @patch('jnpr.junos.transport.tty_netconf.tty_netconf.rpc')
    @patch('jnpr.junos.console.FACT_LIST')
    def test_console_gather_facts(self, mock_fact_list, mock_rpc):
        from jnpr.junos.facts.session import facts_session
        mock_fact_list.__iter__.return_value = [facts_session]
        self.dev._gather_facts()
        self.assertEqual(mock_rpc.call_count, 1)

    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.write')
    @patch('jnpr.junos.transport.tty_netconf.select.select')
    @patch('jnpr.junos.transport.tty_telnet.telnetlib.Telnet.read_until')
    def test_load_console(self, mock_read_until, mock_select, mock_write):
        mock_select.return_value = ([self.dev._tty._rx], [], [])
        xml = """<policy-options>
                  <policy-statement>
                    <name>F5-in</name>
                    <term>
                        <name>testing</name>
                        <then>
                            <accept/>
                        </then>
                    </term>
                    <from>
                        <protocol>mpls</protocol>
                    </from>
                </policy-statement>
                </policy-options>"""

        mock_read_until.return_value = """
        <rpc-reply xmlns="urn:ietf:params:xml:ns:netconf:base:1.0" xmlns:junos="http://xml.juniper.net/junos/15.2I0/junos">
            <load-configuration-results>
            <ok/>
            </load-configuration-results>
            </rpc-reply>
            ]]>]]>"""
        cu = Config(self.dev)
        cu.load(xml, format='xml')
        cu.commit()

 def setUp(self, mock_nc_open, mock_read, mock_flush, mock_write, mock_open):
     self.dev = Console(port="USB/ttyUSB0", baud=9600, mode="Serial")
     mock_read.side_effect = [("login", "login"), ("passwd", "passwd"), ("shell", "shell")]
     self.dev.open()