Currently the avocado-classless plugin always uses a fixed (short) timeout
for tests. However the appropriate timeout really depends on what's being
tested.
Allow tests to set different timeouts, by placing not a plain function, but
an object representing metadata into the classless test manifest. Then
tests can use a parameterized version of the decorator to put additional
information in that metadata. For now, just add the timeout there, but
we might add other things in future.
Signed-off-by: David Gibson
---
.../avocado_classless/manifest.py | 11 ++++++++---
test/avocado_classless/avocado_classless/plugin.py | 4 +---
test/avocado_classless/avocado_classless/test.py | 14 ++++++++++----
test/avocado_classless/examples.py | 8 +++++++-
4 files changed, 26 insertions(+), 11 deletions(-)
diff --git a/test/avocado_classless/avocado_classless/manifest.py b/test/avocado_classless/avocado_classless/manifest.py
index 8f7fe688..34c815cd 100644
--- a/test/avocado_classless/avocado_classless/manifest.py
+++ b/test/avocado_classless/avocado_classless/manifest.py
@@ -12,6 +12,7 @@ Manage the manifest of classless tests in a module
import sys
MANIFEST = "__avocado_classless__"
+DEFAULT_TIMEOUT = 60.0
def manifest(mod):
@@ -24,14 +25,18 @@ def manifest(mod):
class ManifestTestInfo: # pylint: disable=R0903
"""Metadata about a classless test"""
- def __init__(self, func):
+ def __init__(self, func, timeout=DEFAULT_TIMEOUT):
self.func = func
+ self.__timeout = float(timeout)
def run_test(self):
self.func()
+ def timeout(self):
+ return self.__timeout
-def manifest_add(mod, name, func):
+
+def manifest_add(mod, name, func, **kwargs):
"""Register a function as a classless test"""
if isinstance(mod, str):
@@ -40,4 +45,4 @@ def manifest_add(mod, name, func):
mfest = manifest(mod)
if name in mfest:
raise ValueError(f"Duplicate classless test name {name}")
- mfest[name] = ManifestTestInfo(func)
+ mfest[name] = ManifestTestInfo(func, **kwargs)
diff --git a/test/avocado_classless/avocado_classless/plugin.py b/test/avocado_classless/avocado_classless/plugin.py
index 442642d5..e2c9b1eb 100644
--- a/test/avocado_classless/avocado_classless/plugin.py
+++ b/test/avocado_classless/avocado_classless/plugin.py
@@ -38,7 +38,6 @@ from avocado.core.utils import messages
from .manifest import manifest
SHBANG = b'#! /usr/bin/env avocado-runner-avocado-classless'
-DEFAULT_TIMEOUT = 5.0
def load_mod(path):
@@ -184,7 +183,6 @@ class ClasslessRunner(BaseRunner):
process.start()
time_started = time.monotonic()
- timeout = DEFAULT_TIMEOUT
next_status_time = None
while True:
time.sleep(RUNNER_RUN_CHECK_INTERVAL)
@@ -196,7 +194,7 @@ class ClasslessRunner(BaseRunner):
):
next_status_time = now + RUNNER_RUN_STATUS_INTERVAL
yield messages.RunningMessage.get()
- if (now - time_started) > timeout:
+ if (now - time_started) > testinfo.timeout():
process.terminate()
yield messages.FinishedMessage.get("interrupted",
"timeout")
diff --git a/test/avocado_classless/avocado_classless/test.py b/test/avocado_classless/avocado_classless/test.py
index e5251960..04fa52ab 100644
--- a/test/avocado_classless/avocado_classless/test.py
+++ b/test/avocado_classless/avocado_classless/test.py
@@ -14,10 +14,16 @@ from collections import Counter
from .manifest import manifest_add
-def test(func):
- """Function decorator to mark a function as a classless test"""
- manifest_add(func.__module__, func.__name__, func)
- return func
+def classless_test(**kwargs):
+ """Mark a function as a classless test with options"""
+ def dec(func):
+ manifest_add(func.__module__, func.__name__, func, **kwargs)
+ return func
+ return dec
+
+
+# Simple version of classless_test with default options
+test = classless_test()
def test_output(*tests):
diff --git a/test/avocado_classless/examples.py b/test/avocado_classless/examples.py
index a0e883f3..8625c9d3 100644
--- a/test/avocado_classless/examples.py
+++ b/test/avocado_classless/examples.py
@@ -7,8 +7,9 @@ are expected to fail.
"""
import sys
+import time
-from avocado_classless.test import test, test_output
+from avocado_classless.test import classless_test, test, test_output
@test
@@ -49,3 +50,8 @@ def positive_fraction():
@test_output(is_integer, is_positive)
def negative_fraction():
return -3.5
+
+
+@classless_test(timeout=1.0)
+def timeout():
+ time.sleep(3)
--
2.41.0