def set_region_name(self, region_name=None):
'we need to set a region name if it is not set yet'
env = Env()
self.region_name = region_name
if self.region_name is None:
self.region_name = env.get_env('AWS_REGION')
def set_region(self, region_name=None):
'''we need to set a region if it is unset at this point'''
env = Env()
self.region_name = region_name
if region_name is None:
self.region_name = env.get_env('AWS_REGION')
return self.region_name
def _set_master_account(self, account_number=None):
'set the AWS Organization master account number'
try:
env = Env()
sts = Sts()
acc_num = sts.get_account_id()
self.master_account = acc_num or account_number or env.get_env(
'AW_MASTER')
except Exception as e:
raise Exception('Could not find master account info', e)
if self.master_account is None:
raise Exception('Master Account not set, check AW_MASTER')
def test_env_get():
os.environ['TEST_ENV'] = 'just a test'
env = Env()
assert env.get_env('TEST_ENV') == 'just a test'
assert env.get_env('ENV_TEST') is None
os.environ.pop('TEST_ENV')
def test_env_init():
os.environ['TEST_ENV'] = 'just a test'
env = Env()
assert env.env['TEST_ENV'] == 'just a test'
os.environ.pop('TEST_ENV')
def test_set_env_unset_error(env):
env = Env()
env.set_env('HOWDY', None)
with pytest.raises(KeyError):
os.environ['HOWDY']
def test_set_env_unset(env):
env = Env()
env.set_env('BLARGH', None)
with pytest.raises(KeyError):
os.environ['BLARGH']
def test_set_env(env):
env = Env()
env.set_env('BLARGH', 'go away!!!')
assert os.environ['BLARGH'] == 'go away!!!'
def test_get_env(env):
env = Env()
assert env.get_env('BLARGH') == 'hello!'
assert env.get_env('HGRALB') is None
def test_init(env):
env = Env()
assert type(env) == Env
assert env.env['BLARGH'] == 'hello!'