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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# Authors:
# Petr Viktorin <pviktori@redhat.com>
#
# Copyright (C) 2014 Red Hat
# see file 'COPYING' for use and warranty information
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""Pytest plugin for IPA
Adds support for the @pytest.mark.source_order decorator which,
when applied to a class, runs the test methods in source order.
See test_ordering for an example.
"""
import unittest
import pytest
def ordered(cls):
"""Decorator that marks a test class as ordered
Methods within the marked class will be executed in definition order
(or more strictly, in ordered by the line number where they're defined).
Subclasses of unittest.TestCase can not be ordered.
Generator methods will not be ordered by this plugin.
"""
cls._order_plugin__ordered = True
assert not isinstance(cls, unittest.TestCase), (
"A unittest.TestCase may not be ordered.")
cls = pytest.mark.source_order(cls)
return cls
def decorate_items(items):
node_indexes = {}
for index, item in enumerate(items):
try:
func = item.function
except AttributeError:
yield (index, ), item
continue
key = (index, )
for node in reversed(item.listchain()):
# Find the corresponding class
if isinstance(node, pytest.Class):
cls = node.cls
else:
continue
if getattr(cls, '_order_plugin__ordered', False):
node_index = node_indexes.setdefault(node, index)
# Find first occurence of the method in class hierarchy
for i, parent_class in enumerate(reversed(cls.mro())):
if getattr(parent_class, '_order_plugin__ordered', False):
method = getattr(parent_class, func.__name__, None)
if method:
# Sort methods as tuples (position of the class
# in the inheritance chain, position of the method
# within that class)
key = (node_index, 0,
i, method.func_code.co_firstlineno, node)
break
else:
# Weird case fallback
# Method name not in any of the classes in MRO, run it last
key = node_index, 1, func.func_code.co_firstlineno, node
break
yield key, item
def pytest_collection_modifyitems(session, config, items):
items[:] = [item for i, item in sorted(decorate_items(items))]
|
