def test_env_aws_retrieve_no_token(self):
# clear environement
os.environ.clear()
# set environment variables
os.environ["AWS_ACCESS_KEY_ID"] = "access"
os.environ["AWS_SECRET_ACCESS_KEY"] = "secret"
# Init new env_aws provider
provider = EnvAWS()
# assert expired true for newly created provider
eq_(provider.is_expired(), True)
# retrieve rpovider credentials
creds = provider.retrieve()
# assert expected data
expected_creds = Value(access_key="access",
secret_key="secret",
session_token=None)
eq_(creds.access_key, expected_creds.access_key)
eq_(creds.secret_key, expected_creds.secret_key)
eq_(creds.session_token, expected_creds.session_token)
# assert expired true for retrieved credentials
eq_(provider.is_expired(), False)
def test_chain_is_expired(self):
# clear environment
os.environ.clear()
# prepare env for env_aws
os.environ["AWS_ACCESS_KEY_ID"] = "access_aws"
os.environ["AWS_SECRET_ACCESS_KEY"] = "secret_aws"
# create chain provider
chain = Chain(providers=[EnvAWS()])
# is_expired should be True before retrieve()
eq_(chain.is_expired(), True)
# retieve single env_aws provider
chain.retrieve()
# is_expired should be False after retrieve()
eq_(chain.is_expired(), False)
def test_chain_retrieve(self):
# clear environment
os.environ.clear()
# prepare env for env_aws provider
os.environ["AWS_ACCESS_KEY_ID"] = "access_aws"
os.environ["AWS_SECRET_ACCESS_KEY"] = "secret_aws"
os.environ["AWS_SESSION_TOKEN"] = "token_aws"
# prepare env for env_minio
os.environ['MINIO_ACCESS_KEY'] = "access_minio"
os.environ["MINIO_SECRET_KEY"] = "secret_minio"
# create chain provider with env_aws and env_minio providers
chain = Chain(providers=[EnvAWS(), EnvMinio()])
# retireve provider (env_aws) has priority
creds = chain.retrieve()
# assert provider credentials
eq_(creds.access_key, "access_aws")
eq_(creds.secret_key, "secret_aws")
eq_(creds.session_token, "token_aws")
# assert is_expired
eq_(chain.is_expired(), False)