def test_generate_rsa_public_and_private(self): pub, priv = pycrypto.generate_rsa_public_and_private() # Check format of 'pub' and 'priv'. self.assertEqual(None, tuf.formats.PEMRSA_SCHEMA.check_match(pub), FORMAT_ERROR_MSG) self.assertEqual(None, tuf.formats.PEMRSA_SCHEMA.check_match(priv), FORMAT_ERROR_MSG) # Check for invalid bits argument. bit >= 2048 and a multiple of 256. self.assertRaises(tuf.FormatError, pycrypto.generate_rsa_public_and_private, 1024) self.assertRaises(ValueError, pycrypto.generate_rsa_public_and_private, 2049) self.assertRaises(tuf.FormatError, pycrypto.generate_rsa_public_and_private, '2048')
from __future__ import print_function from __future__ import absolute_import from __future__ import division from __future__ import unicode_literals import unittest import logging import tuf import tuf.log import tuf.formats import tuf.pycrypto_keys as pycrypto logger = logging.getLogger('tuf.test_pycrypto_keys') public_rsa, private_rsa = pycrypto.generate_rsa_public_and_private() FORMAT_ERROR_MSG = 'tuf.FormatError raised. Check object\'s format.' class TestPycrypto_keys(unittest.TestCase): def setUp(self): pass def test_generate_rsa_public_and_private(self): pub, priv = pycrypto.generate_rsa_public_and_private() # Check format of 'pub' and 'priv'. self.assertEqual(None, tuf.formats.PEMRSA_SCHEMA.check_match(pub), FORMAT_ERROR_MSG) self.assertEqual(None, tuf.formats.PEMRSA_SCHEMA.check_match(priv),