Added more test cases.

lextudio · Feb 4, 2024 · 806132b · 806132b
1 parent ccfd8c5
commit 806132b
Show file tree

Hide file tree

Showing 6 changed files with 112 additions and 72 deletions.
diff --git a/tests/hlapi/asyncio/manager/cmdgen/test_usm_none_none.py b/tests/hlapi/asyncio/manager/cmdgen/test_usm_none_none.py
@@ -1,5 +1,6 @@
 import pytest
 from pysnmp.hlapi.asyncio import *
+from pysnmp.proto.errind import UnknownUserName
 
 @pytest.mark.asyncio
 async def test_usm_no_auth_no_priv():
@@ -19,4 +20,22 @@ async def test_usm_no_auth_no_priv():
     assert errorStatus == 0
     assert len(varBinds) == 1
     assert varBinds[0][0].prettyPrint() == "SNMPv2-MIB::sysDescr.0"
-    isinstance(varBinds[0][1], OctetString)
+    isinstance(varBinds[0][1], OctetString)
+
+@pytest.mark.asyncio
+async def test_usm_no_auth_no_priv_wrong_user():
+    snmpEngine = SnmpEngine()
+    authData = UsmUserData(
+        "usr-none-none-not-exist"
+    )
+    errorIndication, errorStatus, errorIndex, varBinds = await getCmd(
+        snmpEngine,
+        authData,
+        UdpTransportTarget(("demo.pysnmp.com", 161), retries=0),
+        ContextData(),
+        ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
+    )
+
+    assert isinstance(errorIndication, UnknownUserName)
+    assert str(errorIndication) == 'Unknown USM user'
+    snmpEngine.transportDispatcher.closeDispatcher()
diff --git a/tests/hlapi/asyncio/manager/cmdgen/test_usm_none_none_wrong_user.py b/tests/hlapi/asyncio/manager/cmdgen/test_usm_none_none_wrong_user.py
diff --git a/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_aes128.py b/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_aes128.py
@@ -1,5 +1,6 @@
 import pytest
 from pysnmp.hlapi.asyncio import *
+from pysnmp.proto.errind import RequestTimedOut, UnknownUserName
 
 @pytest.mark.asyncio
 async def test_usm_sha_aes128():
@@ -26,3 +27,72 @@ async def test_usm_sha_aes128():
     isinstance(varBinds[0][1], OctetString)
 
     snmpEngine.transportDispatcher.closeDispatcher()
+
+@pytest.mark.asyncio
+async def test_usm_sha_aes128_wrong_auth():
+    snmpEngine = SnmpEngine()
+    authData = UsmUserData(
+        "usr-sha-aes",
+        "authkey1",
+        "privkey1",
+        authProtocol=usmHMACMD5AuthProtocol, # wrongly use usmHMACMD5AuthProtocol
+        privProtocol=usmAesCfb128Protocol
+    )
+    errorIndication, errorStatus, errorIndex, varBinds = await getCmd(
+        snmpEngine,
+        authData,
+        UdpTransportTarget(("demo.pysnmp.com", 161), retries=0),
+        ContextData(),
+        ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
+    )
+
+    assert isinstance(errorIndication, RequestTimedOut)
+    assert str(errorIndication) == 'No SNMP response received before timeout'
+
+    snmpEngine.transportDispatcher.closeDispatcher()
+
+@pytest.mark.asyncio
+async def test_usm_sha_aes128_wrong_priv():
+    snmpEngine = SnmpEngine()
+    authData = UsmUserData(
+        "usr-sha-aes",
+        "authkey1",
+        "privkey1",
+        authProtocol=usmHMACSHAAuthProtocol,
+        privProtocol=usmDESPrivProtocol, # wrongly use usmDESPrivProtocol
+    )
+    errorIndication, errorStatus, errorIndex, varBinds = await getCmd(
+        snmpEngine,
+        authData,
+        UdpTransportTarget(("demo.pysnmp.com", 161), retries=0),
+        ContextData(),
+        ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
+    )
+
+    assert isinstance(errorIndication, RequestTimedOut)
+    assert str(errorIndication) == 'No SNMP response received before timeout'
+
+    snmpEngine.transportDispatcher.closeDispatcher()
+
+@pytest.mark.asyncio
+async def test_usm_sha_aes128_wrong_user():
+    snmpEngine = SnmpEngine()
+    authData = UsmUserData(
+        "usr-sha-aes-not-exist",
+        "authkey1",
+        "privkey1",
+        authProtocol=usmHMACSHAAuthProtocol,
+        privProtocol=usmAesCfb128Protocol,
+    )
+    errorIndication, errorStatus, errorIndex, varBinds = await getCmd(
+        snmpEngine,
+        authData,
+        UdpTransportTarget(("demo.pysnmp.com", 161), retries=0),
+        ContextData(),
+        ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
+    )
+
+    assert isinstance(errorIndication, UnknownUserName)
+    assert str(errorIndication) == 'Unknown USM user'
+
+    snmpEngine.transportDispatcher.closeDispatcher()
diff --git a/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_aes128_wrong_user.py b/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_aes128_wrong_user.py
diff --git a/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_none.py b/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_none.py
@@ -1,5 +1,6 @@
 import pytest
 from pysnmp.hlapi.asyncio import *
+from pysnmp.proto.errind import UnknownUserName
 
 @pytest.mark.asyncio
 async def test_usm_sha_none():
@@ -24,3 +25,24 @@ async def test_usm_sha_none():
     isinstance(varBinds[0][1], OctetString)
 
     snmpEngine.transportDispatcher.closeDispatcher()
+
+@pytest.mark.asyncio
+async def test_usm_sha_none_wrong_user():
+    snmpEngine = SnmpEngine()
+    authData = UsmUserData(
+        "usr-sha-none-not-exist",
+        "authkey1",
+        authProtocol=usmHMACSHAAuthProtocol,
+    )
+    errorIndication, errorStatus, errorIndex, varBinds = await getCmd(
+        snmpEngine,
+        authData,
+        UdpTransportTarget(("demo.pysnmp.com", 161), retries=0),
+        ContextData(),
+        ObjectType(ObjectIdentity("SNMPv2-MIB", "sysDescr", 0)),
+    )
+
+    assert isinstance(errorIndication, UnknownUserName)
+    assert str(errorIndication) == 'Unknown USM user'
+
+    snmpEngine.transportDispatcher.closeDispatcher()
diff --git a/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_none_wrong_user.py b/tests/hlapi/asyncio/manager/cmdgen/test_usm_sha_none_wrong_user.py