def test_with_aws_session_credentials(gdalenv):
"""Create an Env with a boto3 session."""
env = rasterio.Env(
aws_access_key_id='id', aws_secret_access_key='key',
aws_session_token='token', region_name='null-island-1')
expected = default_options.copy()
with env:
expected.update({
'AWS_ACCESS_KEY_ID': 'id', 'AWS_REGION': 'null-island-1',
'AWS_SECRET_ACCESS_KEY': 'key', 'AWS_SESSION_TOKEN': 'token'})
assert getenv() == expected
def test_with_aws_session_credentials(gdalenv):
"""Create an Env with a boto3 session."""
with rasterio.Env(
aws_access_key_id='id', aws_secret_access_key='key',
aws_session_token='token', region_name='null-island-1') as s:
expected = default_options.copy()
assert getenv() == rasterio.env._env.options == expected
s.get_aws_credentials()
expected.update({
'aws_access_key_id': 'id', 'aws_region': 'null-island-1',
'aws_secret_access_key': 'key', 'aws_session_token': 'token'})
assert getenv() == rasterio.env._env.options == expected
def test_env_options(gdalenv):
"""Test env options."""
env = rasterio.Env(foo='x')
assert env.options == {'foo': 'x'}
assert not env.previous_options
expected = default_options.copy()
assert getenv() == rasterio.env._env.options == expected
expected.update({'foo': 'x'})
with env:
assert getenv() == rasterio.env._env.options == expected
del expected['foo']
assert getenv() == rasterio.env._env.options == expected
def test_with_aws_session_credentials(gdalenv):
"""Create an Env with a boto3 session."""
with rasterio.Env(
aws_access_key_id='id', aws_secret_access_key='key',
aws_session_token='token', region_name='null-island-1') as s:
expected = default_options.copy()
assert getenv() == rasterio.env.local._env.options == expected
s.get_aws_credentials()
expected.update({
'aws_access_key_id': 'id', 'aws_region': 'null-island-1',
'aws_secret_access_key': 'key', 'aws_session_token': 'token'})
assert getenv() == rasterio.env.local._env.options == expected
def test_env_accessors(gdalenv):
"""High level GDAL env access."""
defenv()
setenv(foo='1', bar='2')
expected = default_options.copy()
expected.update({'foo': '1', 'bar': '2'})
assert getenv() == rasterio.env._env.options
assert getenv() == expected
assert get_gdal_config('foo') == '1'
assert get_gdal_config('bar') == '2'
delenv()
with pytest.raises(EnvError):
getenv()
assert get_gdal_config('foo') is None
assert get_gdal_config('bar') is None
def test_with_aws_session_credentials(gdalenv):
"""Create an Env with a boto3 session."""
env = rasterio.Env(aws_access_key_id='id',
aws_secret_access_key='key',
aws_session_token='token',
region_name='null-island-1')
expected = default_options.copy()
with env:
expected.update({
'AWS_ACCESS_KEY_ID': 'id',
'AWS_REGION': 'null-island-1',
'AWS_SECRET_ACCESS_KEY': 'key',
'AWS_SESSION_TOKEN': 'token'
})
assert getenv() == expected
def test_env_accessors(gdalenv):
"""High level GDAL env access."""
defenv()
setenv(foo='1', bar='2')
expected = default_options.copy()
expected.update({'foo': '1', 'bar': '2'})
assert getenv() == rasterio.env.local._env.options
assert getenv() == expected
assert get_gdal_config('foo') == 1
assert get_gdal_config('bar') == 2
delenv()
with pytest.raises(EnvError):
getenv()
assert get_gdal_config('foo') is None
assert get_gdal_config('bar') is None
def test_env_accessors(gdalenv):
"""High level GDAL env access."""
defenv()
setenv(foo='1', bar='2')
expected = default_options.copy()
expected.update({'foo': '1', 'bar': '2'})
items = getenv()
assert items == rasterio.env.local._env.options
# Comparison below requires removal of GDAL_DATA.
items.pop('GDAL_DATA', None)
assert items == expected
assert get_gdal_config('foo') == 1
assert get_gdal_config('bar') == 2
delenv()
with pytest.raises(EnvError):
getenv()
assert get_gdal_config('foo') is None
assert get_gdal_config('bar') is None
def test_with_aws_session_credentials(gdalenv):
"""Create an Env with a boto3 session."""
env = rasterio.Env(aws_access_key_id='id',
aws_secret_access_key='key',
aws_session_token='token',
region_name='null-island-1')
expected = default_options.copy()
with env:
expected.update({
'AWS_ACCESS_KEY_ID': 'id',
'AWS_REGION': 'null-island-1',
'AWS_SECRET_ACCESS_KEY': 'key',
'AWS_SESSION_TOKEN': 'token'
})
items = getenv()
# Comparison below requires removal of GDAL_DATA.
items.pop('GDAL_DATA', None)
assert items == expected
