Problème de decorateur entre unittest et nosetests

Version imprimable

Bonjour,

j'ai un petit problème qui mérite un oeil externe :roll:

J'ai une classe de test avec des méthodes de test.
Chaque méthode de test possède un décorateur timeout

La definition du décorateur se fait en déhors de la classe de test.

Quand je lance unittest, aucun soucis, les tests se lance et j'ai bien le timeout qui fonctionne ... par contre avec nosetest plus rien ne se lance :(
(et bien sûr en enlevant les décorateurs tout fonctionne bien :-/ )

Si vous avez une idée je suis preneur

Voici le bout de code

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
import unittest
import signal
import time
 
class TimeoutException(Exception):
    pass
 
def timeout(timeout_time):
    """
    decorator function catching the argument
    """
    def timeout_function(func):
        """
        decorator function
        """
        def _timeout_function(self):
            """
            create a signal handler
            set the timeout with the argument given while calling the decorator @timeout
            call the function
            catch a timeout exception if necessary
            """
            def timeout_handler(signum, frame):
                print 'Timeout (%s sec) reached' % str(timeout_time)
                raise TimeoutException()
 
            old_handler = signal.signal(signal.SIGALRM, timeout_handler)
            signal.alarm(timeout_time) # triger alarm in timeout_time seconds
            try:
                retval = func(self)
            except TimeoutException:
                return 'Timeout (%s sec) reached' % str(timeout_time)
            finally:
                signal.signal(signal.SIGALRM, old_handler)
            signal.alarm(0)
            return retval
        return _timeout_function
    return timeout_function
 
 
class Test_MaclasseDeTest(unittest.TestCase):
    @timeout(5)
    def test_OK(self):
        print 'dummy test ok'
        #content of the test
        pass
 
    @timeout(5)
    def test_KO(self):
        print 'dummy test ko'
        time.sleep(10)
        #content of the test
        pass
 
 
if __name__ == "__main__":
    unittest.main()

23/02/2011, 18h01
wiztricks
Salut,

Lire la discussion qui décrit les précautions à prendre avec les décorateurs.

Si vous utilisez nose, la chose existe déjà (enfin c'est la doc qui le dit):
Citation:
nose.tools.timed(limit)
Test must finish within specified time limit to pass.
Example use:

Code:

1 2 3 @timed(.1) def test_that_fails(): time.sleep(.2)
- W
23/02/2011, 20h10
Makiavelik

Merci pour la réponse, je vais regarder la discussion.

Pour le décorateur @timed, en fait c'est juste un check sur le temps que le test met à s'exécuter ... du coup s'il y a une attente infinie, on ne sort pas 'proprement'

C'est résolu grâce au lien de la discussion

voici le code qui fonctionne à la fois sous unittest et sous nosetests

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import unittest
import signal
import time
from functools import wraps
 
class TimeoutException(Exception):
    pass
 
def timeout(timeout_time):
    """
    decorator function catching the argument
    """
    def timeout_function(func):
        """
        decorator function
        """
        @wraps(func)
        def _timeout_function(self):
            """
            create a signal handler
            set the timeout with the argument given while calling the decorator @timeout
            call the function
            catch a timeout exception if necessary
            """
            def timeout_handler(signum, frame):
                print 'Timeout (%s sec) reached' % str(timeout_time)
                raise TimeoutException()
 
            old_handler = signal.signal(signal.SIGALRM, timeout_handler)
            signal.alarm(timeout_time) # triger alarm in timeout_time seconds
            try:
                retval = func(self)
            except TimeoutException:
		self.fail()
                return 'Timeout (%s sec) reached' % str(timeout_time)
            finally:
                signal.signal(signal.SIGALRM, old_handler)
            signal.alarm(0)
            return retval
        return _timeout_function
    return timeout_function
 
 
class Test_MaclasseDeTest(unittest.TestCase):
    @timeout(5)
    def test_OK(self):
	"""docString test_OK"""
        pass
 
    @timeout(5)
    def test_KO(self):
        """docString test_KO"""
        time.sleep(10)
        pass
 
 
if __name__ == "__main__":
    unittest.main()