def test_set_values_1(self): module = Mock() self.assertTrue(isinstance(module.MY_SECRET_KEY1, Mock)) set_values(KEYSET_PATH, ENCRYPTED_SETTINGS_MODULES, module, ['MY_SECRET_KEY1', 'MY_SECRET_KEY3']) self.assertEqual(module.MY_SECRET_KEY1, 'poniesrox1') self.assertEqual(module.MY_SECRET_KEY3, 'poniesrox3')
def test_set_values_2(self): module = Mock() self.assertTrue(isinstance(module.MY_SECRET_KEY1, Mock)) self.assertTrue(isinstance(module.MY_SECRET_KEY2, Mock)) self.assertTrue(isinstance(module.MY_SECRET_KEY3, Mock)) set_values(KEYSET_PATH, ENCRYPTED_SETTINGS_MODULES, module, SETTING_NAME_TO_DECRYPTED_VALUE_MAP.keys()) for key, expected in SETTING_NAME_TO_DECRYPTED_VALUE_MAP.items(): actual = getattr(module, key) self.assertEqual(actual, expected)
""" Method 3 - Automatically decrypt values and set attributes on the current module. """ import sys from encrypted_settings.config import set_values KEYSET_PATH = '/my/path/to/keyczar_keyset' ENCRYPTED_SETTINGS_MODULE = 'myapp.configs.encrypted_settings' FACEBOOK_API_KEY = None set_values(KEYSET_PATH, ENCRYPTED_SETTINGS_MODULE, sys.modules[__name__], ['FACEBOOK_API_KEY'])