def setUp(self):
self.plugin = SimpleCryptoPlugin()
def setUp(self):
self.plugin = SimpleCryptoPlugin()
class WhenTestingSimpleCryptoPlugin(unittest.TestCase):
def setUp(self):
self.plugin = SimpleCryptoPlugin()
def test_pad_binary_string(self):
binary_string = b'some_binary_string'
padded_string = (
b'some_binary_string' +
b'\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e'
)
self.assertEqual(self.plugin._pad(binary_string), padded_string)
def test_pad_random_bytes(self):
random_bytes = Random.get_random_bytes(10)
padded_bytes = random_bytes + b'\x06\x06\x06\x06\x06\x06'
self.assertEqual(self.plugin._pad(random_bytes), padded_bytes)
def test_strip_padding_from_binary_string(self):
binary_string = b'some_binary_string'
padded_string = (
b'some_binary_string' +
b'\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e'
)
self.assertEqual(self.plugin._strip_pad(padded_string), binary_string)
def test_strip_padding_from_random_bytes(self):
random_bytes = Random.get_random_bytes(10)
padded_bytes = random_bytes + b'\x06\x06\x06\x06\x06\x06'
self.assertEqual(self.plugin._strip_pad(padded_bytes), random_bytes)
def test_encrypt_unicode_raises_value_error(self):
unencrypted = u'unicode_beer\U0001F37A'
secret = MagicMock()
secret.mime_type = 'text/plain'
with self.assertRaises(ValueError):
self.plugin.encrypt(unencrypted, MagicMock())
def test_byte_string_encryption(self):
unencrypted = b'some_secret'
encrypted, kek_metadata = self.plugin.encrypt(unencrypted, MagicMock())
decrypted = self.plugin.decrypt(encrypted, kek_metadata, MagicMock())
self.assertEqual(unencrypted, decrypted)
def test_random_bytes_encryption(self):
unencrypted = Random.get_random_bytes(10)
encrypted, kek_metadata = self.plugin.encrypt(unencrypted, MagicMock())
decrypted = self.plugin.decrypt(encrypted, kek_metadata, MagicMock())
self.assertEqual(unencrypted, decrypted)
def test_create_256_bit_key(self):
key = self.plugin.create("aes", 256)
self.assertEqual(len(key), 32)
def test_create_192_bit_key(self):
key = self.plugin.create("aes", 192)
self.assertEqual(len(key), 24)
def test_create_128_bit_key(self):
key = self.plugin.create("aes", 128)
self.assertEqual(len(key), 16)
def test_create_unsupported_bit_key(self):
with self.assertRaises(ValueError):
self.plugin.create("aes", 129)
class WhenTestingSimpleCryptoPlugin(unittest.TestCase):
def setUp(self):
self.plugin = SimpleCryptoPlugin()
def test_pad_binary_string(self):
binary_string = b"some_binary_string"
padded_string = b"some_binary_string" + b"\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e"
self.assertEqual(self.plugin._pad(binary_string), padded_string)
def test_pad_random_bytes(self):
random_bytes = Random.get_random_bytes(10)
padded_bytes = random_bytes + b"\x06\x06\x06\x06\x06\x06"
self.assertEqual(self.plugin._pad(random_bytes), padded_bytes)
def test_strip_padding_from_binary_string(self):
binary_string = b"some_binary_string"
padded_string = b"some_binary_string" + b"\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e\x0e"
self.assertEqual(self.plugin._strip_pad(padded_string), binary_string)
def test_strip_padding_from_random_bytes(self):
random_bytes = Random.get_random_bytes(10)
padded_bytes = random_bytes + b"\x06\x06\x06\x06\x06\x06"
self.assertEqual(self.plugin._strip_pad(padded_bytes), random_bytes)
def test_encrypt_unicode_raises_value_error(self):
unencrypted = u"unicode_beer\U0001F37A"
secret = MagicMock()
secret.mime_type = "text/plain"
with self.assertRaises(ValueError):
self.plugin.encrypt(unencrypted, MagicMock())
def test_byte_string_encryption(self):
unencrypted = b"some_secret"
encrypted, kek_metadata = self.plugin.encrypt(unencrypted, MagicMock())
decrypted = self.plugin.decrypt(encrypted, kek_metadata, MagicMock())
self.assertEqual(unencrypted, decrypted)
def test_random_bytes_encryption(self):
unencrypted = Random.get_random_bytes(10)
encrypted, kek_metadata = self.plugin.encrypt(unencrypted, MagicMock())
decrypted = self.plugin.decrypt(encrypted, kek_metadata, MagicMock())
self.assertEqual(unencrypted, decrypted)
def test_create_256_bit_key(self):
key = self.plugin.create("aes", 256)
self.assertEqual(len(key), 32)
def test_create_192_bit_key(self):
key = self.plugin.create("aes", 192)
self.assertEqual(len(key), 24)
def test_create_128_bit_key(self):
key = self.plugin.create("aes", 128)
self.assertEqual(len(key), 16)
def test_create_unsupported_bit_key(self):
with self.assertRaises(ValueError):
self.plugin.create("aes", 129)
