diff --git a/virtualpdu/tests/snmp_error_indications.py b/virtualpdu/tests/snmp_error_indications.py index 9cf72d5..4b74736 100644 --- a/virtualpdu/tests/snmp_error_indications.py +++ b/virtualpdu/tests/snmp_error_indications.py @@ -175,3 +175,15 @@ class SNMPApplicationError(SNMPErrorIndication): class OidNotIncreasing(SNMPApplicationError): pass + + +class ReportPduReceived(SNMPErrorIndication): + pass + + +class UnknownUserName(SNMPErrorIndication): + pass + + +class WrongDigest(SNMPErrorIndication): + pass diff --git a/virtualpdu/tests/unit/test_snmp_client.py b/virtualpdu/tests/unit/test_snmp_client.py index c0159b5..bc4536e 100644 --- a/virtualpdu/tests/unit/test_snmp_client.py +++ b/virtualpdu/tests/unit/test_snmp_client.py @@ -17,6 +17,9 @@ from mock import sentinel from pysnmp.proto import errind from pysnmp.proto.errind import ErrorIndication from pysnmp.proto.rfc1905 import NoSuchInstance + +import testtools + from virtualpdu.tests import base from virtualpdu.tests import snmp_client from virtualpdu.tests import snmp_error_indications @@ -92,8 +95,8 @@ class TestSnmpClient(base.TestCase): exception_class = snmp_error_indications.__dict__.get(class_name) - self.assertRaises(exception_class, - self.snmp_client.get_one, oid) + with testtools.ExpectedException(exception_class): + self.snmp_client.get_one(oid) def test_set(self): oid = (1, 3, 6)