diff options
Diffstat (limited to 'tests/hlapi/test_auth.py')
| -rw-r--r-- | tests/hlapi/test_auth.py | 143 |
1 files changed, 143 insertions, 0 deletions
diff --git a/tests/hlapi/test_auth.py b/tests/hlapi/test_auth.py new file mode 100644 index 00000000..ad5323fd --- /dev/null +++ b/tests/hlapi/test_auth.py @@ -0,0 +1,143 @@ +# +# This file is part of pysnmp software. +# +# Copyright (c) 2005-2018, Ilya Etingof <etingof@gmail.com> +# License: http://snmplabs.com/pysnmp/license.html +# +import sys + +try: + import unittest2 as unittest + +except ImportError: + import unittest + +from tests.base import BaseTestCase + +from pysnmp import hlapi +from pysnmp import debug +from pysnmp import error + + +class ProtocolConstantsTestCase(BaseTestCase): + def testEnsureProtocols(self): + self.assertTrue(hasattr(hlapi, 'usm3DESEDEPrivProtocol')) + self.assertTrue(hasattr(hlapi, 'usm3DESEDEPrivProtocol')) + self.assertTrue(hasattr(hlapi, 'usmAesCfb128Protocol')) + self.assertTrue(hasattr(hlapi, 'usmAesCfb192Protocol')) + self.assertTrue(hasattr(hlapi, 'usmAesCfb256Protocol')) + self.assertTrue(hasattr(hlapi, 'usmAesBlumenthalCfb192Protocol')) + self.assertTrue(hasattr(hlapi, 'usmAesBlumenthalCfb256Protocol')) + self.assertTrue(hasattr(hlapi, 'usmDESPrivProtocol')) + self.assertTrue(hasattr(hlapi, 'usmHMACMD5AuthProtocol')) + self.assertTrue(hasattr(hlapi, 'usmHMACSHAAuthProtocol')) + self.assertTrue(hasattr(hlapi, 'usmHMAC128SHA224AuthProtocol')) + self.assertTrue(hasattr(hlapi, 'usmHMAC192SHA256AuthProtocol')) + self.assertTrue(hasattr(hlapi, 'usmHMAC256SHA384AuthProtocol')) + self.assertTrue(hasattr(hlapi, 'usmHMAC384SHA512AuthProtocol')) + self.assertTrue(hasattr(hlapi, 'usmNoAuthProtocol')) + self.assertTrue(hasattr(hlapi, 'usmNoPrivProtocol')) + + +class CommunityDataTestCase(BaseTestCase): + def testVersionOne(self): + auth = hlapi.CommunityData( + 'public', mpModel=0, contextEngineId='0123456789', contextName='abc', tag='x' + ) + self.assertTrue(auth.communityIndex) + self.assertEqual(auth.communityName, 'public') + self.assertEqual(auth.contextEngineId, '0123456789') + self.assertEqual(auth.contextName, 'abc') + self.assertEqual(auth.mpModel, 0) + self.assertEqual(auth.securityLevel, 'noAuthNoPriv') + self.assertEqual(auth.securityModel, 1) + self.assertTrue(auth.securityName) + self.assertEqual(auth.tag, 'x') + + def testVersionTwoC(self): + auth = hlapi.CommunityData( + 'public', contextEngineId='0123456789', contextName='abc', tag='x' + ) + self.assertTrue(auth.communityIndex) + self.assertEqual(auth.communityName, 'public') + self.assertEqual(auth.contextEngineId, '0123456789') + self.assertEqual(auth.contextName, 'abc') + self.assertEqual(auth.mpModel, 1) + self.assertEqual(auth.securityLevel, 'noAuthNoPriv') + self.assertEqual(auth.securityModel, 2) + self.assertTrue(auth.securityName) + self.assertEqual(auth.tag, 'x') + + def testClone(self): + auth = hlapi.CommunityData('public').clone('private') + self.assertTrue(auth.communityIndex) + self.assertEqual(auth.communityName, 'private') + self.assertEqual(auth.contextEngineId, None) + self.assertEqual(auth.contextName, hlapi.OctetString('')) + self.assertEqual(auth.mpModel, 1) + self.assertEqual(auth.securityLevel, 'noAuthNoPriv') + self.assertEqual(auth.securityModel, 2) + self.assertTrue(auth.securityName) + self.assertEqual(auth.tag, hlapi.OctetString('')) + + +class UsmUserDataTestCase(BaseTestCase): + def testAuthPrivDefaults(self): + auth = hlapi.UsmUserData('testuser', 'authkey1', 'privkey1') + self.assertEqual(auth.authKey, 'authkey1') + self.assertEqual(auth.authProtocol, (1, 3, 6, 1, 6, 3, 10, 1, 1, 2)) + self.assertEqual(auth.mpModel, 3) + self.assertEqual(auth.privKey, 'privkey1') + self.assertEqual(auth.privProtocol, (1, 3, 6, 1, 6, 3, 10, 1, 2, 2)) + self.assertFalse(auth.securityEngineId) + self.assertEqual(auth.securityLevel, 'authPriv') + self.assertEqual(auth.securityModel, 3) + self.assertEqual(auth.securityName, 'testuser') + self.assertEqual(auth.userName, 'testuser') + + def testAuthNoPrivDefaults(self): + auth = hlapi.UsmUserData('testuser', 'authkey1') + self.assertEqual(auth.authKey, 'authkey1') + self.assertEqual(auth.authProtocol, (1, 3, 6, 1, 6, 3, 10, 1, 1, 2)) + self.assertEqual(auth.mpModel, 3) + self.assertFalse(auth.privKey) + self.assertEqual(auth.privProtocol, (1, 3, 6, 1, 6, 3, 10, 1, 2, 1)) + self.assertFalse(auth.securityEngineId) + self.assertEqual(auth.securityLevel, 'authNoPriv') + self.assertEqual(auth.securityModel, 3) + self.assertEqual(auth.securityName, 'testuser') + self.assertEqual(auth.userName, 'testuser') + + def testNoAuthNoPrivDefaults(self): + auth = hlapi.UsmUserData('testuser') + self.assertFalse(auth.authKey) + self.assertEqual(auth.authProtocol, (1, 3, 6, 1, 6, 3, 10, 1, 1, 1)) + self.assertEqual(auth.mpModel, 3) + self.assertFalse(auth.privKey) + self.assertEqual(auth.privProtocol, (1, 3, 6, 1, 6, 3, 10, 1, 2, 1)) + self.assertFalse(auth.securityEngineId) + self.assertEqual(auth.securityLevel, 'noAuthNoPriv') + self.assertEqual(auth.securityModel, 3) + self.assertEqual(auth.securityName, 'testuser') + self.assertEqual(auth.userName, 'testuser') + + def testAuthPriv(self): + auth = hlapi.UsmUserData('testuser', 'authkey1', 'privkey1', + authProtocol=hlapi.usmHMACSHAAuthProtocol, + privProtocol=hlapi.usmAesCfb256Protocol) + self.assertEqual(auth.authKey, 'authkey1') + self.assertEqual(auth.authProtocol, (1, 3, 6, 1, 6, 3, 10, 1, 1, 3)) + self.assertEqual(auth.mpModel, 3) + self.assertEqual(auth.privKey, 'privkey1') + self.assertEqual(auth.privProtocol, (1, 3, 6, 1, 4, 1, 9, 12, 6, 1, 102)) + self.assertFalse(auth.securityEngineId) + self.assertEqual(auth.securityLevel, 'authPriv') + self.assertEqual(auth.securityModel, 3) + self.assertEqual(auth.securityName, 'testuser') + self.assertEqual(auth.userName, 'testuser') + + +suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__]) + +if __name__ == '__main__': + unittest.TextTestRunner(verbosity=2).run(suite) |