def stop_infra():
global INFRA_STOPPED
if INFRA_STOPPED:
return
generic_proxy.QUIET = True
common.cleanup(files=True, quiet=True)
common.cleanup_resources()
lambda_api.cleanup()
time.sleep(1)
# TODO: optimize this (takes too long currently)
# check_infra(retries=2, expect_shutdown=True)
INFRA_STOPPED = True
def start_dynamodbstreams(port=DEFAULT_PORT_DYNAMODBSTREAMS, async=False):
print("Starting mock DynamoDB Streams (port %s)..." % port)
if async:
thread = FuncThread(dynamodbstreams_api.serve, port, quiet=True)
thread.start()
TMP_THREADS.append(thread)
return thread
else:
firehose_api.serve(port)
def start_lambda(port=DEFAULT_PORT_LAMBDA, async=False):
print("Starting mock Lambda (port %s)..." % port)
lambda_api.cleanup()
if async:
thread = FuncThread(lambda_api.serve, port, quiet=True)
thread.start()
TMP_THREADS.append(thread)
return thread
else:
lambda_api.serve(port)
def stop_infra():
global INFRA_STOPPED
if INFRA_STOPPED:
return
generic_proxy.QUIET = True
common.cleanup(files=True, quiet=True)
def stop_infra():
generic_proxy.QUIET = True
common.cleanup(files=True, quiet=True)
common.cleanup_resources()
lambda_api.cleanup()
time.sleep(1)