def test_encrypt_listener_dict(self, mock_fernet):
mock_fern = mock.MagicMock()
mock_fernet.return_value = mock_fern
TEST_DATA = {'cert': b'some data'}
TEST_DATA2 = {'test': 'more data'}
FAKE_ENCRYPTED_DATA = b'alqwkhjetrhth'
mock_fern.encrypt.return_value = FAKE_ENCRYPTED_DATA
# We need a class instance with the mock
amp_driver = driver.AmphoraProviderDriver()
# Test just default_tls_container_data
list_dict = {consts.DEFAULT_TLS_CONTAINER_DATA: TEST_DATA}
amp_driver._encrypt_listener_dict(list_dict)
mock_fern.encrypt.assert_called_once_with(b'some data')
self.assertEqual({'cert': FAKE_ENCRYPTED_DATA},
list_dict[consts.DEFAULT_TLS_CONTAINER_DATA])
mock_fern.reset_mock()
# Test just sni_container_data
TEST_DATA = {'cert': b'some data'}
sni_dict = {consts.SNI_CONTAINER_DATA: [TEST_DATA, TEST_DATA2]}
amp_driver._encrypt_listener_dict(sni_dict)
mock_fern.encrypt.assert_called_once_with(b'some data')
encrypted_sni = [{'cert': FAKE_ENCRYPTED_DATA}, TEST_DATA2]
self.assertEqual(encrypted_sni, sni_dict[consts.SNI_CONTAINER_DATA])
def test_encrypt_listener_dict(self, mock_fernet):
mock_fern = mock.MagicMock()
mock_fernet.return_value = mock_fern
TEST_DATA = 'some data'
TEST_DATA2 = 'more data'
FAKE_ENCRYPTED_DATA = 'alqwkhjetrhth'
mock_fern.encrypt.return_value = FAKE_ENCRYPTED_DATA
# We need a class instance with the mock
amp_driver = driver.AmphoraProviderDriver()
# Test just default_tls_container_data
list_dict = {consts.DEFAULT_TLS_CONTAINER_DATA: TEST_DATA}
amp_driver._encrypt_listener_dict(list_dict)
mock_fern.encrypt.assert_called_once_with(TEST_DATA)
self.assertEqual(FAKE_ENCRYPTED_DATA,
list_dict[consts.DEFAULT_TLS_CONTAINER_DATA])
mock_fern.reset_mock()
# Test just sni_container_data
list_dict = {consts.SNI_CONTAINER_DATA: [TEST_DATA, TEST_DATA2]}
amp_driver._encrypt_listener_dict(list_dict)
calls = [mock.call(TEST_DATA), mock.call(TEST_DATA2)]
mock_fern.encrypt.assert_has_calls(calls)
encrypted_sni = [FAKE_ENCRYPTED_DATA, FAKE_ENCRYPTED_DATA]
self.assertEqual(encrypted_sni, list_dict[consts.SNI_CONTAINER_DATA])
def setUp(self):
super().setUp()
self.amp_driver = driver.AmphoraProviderDriver()
self.sample_data = sample_data_models.SampleDriverDataModels()
