def testAreEncryptionKeysAvailableWithLoadKeysError(self):
self.mox.StubOutWithMock(crypto, 'CauliflowerVestReader', True)
self.mox.StubOutWithMock(crypto.CauliflowerVestReader, 'LoadKeys')
t = 'footype'
mock_reader = self.mox.CreateMockAnything()
crypto.CauliflowerVestReader().AndReturn(mock_reader)
mock_reader.LoadKeys(t).AndRaise(crypto.Error)
self.mox.ReplayAll()
self.assertFalse(crypto.AreEncryptionKeysAvailable(key_type=t))
self.mox.VerifyAll()
def testEncrypt(self):
self.mox.StubOutWithMock(crypto, 'CauliflowerVestReader', True)
self.mox.StubOutWithMock(crypto.CauliflowerVestReader, 'LoadKeys')
self.mox.StubOutWithMock(crypto.keyczar, 'Crypter', True)
data = 'foodata'
key_type = 'footype'
mock_reader = self.mox.CreateMockAnything()
mock_crypter = self.mox.CreateMockAnything()
crypto.CauliflowerVestReader().AndReturn(mock_reader)
mock_reader.LoadKeys(key_type).AndReturn(None)
crypto.keyczar.Crypter(reader=mock_reader).AndReturn(mock_crypter)
mock_crypter.Encrypt(data).AndReturn('result')
self.mox.ReplayAll()
r = crypto.Encrypt(data, encryption_key_type=key_type)
self.assertEqual(r, 'result')
self.mox.VerifyAll()
def setUp(self): super(CauliflowerVestReaderTest, self).setUp() self.r = crypto.CauliflowerVestReader()
def setUp(self):
mox.MoxTestBase.setUp(self)
self.stubs = stubout.StubOutForTesting()
self.r = crypto.CauliflowerVestReader()