You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
"""
|
|
All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
its licensors.
|
|
|
|
For complete copyright and license terms please see the LICENSE at the root of this
|
|
distribution (the "License"). All use of this software is governed by the License,
|
|
or, if provided, by the license below or the license accompanying this file. Do not
|
|
remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
Pytest fixture for accessing an optional "--timeout" argument used for specifying
|
|
the timeout (in seconds) for any one AP Batch call.
|
|
If not specified, defaults to 35 minutes
|
|
"""
|
|
|
|
import pytest
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def timeout_option_fixture(request) -> int:
|
|
"""
|
|
Fixture for accessing the command line argument "--ap_timeout"
|
|
Defaults to 35 minutes
|
|
"""
|
|
default_timeout = 35 * 60 # 35 minutes (2100 seconds)
|
|
timeout = request.config.getoption("--ap_timeout")
|
|
if timeout is not None:
|
|
try:
|
|
return int(timeout)
|
|
except ValueError:
|
|
# failed casting timeout to integer
|
|
# fmt:off
|
|
logger.error(f"Failed parsing --ap_timeout cmdline argument {timeout} to integer. "
|
|
f"Using default value of {default_timeout} seconds")
|
|
# fmt:on
|
|
return default_timeout
|