test: added test for better coverage

Author: rapejim

tests/test_crypt.py

@@ -7,7 +7,8 @@
 
 from pyms.cloud.aws.kms import Crypt as CryptAws
 from pyms.config import get_conf
-from pyms.constants import LOGGER_NAME, CONFIGMAP_FILE_ENVIRONMENT, CRYPT_FILE_KEY_ENVIRONMENT, CONFIG_BASE
+from pyms.constants import LOGGER_NAME, CONFIGMAP_FILE_ENVIRONMENT, CRYPT_FILE_KEY_ENVIRONMENT, CONFIG_BASE, \
+    CRYPT_FILE_KEY_ENVIRONMENT_LEGACY
 from pyms.crypt.driver import CryptAbstract, CryptResource
 from pyms.crypt.fernet import Crypt as CryptFernet
 from pyms.exceptions import FileDoesNotExistException
@@ -78,7 +79,10 @@ def setUp(self):
 
     def tearDown(self):
         del os.environ[CONFIGMAP_FILE_ENVIRONMENT]
-        del os.environ[CRYPT_FILE_KEY_ENVIRONMENT]
+        if os.getenv(CRYPT_FILE_KEY_ENVIRONMENT):
+            del os.environ[CRYPT_FILE_KEY_ENVIRONMENT]
+        if os.getenv(CRYPT_FILE_KEY_ENVIRONMENT_LEGACY):
+            del os.environ[CRYPT_FILE_KEY_ENVIRONMENT_LEGACY]
 
     def test_encrypt_conf(self):
         crypt = CryptFernet(path=self.BASE_DIR)
@@ -87,6 +91,15 @@ def test_encrypt_conf(self):
         crypt.delete_key()
         assert config.database_url == "http://database-url"
 
+    def test_encrypt_conf_deprecated_env(self):
+        os.environ[CRYPT_FILE_KEY_ENVIRONMENT_LEGACY] = os.getenv(CRYPT_FILE_KEY_ENVIRONMENT)
+        del os.environ[CRYPT_FILE_KEY_ENVIRONMENT]
+        crypt = CryptFernet(path=self.BASE_DIR)
+        crypt._loader.put_file(b"9IXx2F5d5Ob-h5xdCnFSUXhuFKLGRibvLfSbixpcfCw=", "wb")
+        config = get_conf(service=CONFIG_BASE, uppercase=True, crypt=CryptFernet)
+        crypt.delete_key()
+        assert config.database_url == "http://database-url"
+
 
 class GetConfigEncryptedAWS(unittest.TestCase):
     BASE_DIR = os.path.dirname(os.path.abspath(__file__))