Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ServerMode - dont bother resetting multiple times in parallel (#5264)

Browse Source
This commit is contained in:
Bert Blommers 2022-06-26 13:05:08 +00:00 committed by GitHub
parent a92dcd149d
commit e5830181dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
moto/core/models.py
View File

@ -333,16 +333,23 @@ MockAWS = BotocoreEventMockAWS
class ServerModeMockAWS(BaseMockAWS): class ServerModeMockAWS(BaseMockAWS):
RESET_IN_PROGRESS = False
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self.test_server_mode_endpoint = settings.test_server_mode_endpoint() self.test_server_mode_endpoint = settings.test_server_mode_endpoint()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def reset(self): def reset(self):
call_reset_api = os.environ.get("MOTO_CALL_RESET_API") call_reset_api = os.environ.get("MOTO_CALL_RESET_API")
if not call_reset_api or call_reset_api.lower() != "false": call_reset_api = not call_reset_api or call_reset_api.lower() != "false"
if call_reset_api:
if not ServerModeMockAWS.RESET_IN_PROGRESS:
ServerModeMockAWS.RESET_IN_PROGRESS = True
import requests import requests
requests.post(f"{self.test_server_mode_endpoint}/moto-api/reset") requests.post(f"{self.test_server_mode_endpoint}/moto-api/reset")
ServerModeMockAWS.RESET_IN_PROGRESS = False
def enable_patching(self, reset=True): def enable_patching(self, reset=True):
if self.__class__.nested_count == 1 and reset: if self.__class__.nested_count == 1 and reset: