@@ -25,8 +25,14 @@ def test_time_in_draft_with_ready_for_review(self):
2525 Test measure_time_in_draft with one draft and review interval.
2626 """
2727 self .issue .events .return_value = [
28- MagicMock (event = "converted_to_draft" , created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc )),
29- MagicMock (event = "ready_for_review" , created_at = datetime (2021 , 1 , 3 , tzinfo = pytz .utc )),
28+ MagicMock (
29+ event = "converted_to_draft" ,
30+ created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc ),
31+ ),
32+ MagicMock (
33+ event = "ready_for_review" ,
34+ created_at = datetime (2021 , 1 , 3 , tzinfo = pytz .utc ),
35+ ),
3036 ]
3137 result = measure_time_in_draft (self .issue )
3238 expected = timedelta (days = 2 )
@@ -37,7 +43,10 @@ def test_time_in_draft_without_ready_for_review(self):
3743 Test measure_time_in_draft when ready_for_review_at is not provided and issue is still open.
3844 """
3945 self .issue .events .return_value = [
40- MagicMock (event = "converted_to_draft" , created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc )),
46+ MagicMock (
47+ event = "converted_to_draft" ,
48+ created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc ),
49+ ),
4150 ]
4251 now = datetime (2021 , 1 , 4 , tzinfo = pytz .utc )
4352 with unittest .mock .patch ("time_in_draft.datetime" ) as mock_datetime :
@@ -51,10 +60,22 @@ def test_time_in_draft_multiple_intervals(self):
5160 Test measure_time_in_draft with multiple draft intervals.
5261 """
5362 self .issue .events .return_value = [
54- MagicMock (event = "converted_to_draft" , created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc )),
55- MagicMock (event = "ready_for_review" , created_at = datetime (2021 , 1 , 3 , tzinfo = pytz .utc )),
56- MagicMock (event = "converted_to_draft" , created_at = datetime (2021 , 1 , 5 , tzinfo = pytz .utc )),
57- MagicMock (event = "ready_for_review" , created_at = datetime (2021 , 1 , 7 , tzinfo = pytz .utc )),
63+ MagicMock (
64+ event = "converted_to_draft" ,
65+ created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc ),
66+ ),
67+ MagicMock (
68+ event = "ready_for_review" ,
69+ created_at = datetime (2021 , 1 , 3 , tzinfo = pytz .utc ),
70+ ),
71+ MagicMock (
72+ event = "converted_to_draft" ,
73+ created_at = datetime (2021 , 1 , 5 , tzinfo = pytz .utc ),
74+ ),
75+ MagicMock (
76+ event = "ready_for_review" ,
77+ created_at = datetime (2021 , 1 , 7 , tzinfo = pytz .utc ),
78+ ),
5879 ]
5980 result = measure_time_in_draft (self .issue )
6081 expected = timedelta (days = 4 )
@@ -65,32 +86,45 @@ def test_time_in_draft_ongoing_draft(self):
6586 Test measure_time_in_draft with an ongoing draft interval.
6687 """
6788 self .issue .events .return_value = [
68- MagicMock (event = "converted_to_draft" , created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc )),
89+ MagicMock (
90+ event = "converted_to_draft" ,
91+ created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc ),
92+ ),
6993 ]
7094 with unittest .mock .patch ("time_in_draft.datetime" ) as mock_datetime :
7195 mock_datetime .now .return_value = datetime (2021 , 1 , 4 , tzinfo = pytz .utc )
7296 result = measure_time_in_draft (self .issue )
7397 expected = timedelta (days = 3 )
74- self .assertEqual (result , expected , "The ongoing draft time should be 3 days." )
98+ self .assertEqual (
99+ result , expected , "The ongoing draft time should be 3 days."
100+ )
75101
76102 def test_time_in_draft_no_draft_events (self ):
77103 """
78104 Test measure_time_in_draft with no draft-related events.
79105 """
80106 self .issue .events .return_value = []
81107 result = measure_time_in_draft (self .issue )
82- self .assertIsNone (result , "The result should be None when there are no draft events." )
108+ self .assertIsNone (
109+ result , "The result should be None when there are no draft events."
110+ )
83111
84112 def test_time_in_draft_without_ready_for_review_and_closed (self ):
85113 """
86114 Test measure_time_in_draft for a closed issue with an ongoing draft and ready_for_review_at is not provided.
87115 """
88116 self .issue .events .return_value = [
89- MagicMock (event = "converted_to_draft" , created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc )),
117+ MagicMock (
118+ event = "converted_to_draft" ,
119+ created_at = datetime (2021 , 1 , 1 , tzinfo = pytz .utc ),
120+ ),
90121 ]
91122 self .issue .issue .state = "closed"
92123 result = measure_time_in_draft (self .issue )
93- self .assertIsNone (result , "The result should be None for a closed issue with an ongoing draft." )
124+ self .assertIsNone (
125+ result ,
126+ "The result should be None for a closed issue with an ongoing draft." ,
127+ )
94128
95129
96130class TestGetStatsTimeInDraft (unittest .TestCase ):
0 commit comments