@@ -16,7 +16,7 @@ class TestLabels(unittest.TestCase):
16
16
def setUp (self ):
17
17
self .issue = MagicMock () # type: ignore
18
18
self .issue .issue = MagicMock (spec = github3 .issues .Issue ) # type: ignore
19
- self .issue .created_at = "2020 -01-01T00:00:00Z"
19
+ self .issue .created_at = "2021 -01-01T00:00:00Z"
20
20
self .issue .closed_at = "2021-01-05T00:00:00Z"
21
21
self .issue .state = "closed"
22
22
self .issue .issue .events .return_value = [
@@ -35,29 +35,42 @@ def setUp(self):
35
35
label = {"name" : "bug" },
36
36
created_at = datetime (2021 , 1 , 3 , tzinfo = pytz .UTC ),
37
37
),
38
+ MagicMock (
39
+ event = "labeled" ,
40
+ label = {"name" : "bug" },
41
+ created_at = datetime (2021 , 1 , 4 , tzinfo = pytz .UTC ),
42
+ ),
38
43
]
39
44
40
45
def test_get_label_events (self ):
41
46
"""Test get_label_events"""
42
47
labels = ["bug" ]
43
48
events = get_label_events (self .issue , labels )
44
- self .assertEqual (len (events ), 2 )
49
+ self .assertEqual (len (events ), 3 )
45
50
self .assertEqual (events [0 ].label ["name" ], "bug" )
46
51
self .assertEqual (events [1 ].label ["name" ], "bug" )
52
+ self .assertEqual (events [2 ].label ["name" ], "bug" )
47
53
48
54
def test_get_label_metrics_closed_issue (self ):
49
55
"""Test get_label_metrics using a closed issue"""
50
56
labels = ["bug" , "feature" ]
51
57
metrics = get_label_metrics (self .issue , labels )
52
- self .assertEqual (metrics ["bug" ], timedelta (days = 2 ))
58
+ self .assertEqual (metrics ["bug" ], timedelta (days = 3 ))
53
59
self .assertEqual (metrics ["feature" ], timedelta (days = 3 ))
54
60
55
61
def test_get_label_metrics_open_issue (self ):
56
62
"""Test get_label_metrics using an open issue"""
57
63
self .issue .state = "open"
58
64
labels = ["bug" , "feature" ]
59
65
metrics = get_label_metrics (self .issue , labels )
60
- self .assertEqual (metrics ["bug" ], timedelta (days = 2 ))
66
+ self .assertLessEqual (
67
+ metrics ["bug" ],
68
+ datetime .now (pytz .utc ) - datetime (2021 , 1 , 2 , tzinfo = pytz .UTC ),
69
+ )
70
+ self .assertGreater (
71
+ metrics ["bug" ],
72
+ datetime .now (pytz .utc ) - datetime (2021 , 1 , 3 , tzinfo = pytz .UTC ),
73
+ )
61
74
self .assertLessEqual (
62
75
metrics ["feature" ],
63
76
datetime .now (pytz .utc ) - datetime (2021 , 1 , 2 , tzinfo = pytz .UTC ),
0 commit comments