1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
import pytest
from pysnmp.hlapi.v3arch.asyncio import *
from pysnmp.proto.errind import UnknownUserName, WrongDigest
from tests.agent_context import AGENT_PORT, AgentContextManager
@pytest.mark.asyncio
async def test_usm_sha_none():
async with AgentContextManager():
with SnmpEngine() as snmpEngine:
authData = UsmUserData(
"usr-sha-none",
"authkey1",
authProtocol=USM_AUTH_HMAC96_SHA,
)
errorIndication, errorStatus, errorIndex, varBinds = await get_cmd(
snmpEngine,
authData,
await UdpTransportTarget.create(("localhost", AGENT_PORT), retries=0),
ContextData(),
ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
)
assert errorIndication is None
assert errorStatus == 0
assert len(varBinds) == 1
assert varBinds[0][0].prettyPrint() == "SNMPv2-MIB::sysDescr.0"
isinstance(varBinds[0][1], OctetString)
@pytest.mark.asyncio
async def test_usm_sha_none_wrong_auth():
async with AgentContextManager():
with SnmpEngine() as snmpEngine:
authData = UsmUserData(
"usr-sha-none",
"authkey1",
authProtocol=USM_AUTH_HMAC96_MD5, # wrongly use usmHMACMD5AuthProtocol
)
errorIndication, errorStatus, errorIndex, varBinds = await get_cmd(
snmpEngine,
authData,
await UdpTransportTarget.create(("localhost", AGENT_PORT), retries=0),
ContextData(),
ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
)
assert isinstance(errorIndication, WrongDigest)
assert str(errorIndication) == "Wrong SNMP PDU digest"
@pytest.mark.asyncio
async def test_usm_sha_none_wrong_user():
async with AgentContextManager():
with SnmpEngine() as snmpEngine:
authData = UsmUserData(
"usr-sha-none-not-exist",
"authkey1",
authProtocol=USM_AUTH_HMAC96_SHA,
)
errorIndication, errorStatus, errorIndex, varBinds = await get_cmd(
snmpEngine,
authData,
await UdpTransportTarget.create(("localhost", AGENT_PORT), retries=0),
ContextData(),
ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
)
assert isinstance(errorIndication, UnknownUserName)
assert str(errorIndication) == "Unknown USM user"
|
