Example #1
0
class TestSetIdObj(AddressimoTestCase):
    def setUp(self):
        self.patcher1 = patch('addressimo.signer.LocalSigner.crypto')

        self.mockCrypto = self.patcher1.start()

        self.id_obj = Mock()

        self.ls = LocalSigner()

    def test_go_right(self):

        self.ls.set_id_obj(self.id_obj)

        # Validate calls
        self.assertEqual(self.mockCrypto.load_privatekey.return_value, self.ls.privkey)
        self.assertEqual(1, self.mockCrypto.load_privatekey.call_count)
        self.assertEqual(self.mockCrypto.FILETYPE_PEM, self.mockCrypto.load_privatekey.call_args[0][0])
        self.assertEqual(self.id_obj.private_key, self.mockCrypto.load_privatekey.call_args[0][1])

    def test_id_obj_missing(self):

        self.assertRaisesRegexp(ValueError, 'LocalSigner Requires ID Configuration', self.ls.set_id_obj, '')

    def test_private_key_missing(self):

        self.id_obj.private_key = None

        self.assertRaisesRegexp(ValueError, 'id_obj missing private key', self.ls.set_id_obj, self.id_obj)
Example #2
0
    def setUp(self):
        self.patcher1 = patch('addressimo.signer.LocalSigner.crypto')

        self.mockCrypto = self.patcher1.start()

        self.id_obj = Mock()
        self.ls = LocalSigner()
        self.ls.set_id_obj(self.id_obj)
Example #3
0
class TestFunctions(AddressimoTestCase):
    def setUp(self):
        self.patcher1 = patch('addressimo.signer.LocalSigner.crypto')

        self.mockCrypto = self.patcher1.start()

        self.id_obj = Mock()
        self.ls = LocalSigner()
        self.ls.set_id_obj(self.id_obj)

    def test_get_plugin_name(self):
        self.assertEqual('LOCAL', self.ls.get_plugin_name())

    def test_sign(self):

        ret_val = self.ls.sign('data')

        self.assertEqual(self.mockCrypto.sign.return_value, ret_val)
        self.assertEqual(self.mockCrypto.load_privatekey.return_value, self.mockCrypto.sign.call_args[0][0])
        self.assertEqual('data', self.mockCrypto.sign.call_args[0][1])
        self.assertEqual('sha256', self.mockCrypto.sign.call_args[0][2])

    def test_get_pki_type(self):
        self.assertEqual('x509+sha256', self.ls.get_pki_type())