def test_getPfcSubtree(self):
# Subtree for port
oid = ObjectIdentifier(32, 0, 0, 0, (1, 3, 6, 1, 4, 1, 9, 9, 813, 1, 1))
expected_oid = ObjectIdentifier(32, 0, 0, 0, (1, 3, 6, 1, 4, 1, 9, 9, 813, 1, 1, 1, 1, 1))
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET_NEXT, 16, 0, 42, 0, 0, 0),
oids=[oid]
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut_port)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.COUNTER_64)
self.assertEqual(str(value0.name), str(expected_oid))
self.assertEqual(value0.data, 4)
# Subtree for Priority
oid = ObjectIdentifier(33, 0, 0, 0, (1, 3, 6, 1, 4, 1, 9, 9, 813, 1, 2))
expected_oid = ObjectIdentifier(33, 0, 0, 0, (1, 3, 6, 1, 4, 1, 9, 9, 813, 1, 2, 1, 2, 1, 0))
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET_NEXT, 16, 0, 42, 0, 0, 0),
oids=[oid]
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut_prio)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.COUNTER_64)
self.assertEqual(str(value0.name), str(expected_oid))
self.assertEqual(value0.data, 209347219842134092490 % pow(2, 64)) # Test integer truncation
def test_endianness(self):
"""
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| h.version | h.type | h.flags | <reserved> |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
"""
one = 1
one_little_endian = one.to_bytes(4, 'little')
one_big_endian = one.to_bytes(4, 'big')
# flag unset
pdu_little_endian = PDUHeader.from_bytes(b'\x00\x00\x00\x00' + one_little_endian * 4)
self.assertEqual(pdu_little_endian.session_id, 1)
self.assertEqual(pdu_little_endian.transaction_id, 1)
self.assertEqual(pdu_little_endian.packet_id, 1)
self.assertEqual(pdu_little_endian.payload_length, 1)
print(pdu_little_endian)
nbo_set = struct.pack('!BBBB', 0, 0, 0 | PDUHeaderTags.MASK_NEWORK_BYTE_ORDER, 0)
# network byte flag set
pdu_big_endian = PDUHeader.from_bytes(nbo_set + one_big_endian * 4)
self.assertEqual(pdu_big_endian.session_id, 1)
self.assertEqual(pdu_big_endian.transaction_id, 1)
self.assertEqual(pdu_big_endian.packet_id, 1)
self.assertEqual(pdu_big_endian.payload_length, 1)
print(pdu_big_endian)
def test_bad_unpack(self):
with self.assertRaises(exceptions.PDUUnpackError):
# missing IDs
PDUHeader.from_bytes(b'\x00\x00\x00\x00')
with self.assertRaises(exceptions.PDUUnpackError):
# missing payload length
PDUHeader.from_bytes(b'\x00\x00\x00\x00' * 3)
def test_roundtrip(self):
"""
1.3.6.1.2.1.2.2.1.[1-22].7
index 10
6,2,2,0,(1,2,2,1,1,7)
upper bound 22
:return:
"""
register = RegisterPDU(
header=PDUHeader(1, PduTypes.REGISTER, 16, 0, 42, 0, 0, 0),
timeout=40,
priority=50,
range_subid=6,
subtree=ObjectIdentifier(4, 2, 0, 0, (1, 1, 1, 0)),
upper_bound=7
)
encoded = register.encode()
decoded = PDU.decode(encoded)
self.assertEqual(decoded.header.type_, PduTypes.REGISTER)
self.assertIsInstance(register, RegisterPDU)
self.assertEqual(register, decoded)
print(register)
def test_getpdu_xcvr_info(self):
sub_id = 1000 * 1 # sub id for Ethernet100
expected_mib = {
2: (ValueType.OCTET_STRING, "QSFP+ for Ethernet0"),
5: (ValueType.INTEGER, PhysicalClass.PORT),
7: (ValueType.OCTET_STRING, ""), # skip
8: (ValueType.OCTET_STRING, "A1"),
9: (ValueType.OCTET_STRING, ""), # skip
10: (ValueType.OCTET_STRING, ""), # skip
11: (ValueType.OCTET_STRING, "SERIAL_NUM"),
12: (ValueType.OCTET_STRING, "VENDOR_NAME"),
13: (ValueType.OCTET_STRING, "MODEL_NAME")
}
oids = [ObjectIdentifier(12, 0, 1, 0, (1, 3, 6, 1, 2, 1, 47, 1, 1, 1, 1, field_sub_id, sub_id))
for field_sub_id in expected_mib]
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=oids
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
for mib_key, value in zip(expected_mib, response.values):
expected_oid = ObjectIdentifier(12, 0, 1, 0, (1, 3, 6, 1, 2, 1, 47, 1, 1, 1, 1, mib_key, sub_id))
expected_type, expected_value = expected_mib[mib_key]
self.assertEqual(str(value.name), str(expected_oid))
self.assertEqual(value.type_, expected_type)
self.assertEqual(str(value.data), str(expected_value))
def test_getnextpdu_lag(self):
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=(ObjectIdentifier(
20, 0, 0, 0, (1, 3, 6, 1, 2, 1, 17, 7, 1, 2, 2, 1, 2, 101)), ))
response = get_pdu.make_response(self.lut)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.INTEGER)
self.assertEqual(value0.data, BaseIdx.portchannel_base_idx + 101)
def test_subtype_lldp_loc_man_addr_table(self):
oid = ObjectIdentifier(13, 0, 1, 0, (1, 0, 8802, 1, 1, 2, 1, 3, 8, 1, 3, 1, 4))
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid]
)
response = get_pdu.make_response(self.lut)
value0 = response.values[0]
self.assertEqual(str(value0.name), str(ObjectIdentifier(13, 0, 1, 0, (1, 0, 8802, 1, 1, 2, 1, 3, 8, 1, 3, 1, 4, 10, 224, 25, 26))))
self.assertEqual(value0.type_, ValueType.INTEGER)
self.assertEqual(value0.data, 5)
def test_getpdu_noinstance_eth0_docker0(self):
oid = ObjectIdentifier(
20, 0, 1, 0, (1, 3, 6, 1, 2, 1, 4, 22, 1, 2, 9, 240, 127, 1, 1))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
n = len(response.values)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.NO_SUCH_INSTANCE)
def test_getnextpdu_empty(self):
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=(ObjectIdentifier(12, 0, 0, 0,
(1, 3, 6, 1, 2, 1, 4, 24, 4, 1, 1, 255)), ))
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
n = len(response.values)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.END_OF_MIB_VIEW)
def test_getPsu2Status(self):
oid = ObjectIdentifier(2, 0, 0, 0,
(1, 3, 6, 1, 4, 1, 9, 9, 117, 1, 1, 2, 1, 2, 2))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.INTEGER)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data, 2)
def test_getnextpdu_local_port_identification(self):
# oid.include = 1
oid = ObjectIdentifier(11, 0, 1, 0,
(1, 0, 8802, 1, 1, 2, 1, 3, 7, 1, 3))
get_pdu = GetNextPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0,
0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.OCTET_STRING)
self.assertEqual(str(value0.data), "etp1")
def test_getpdu_internal_noinstance(self):
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=(ObjectIdentifier(
20, 0, 0, 0, (1, 3, 6, 1, 2, 1, 4, 24, 4, 1, 1, 0,
0, 0, 0, 0, 10, 10, 0, 5)), ))
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
n = len(response.values)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.NO_SUCH_INSTANCE)
def test_getnextpdu_lldpRemSysCapEnabled(self):
oid = ObjectIdentifier(12, 0, 1, 0, (1, 0, 8802, 1, 1, 2, 1, 4, 1, 1, 12, 1, 1))
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid]
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.OCTET_STRING)
self.assertEqual(str(value0.name), str(ObjectIdentifier(12, 0, 1, 0, (1, 0, 8802, 1, 1, 2, 1, 4, 1, 1, 12, 1, 1))))
self.assertEqual(str(value0.data), "\x28\x00")
def test_getpdu_disappeared(self):
oid = ObjectIdentifier(
20, 0, 0, 0, (1, 3, 6, 1, 4, 1, 9, 9, 187, 1, 2, 5, 1, 3, 2, 16,
252, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0x10))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.NO_SUCH_INSTANCE)
def test_getnextpdu_noeth(self):
# oid.include = 1
oid = ObjectIdentifier(
12, 0, 1, 0, (1, 0, 8802, 1, 1, 2, 1, 4, 1, 1, 7, 18545, 126, 1))
get_pdu = GetNextPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0,
0),
oids=[oid])
print("GetNextPDU sr=", get_pdu.sr)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.END_OF_MIB_VIEW)
def test_getpdu(self):
oid = ObjectIdentifier(
20, 0, 0, 0, (1, 3, 6, 1, 2, 1, 4, 22, 1, 2, 37, 10, 0, 0, 19))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.OCTET_STRING)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data.string, b'\x52\x54\x00\x04\x52\x5d')
def test_getnextpdu(self):
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=(ObjectIdentifier(
20, 0, 0, 0, (1, 3, 6, 1, 2, 1, 17, 7, 1, 2, 2, 1, 2, 999)), ))
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
n = len(response.values)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.INTEGER)
self.assertEqual(value0.data, 113)
def test_getPduIndicationForPort(self):
oid = ObjectIdentifier(32, 0, 0, 0,
(1, 3, 6, 1, 4, 1, 9, 9, 813, 1, 1, 1, 2, 1))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut_port)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.COUNTER_64)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data, 4)
def test_getPduRequestForPriority(self):
oid = ObjectIdentifier(8, 0, 0, 0, (1, 3, 6, 1, 4, 1, 9, 9, 813, 1, 2, 1, 2, 9, 0))
get_pdu = GetPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid]
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut_prio)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.COUNTER_64)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data, 209347219842134092490 % pow(2, 64)) # Test integer truncation
def test_getMulticastQueueCounters(self):
oid = ObjectIdentifier(
8, 0, 0, 0,
(1, 3, 6, 1, 4, 1, 9, 9, 580, 1, 5, 5, 1, 4, 5, 2, 1, 9))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.NO_SUCH_INSTANCE)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data, None)
def test_roundtrip(self):
open_pdu = OpenPDU(
header=PDUHeader(1, 1, 16, 0, 0, 0, 0, 0),
timeout=42,
oid=ObjectIdentifier(4, 2, 0, 0, (1, 1, 1, 0)),
descr="HOWDY-HO, NEIGHBOR!"
)
encoded = open_pdu.encode()
decoded = PDU.decode(encoded)
self.assertEqual(decoded.header.type_, PduTypes.OPEN)
self.assertIsInstance(open_pdu, OpenPDU)
self.assertEqual(open_pdu, decoded)
print(open_pdu)
def test_getnextpdu(self):
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=(ObjectIdentifier(
20, 0, 0, 0,
(1, 3, 6, 1, 2, 1, 4, 22, 1, 2, 37, 10, 0, 0, 19)), ))
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
n = len(response.values)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.OCTET_STRING)
self.assertEqual(value0.data.string, b'\x52\x54\x00\xd0\xa0\x8c')
def test_getpdu(self):
oid = ObjectIdentifier(20, 0, 0, 0,
(1, 3, 6, 1, 2, 1, 17, 7, 1, 2, 2, 1, 2, 1000,
124, 254, 144, 128, 159, 4))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.INTEGER)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data, 113)
def test_getpdu(self):
oid = ObjectIdentifier(24, 0, 1, 0, (1, 3, 6, 1, 2, 1, 4, 24, 4, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 10, 0, 0, 15))
get_pdu = GetPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid]
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.IP_ADDRESS)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(str(value0.data), ipaddress.ip_address("0.0.0.0").packed.decode())
def test_getpdu_established_asic2(self):
oid = ObjectIdentifier(
20, 0, 0, 0,
(1, 3, 6, 1, 4, 1, 9, 9, 187, 1, 2, 5, 1, 3, 1, 4, 10, 10, 0, 5))
get_pdu = GetPDU(header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.INTEGER)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data, 6)
def test_getnextpdu(self):
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=(ObjectIdentifier(21, 0, 0, 0,
(1, 3, 6, 1, 2, 1, 4, 24, 4, 1, 1, 0)), ))
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
print(response)
n = len(response.values)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.IP_ADDRESS)
self.assertEqual(str(value0.data),
ipaddress.ip_address("0.0.0.0").packed.decode())
def test_getNextPsu3(self):
oid = ObjectIdentifier(2, 0, 0, 0,
(1, 3, 6, 1, 4, 1, 9, 9, 117, 1, 1, 2, 1, 2, 3))
expected_oid = None
get_pdu = GetNextPDU(header=PDUHeader(1, PduTypes.GET_NEXT, 16, 0, 42,
0, 0, 0),
oids=[oid])
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
value0 = response.values[0]
self.assertEqual(value0.type_, ValueType.END_OF_MIB_VIEW)
self.assertEqual(str(value0.name), str(oid))
self.assertEqual(value0.data, None)
def test_roundtrip(self):
response_pdu = ResponsePDU(
header=PDUHeader(1, 18, 16, 0, 42, 0, 0, 8),
sys_up_time=0,
error=0,
index=0,
)
encoded = response_pdu.encode()
decoded = PDU.decode(encoded)
self.assertEqual(decoded.header.type_, PduTypes.RESPONSE)
self.assertIsInstance(response_pdu, ResponsePDU)
self.assertEqual(response_pdu, decoded)
print(response_pdu)
def test_getnextpdu_chassis_serial_number(self):
oid = ObjectIdentifier(12, 0, 1, 0, (1, 3, 6, 1, 2, 1, 47, 1, 1, 1, 1, 11))
expected_oid = ObjectIdentifier(12, 0, 1, 0, (1, 3, 6, 1, 2, 1, 47, 1, 1, 1, 1, 11, 1))
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET_NEXT, 16, 0, 42, 0, 0, 0),
oids=[oid]
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
value0 = response.values[0]
self.assertEqual(str(value0.name), str(expected_oid))
self.assertEqual(value0.type_, ValueType.OCTET_STRING)
self.assertEqual(str(value0.data), "SAMPLETESTSN")
def test_getpdu_xcvr_dom(self):
expected_mib = {
1000 * 1 + 1: "DOM Temperature Sensor for Ethernet0",
1000 * 1 + 2: "DOM Voltage Sensor for Ethernet0",
1000 * 1 + 11: "DOM RX Power Sensor for Ethernet0/1",
1000 * 1 + 21: "DOM RX Power Sensor for Ethernet0/2",
1000 * 1 + 31: "DOM RX Power Sensor for Ethernet0/3",
1000 * 1 + 41: "DOM RX Power Sensor for Ethernet0/4",
1000 * 1 + 12: "DOM TX Bias Sensor for Ethernet0/1",
1000 * 1 + 22: "DOM TX Bias Sensor for Ethernet0/2",
1000 * 1 + 32: "DOM TX Bias Sensor for Ethernet0/3",
1000 * 1 + 42: "DOM TX Bias Sensor for Ethernet0/4",
1000 * 1 + 13: "DOM TX Power Sensor for Ethernet0/1",
1000 * 1 + 23: "DOM TX Power Sensor for Ethernet0/2",
1000 * 1 + 33: "DOM TX Power Sensor for Ethernet0/3",
1000 * 1 + 43: "DOM TX Power Sensor for Ethernet0/4",
}
phyDescr, phyClass = 2, 5
# check physical class value
oids = [ObjectIdentifier(12, 0, 1, 0, (1, 3, 6, 1, 2, 1, 47, 1, 1, 1, 1, field_id, sub_id))
for field_id in (phyDescr, phyClass)
for sub_id in expected_mib]
get_pdu = GetNextPDU(
header=PDUHeader(1, PduTypes.GET, 16, 0, 42, 0, 0, 0),
oids=oids
)
encoded = get_pdu.encode()
response = get_pdu.make_response(self.lut)
for mib_key, value in zip(expected_mib, response.values):
if phyClass == value.name.subids[-2]:
sub_id = (phyClass, mib_key)
expected_type, expected_value = ValueType.INTEGER, PhysicalClass.SENSOR
elif phyDescr == value.name.subids[-2]:
sub_id = (phyDescr, mib_key)
expected_type, expected_value = ValueType.OCTET_STRING, expected_mib[mib_key]
else:
# received unexpected
self.assertTrue(False)
expected_oid = ObjectIdentifier(12, 0, 1, 0, (1, 3, 6, 1, 2, 1, 47, 1, 1, 1, 1, *sub_id))
self.assertEqual(str(value.name), str(expected_oid))
self.assertEqual(value.type_, expected_type)
self.assertEqual(str(value.data), str(expected_value))
