def test_primitive_wrapper_reset(self):
self.reg.register_primitive_wrapper(mac.MacWrapper())
self.reg.reset()
with self.assertRaisesRegex(
core.TinkError,
'No PrimitiveWrapper registered for primitive Mac.'):
self.reg.wrap(_mac_set([helper.FakeMac()]))
def test_wrap_success(self):
self.reg.register_primitive_wrapper(mac.MacWrapper())
mac1 = helper.FakeMac('FakeMac1')
mac2 = helper.FakeMac('FakeMac2')
wrapped_mac = self.reg.wrap(_mac_set([mac1, mac2]))
wrapped_mac.verify_mac(mac1.compute_mac(b'data1'), b'data1')
wrapped_mac.verify_mac(mac2.compute_mac(b'data2'), b'data2')
wrapped_mac.verify_mac(wrapped_mac.compute_mac(b'data'), b'data')
def _register_primitive_wrappers(): """Registers all primitive wrappers.""" register_primitive_wrapper = core.Registry.register_primitive_wrapper register_primitive_wrapper(aead.AeadWrapper()) register_primitive_wrapper(daead.DeterministicAeadWrapper()) register_primitive_wrapper(hybrid.HybridDecryptWrapper()) register_primitive_wrapper(hybrid.HybridEncryptWrapper()) register_primitive_wrapper(mac.MacWrapper()) register_primitive_wrapper(signature.PublicKeySignWrapper()) register_primitive_wrapper(signature.PublicKeyVerifyWrapper())
def test_register_different_primitive_wrappers_twice_fails(self):
self.reg.register_primitive_wrapper(mac.MacWrapper())
with self.assertRaisesRegex(
core.TinkError,
'A wrapper for primitive Mac has already been added.'):
self.reg.register_primitive_wrapper(DummyMacWrapper())
def test_register_same_primitive_wrapper_twice(self): self.reg.register_primitive_wrapper(mac.MacWrapper()) self.reg.register_primitive_wrapper(mac.MacWrapper())
