[gentoo-commits] repo/gentoo:master commit in: eclass/

[Michał Górny]

commit:     635892e8f096afd7d1f3ea2f91a6906122b2e5ca
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Wed Jan  3 13:32:33 2024 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Fri Jan  5 20:18:53 2024 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=635892e8

python-utils-r1.eclass, distutils-r1.eclass: Add EPYTEST_TIMEOUT

Signed-off-by: Michał Górny <mgorny <AT> gentoo.org>

 eclass/distutils-r1.eclass    |  3 +++
 eclass/python-utils-r1.eclass | 22 ++++++++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/eclass/distutils-r1.eclass b/eclass/distutils-r1.eclass
index 428551662845..c0d1992ccce0 100644
--- a/eclass/distutils-r1.eclass
+++ b/eclass/distutils-r1.eclass
@@ -618,6 +618,9 @@ distutils_enable_tests() {
                        ;;
                pytest)
                        test_pkgs='>=dev-python/pytest-7.3.1[${PYTHON_USEDEP}]'
+                       if [[ -n ${EPYTEST_TIMEOUT} ]]; then
+                               test_pkgs+=' 
dev-python/pytest-timeout[${PYTHON_USEDEP}]'
+                       fi
                        if [[ ${EPYTEST_XDIST} ]]; then
                                test_pkgs+=' 
dev-python/pytest-xdist[${PYTHON_USEDEP}]'
                        fi

diff --git a/eclass/python-utils-r1.eclass b/eclass/python-utils-r1.eclass
index a82379ce876b..3af3cbdb075e 100644
--- a/eclass/python-utils-r1.eclass
+++ b/eclass/python-utils-r1.eclass
@@ -1296,6 +1296,16 @@ _python_check_occluded_packages() {
 # parameter, when calling epytest.  The listed files will be entirely
 # skipped from test collection.
 
+# @ECLASS_VARIABLE: EPYTEST_TIMEOUT
+# @DEFAULT_UNSET
+# @DESCRIPTION:
+# If set to a non-empty value, enables pytest-timeout plugin and sets
+# test timeout to the specified value.  This variable can be either set
+# in ebuilds that are known to hang, or by user to prevent hangs
+# in automated test environments.  If this variable is set prior
+# to calling distutils_enable_tests in distutils-r1, a test dependency
+# on dev-python/pytest-timeout is added automatically.
+
 # @ECLASS_VARIABLE: EPYTEST_XDIST
 # @DEFAULT_UNSET
 # @DESCRIPTION:
@@ -1381,6 +1391,18 @@ epytest() {
                )
        fi
 
+       if [[ -n ${EPYTEST_TIMEOUT} ]]; then
+               if [[ ${PYTEST_PLUGINS} != *pytest_timeout* ]]; then
+                       args+=(
+                               -p timeout
+                       )
+               fi
+
+               args+=(
+                       "--timeout=${EPYTEST_TIMEOUT}"
+               )
+       fi
+
        if [[ ${EPYTEST_XDIST} ]]; then
                local jobs=${EPYTEST_JOBS:-$(makeopts_jobs)}
                if [[ ${jobs} -gt 1 ]]; then