mirror of
https://github.com/hansjone/netx.git
synced 2026-10-09 03:10:46 +08:00
oliver
0d1d3200f1
Require up/down admin columns and known iface name prefixes so device prompts are not treated as ports. Co-authored-by: Cursor <cursoragent@cursor.com>
136 lines
5.4 KiB
Python
136 lines
5.4 KiB
Python
"""Unit tests for ZTE port traffic brief/detail parsers (sample from oclaw Untitled-1.ps1)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from netx_api.port_traffic_commands import commands_for_vendor, detail_command
|
|
from netx_api.port_traffic_parsers import (
|
|
parse_bw_to_bps,
|
|
parse_zte_interface_brief,
|
|
parse_zte_interface_detail,
|
|
)
|
|
|
|
SAMPLE_DETAIL = """\
|
|
AL5458-ACC-6120HS#show interface xgei-1/1/0/1
|
|
13:47:08 Africa/Algiers Fri Jul 31 2026
|
|
xgei-1/1/0/1 is up, ifindex: 8194
|
|
Description: C2930L100-EQ2
|
|
Line protocol is up, IPv4 protocol is up, IPv6 protocol is down,
|
|
detected status is RX-OK/TX-OK
|
|
Last line protocol up time : 2026-05-26 15:14:49
|
|
Hardware is XGigabit Ethernet, address is 00d0.0000.088f
|
|
Internet address is 192.168.1.1/30
|
|
BW 1 Gbit/s
|
|
IP MTU 1500 bytes
|
|
MTU 1562 bytes
|
|
MPLS MTU 1548 bytes
|
|
|
|
Fec-eth : N/A
|
|
Fec-bypass : N/A
|
|
ARP type ARP
|
|
ARP Timeout 04:00:00
|
|
Last Clear Time : 2026-05-26 15:13:48 Last Refresh Time: 2026-07-31 13:47:00
|
|
Rate period : 30 s
|
|
Input : 824 bit/s 1 packet/s
|
|
Output : 824 bit/s 1 packet/s
|
|
Peak rate:
|
|
Input : 3536 bit/s peak time 2026-05-26 15:17:10
|
|
Output : 6896 bit/s peak time 2026-05-26 15:17:10
|
|
Intf utilization: input 0% output 0%
|
|
HardwareCounters:
|
|
In_Bytes 462427066 In_Packets 6529765
|
|
"""
|
|
|
|
SAMPLE_BRIEF = """\
|
|
AL5458-ACC-6120HS#show interface brief
|
|
13:43:00 Africa/Algiers Fri Jul 31 2026
|
|
Interface Attribute Mode BW Admin Phy Prot Description
|
|
xgei-1/1/0/1 optical Duplex/full 1G up up up C2930L100-EQ2
|
|
xgei-1/1/0/2 optical Duplex/full 1G up down down
|
|
xgei-1/1/0/4 optical Duplex/full 10G up down down 123456test11255
|
|
cgei-1/1/0/33 optical Duplex/full 100G up down down
|
|
xxvgei-1/1/0/11 optical Duplex/full 25G up down down
|
|
smartgroup1 N/A N/A up up down
|
|
smartgroup10 N/A N/A 1G up up up
|
|
bvi2 N/A N/A up up up
|
|
"""
|
|
|
|
|
|
class BwParseTests(unittest.TestCase):
|
|
def test_compact(self):
|
|
self.assertEqual(parse_bw_to_bps("1G"), 1_000_000_000)
|
|
self.assertEqual(parse_bw_to_bps("10G"), 10_000_000_000)
|
|
self.assertEqual(parse_bw_to_bps("100G"), 100_000_000_000)
|
|
self.assertEqual(parse_bw_to_bps("25G"), 25_000_000_000)
|
|
self.assertEqual(parse_bw_to_bps("1M"), 1_000_000)
|
|
|
|
def test_detail_line(self):
|
|
self.assertEqual(parse_bw_to_bps("BW 1 Gbit/s"), 1_000_000_000)
|
|
|
|
|
|
class BriefParserTests(unittest.TestCase):
|
|
def test_sample_brief(self):
|
|
rows = parse_zte_interface_brief(SAMPLE_BRIEF)
|
|
by_name = {r.ifname: r for r in rows}
|
|
self.assertIn("xgei-1/1/0/1", by_name)
|
|
r1 = by_name["xgei-1/1/0/1"]
|
|
self.assertEqual(r1.bw_bps, 1_000_000_000)
|
|
self.assertEqual(r1.admin, "up")
|
|
self.assertEqual(r1.phy, "up")
|
|
self.assertEqual(r1.prot, "up")
|
|
self.assertEqual(r1.description, "C2930L100-EQ2")
|
|
self.assertEqual(by_name["xgei-1/1/0/4"].bw_bps, 10_000_000_000)
|
|
self.assertEqual(by_name["cgei-1/1/0/33"].bw_bps, 100_000_000_000)
|
|
self.assertEqual(by_name["xxvgei-1/1/0/11"].bw_bps, 25_000_000_000)
|
|
self.assertEqual(by_name["smartgroup1"].bw_bps, 0)
|
|
self.assertEqual(by_name["smartgroup10"].bw_bps, 1_000_000_000)
|
|
self.assertEqual(by_name["xgei-1/1/0/2"].phy, "down")
|
|
|
|
def test_rejects_trailing_hostname_prompt(self):
|
|
text = SAMPLE_BRIEF + "\nAL5458-ACC-6120HS#\n"
|
|
rows = parse_zte_interface_brief(text)
|
|
names = [r.ifname for r in rows]
|
|
self.assertNotIn("AL5458-ACC-6120HS#", names)
|
|
self.assertNotIn("AL5458-ACC-6120HS", names)
|
|
self.assertTrue(any(n.startswith("xgei-") for n in names))
|
|
|
|
def test_rejects_prompt_without_hash(self):
|
|
# Some captures leave hostname alone after the table.
|
|
text = (
|
|
SAMPLE_BRIEF
|
|
+ "\nAL5458-ACC-6120HS\n"
|
|
+ "AL5458-ACC-6120HS#show something\n"
|
|
)
|
|
rows = parse_zte_interface_brief(text)
|
|
names = [r.ifname for r in rows]
|
|
self.assertNotIn("AL5458-ACC-6120HS", names)
|
|
self.assertNotIn("AL5458-ACC-6120HS#show", names)
|
|
|
|
|
|
class DetailParserTests(unittest.TestCase):
|
|
def test_sample_detail(self):
|
|
d = parse_zte_interface_detail(SAMPLE_DETAIL)
|
|
self.assertEqual(d.ifname, "xgei-1/1/0/1")
|
|
self.assertEqual(d.bw_bps, 1_000_000_000)
|
|
self.assertEqual(d.rate_period_sec, 30)
|
|
self.assertEqual(d.in_bps, 824.0)
|
|
self.assertEqual(d.out_bps, 824.0)
|
|
self.assertEqual(d.in_util_pct, 0.0)
|
|
self.assertEqual(d.out_util_pct, 0.0)
|
|
self.assertEqual(d.description, "C2930L100-EQ2")
|
|
# Peak rates must not override Rate period values
|
|
self.assertNotEqual(d.in_bps, 3536.0)
|
|
|
|
|
|
class CommandsTests(unittest.TestCase):
|
|
def test_zte_matrix(self):
|
|
cmds = commands_for_vendor("ZTE", "zxros")
|
|
assert cmds is not None
|
|
self.assertEqual(cmds.brief, "show interface brief")
|
|
self.assertEqual(detail_command(cmds, "xgei-1/1/0/1"), "show interface xgei-1/1/0/1")
|
|
self.assertIsNone(commands_for_vendor("Cisco", "ios"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|