def test_clear(self):
"""EphemeralUnencryptedDriver - Clear Credentials"""
descriptor = 'descriptor'
credentials = Credentials('aaaa', False)
self._ep_driver.save_credentials(descriptor, credentials)
EphemeralUnencryptedDriver.clear()
assert_false(self._ep_driver.has_credentials(descriptor))
def test_save_and_load_credentials_persists_statically(self):
"""EphemeralUnencryptedDriver - Save and Load Credentials"""
descriptor = 'descriptor'
credentials = Credentials('aaaa', False)
assert_true(self._ep_driver.save_credentials(descriptor, credentials))
driver2 = EphemeralUnencryptedDriver('service') # Create a separate, identical driver
loaded_credentials = driver2.load_credentials(descriptor)
assert_is_not_none(loaded_credentials)
assert_false(loaded_credentials.is_encrypted())
assert_equal(loaded_credentials.data(), 'aaaa')
def setup(self):
EphemeralUnencryptedDriver.clear()
self._ep_driver = EphemeralUnencryptedDriver('service')
def teardown():
EphemeralUnencryptedDriver.clear()
class TestEphemeralUnencryptedDriver(object):
def setup(self):
EphemeralUnencryptedDriver.clear()
self._ep_driver = EphemeralUnencryptedDriver('service')
@staticmethod
def teardown():
EphemeralUnencryptedDriver.clear()
def test_save_and_has_credentials(self):
"""EphemeralUnencryptedDriver - Save and Has Credentials"""
assert_false(self._ep_driver.has_credentials('descriptor'))
credentials = Credentials('aaaa', False)
assert_true(self._ep_driver.save_credentials('descriptor', credentials))
assert_true(self._ep_driver.has_credentials('descriptor'))
def test_save_and_load_credentials(self):
"""EphemeralUnencryptedDriver - Save and Load Credentials"""
descriptor = 'descriptor'
credentials = Credentials('aaaa', False)
assert_true(self._ep_driver.save_credentials(descriptor, credentials))
loaded_credentials = self._ep_driver.load_credentials(descriptor)
assert_is_not_none(loaded_credentials)
assert_false(loaded_credentials.is_encrypted())
assert_equal(loaded_credentials.data(), 'aaaa')
def test_save_and_load_credentials_persists_statically(self):
"""EphemeralUnencryptedDriver - Save and Load Credentials"""
descriptor = 'descriptor'
credentials = Credentials('aaaa', False)
assert_true(self._ep_driver.save_credentials(descriptor, credentials))
driver2 = EphemeralUnencryptedDriver('service') # Create a separate, identical driver
loaded_credentials = driver2.load_credentials(descriptor)
assert_is_not_none(loaded_credentials)
assert_false(loaded_credentials.is_encrypted())
assert_equal(loaded_credentials.data(), 'aaaa')
@mock_kms
def test_save_automatically_decrypts(self):
"""EphemeralUnencryptedDriver - Save Automatically Decrypts"""
raw_credentials_dict = {
'python': 'is very difficult',
'someone': 'save meeeee',
}
descriptor = 'descriptor5'
raw_credentials = json.dumps(raw_credentials_dict)
encrypted_raw_credentials = encrypt_with_kms(raw_credentials, REGION, KMS_ALIAS)
credentials = Credentials(encrypted_raw_credentials, True, REGION)
assert_true(self._ep_driver.save_credentials(descriptor, credentials))
loaded_credentials = self._ep_driver.load_credentials(descriptor)
assert_is_not_none(loaded_credentials)
assert_false(loaded_credentials.is_encrypted())
assert_equal(json.loads(loaded_credentials.data()), raw_credentials_dict)
def test_clear(self):
"""EphemeralUnencryptedDriver - Clear Credentials"""
descriptor = 'descriptor'
credentials = Credentials('aaaa', False)
self._ep_driver.save_credentials(descriptor, credentials)
EphemeralUnencryptedDriver.clear()
assert_false(self._ep_driver.has_credentials(descriptor))