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
|
from unittest import TestCase
from flaky.flaky_decorator import flaky
from flaky.names import FlakyNames
class TestFlakyDecorator(TestCase):
def test_flaky_raises_for_non_positive_min_passes(self):
def test_something():
pass
self.assertRaises(
ValueError,
lambda: flaky(min_passes=0)(test_something),
)
def test_flaky_raises_for_max_runs_less_than_min_passes(self):
def test_something():
pass
self.assertRaises(
ValueError,
lambda: flaky(max_runs=2, min_passes=3)(test_something),
)
def test_flaky_adds_flaky_attributes_to_test_method(self):
min_passes = 4
max_runs = 7
@flaky(max_runs, min_passes)
def test_something():
pass
flaky_attribute = {
attr: getattr(
test_something,
attr,
None
) for attr in FlakyNames()
}
self.assertIsNotNone(flaky_attribute)
self.assertLessEqual(
{
FlakyNames.MIN_PASSES: min_passes,
FlakyNames.MAX_RUNS: max_runs,
FlakyNames.CURRENT_PASSES: 0,
FlakyNames.CURRENT_RUNS: 0,
FlakyNames.CURRENT_ERRORS: None
}.items(),
flaky_attribute.items()
)
|
