Get region from args or kwargs

This commit is contained in:
Bert Blommers 2020-06-27 19:46:26 +01:00
parent 6ab2f58b21
commit 389939bf8a

View File

@ -417,7 +417,8 @@ class ServerModeMockAWS(BaseMockAWS):
import mock
def fake_boto3_client(*args, **kwargs):
service, region = args
region = self._get_region(*args, **kwargs)
if region:
kwargs["config"] = Config(user_agent_extra="region/" + region)
if "endpoint_url" not in kwargs:
kwargs["endpoint_url"] = "http://localhost:5000"
@ -466,6 +467,14 @@ class ServerModeMockAWS(BaseMockAWS):
if six.PY2:
self._httplib_patcher.start()
def _get_region(self, *args, **kwargs):
if "region_name" in kwargs:
return kwargs["region_name"]
if type(args) == tuple:
service, region = args
return region
return None
def disable_patching(self):
if self._client_patcher:
self._client_patcher.stop()