mirror of
https://github.com/hansjone/netx.git
synced 2026-10-14 16:00:51 +08:00
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
"""Tests for vendor paging-disable helpers used by LLDP / collect / hop."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
from netx_api.ne_netmiko import disable_target_paging, paging_disable_commands
|
|
|
|
|
|
class PagingDisableTests(unittest.TestCase):
|
|
def test_routeros_does_not_send_cisco_paging_command(self) -> None:
|
|
conn = MagicMock()
|
|
self.assertEqual(disable_target_paging(conn, device_type="mikrotik_routeros"), "")
|
|
conn.send_command_timing.assert_not_called()
|
|
conn.write_channel.assert_not_called()
|
|
|
|
def test_commands_by_vendor(self) -> None:
|
|
self.assertEqual(
|
|
paging_disable_commands(vendor="Huawei", device_type="huawei"),
|
|
["screen-length 0 temporary"],
|
|
)
|
|
self.assertEqual(
|
|
paging_disable_commands(vendor="ZTE", device_type="zte_zxros"),
|
|
["terminal length 0"],
|
|
)
|
|
self.assertEqual(
|
|
paging_disable_commands(vendor="Cisco", device_type="cisco_ios"),
|
|
["terminal length 0"],
|
|
)
|
|
self.assertEqual(
|
|
paging_disable_commands(vendor="H3C", device_type="hp_comware"),
|
|
["screen-length disable"],
|
|
)
|
|
|
|
def test_disable_target_paging_uses_timing(self) -> None:
|
|
conn = MagicMock()
|
|
conn.send_command_timing.return_value = "ok\n<HUAWEI>"
|
|
out = disable_target_paging(conn, vendor="Huawei", device_type="huawei")
|
|
conn.send_command_timing.assert_called_once_with(
|
|
"screen-length 0 temporary", read_timeout=15
|
|
)
|
|
self.assertIn("ok", out)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|