@@ -44,8 +44,9 @@ def test_can_increment_view_anonymous_user(self, view_tracker_service):
4444 def test_can_increment_view_new_user (self , view_tracker_service , mock_session ):
4545 """First-time viewers should be able to increment views."""
4646 # Mock no existing record
47- mock_session .query .return_value .filter_by .return_value .one_or_none .\
48- return_value = None
47+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
48+ None
49+ )
4950
5051 result = view_tracker_service .can_increment_view ("owner/repo" , "user123" )
5152 assert result is True
@@ -57,8 +58,9 @@ def test_can_increment_view_within_cooldown(
5758 # Mock existing record from 1 hour ago (within 24-hour cooldown)
5859 mock_record = MagicMock (spec = RoadmapViewTracker )
5960 mock_record .viewed_at = datetime .now (timezone .utc ) - timedelta (hours = 1 )
60- mock_session .query .return_value .filter_by .return_value .one_or_none .\
61- return_value = mock_record
61+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
62+ mock_record
63+ )
6264
6365 result = view_tracker_service .can_increment_view ("owner/repo" , "user123" )
6466 assert result is False
@@ -70,8 +72,9 @@ def test_can_increment_view_after_cooldown(
7072 # Mock existing record from 25 hours ago (outside 24-hour cooldown)
7173 mock_record = MagicMock (spec = RoadmapViewTracker )
7274 mock_record .viewed_at = datetime .now (timezone .utc ) - timedelta (hours = 25 )
73- mock_session .query .return_value .filter_by .return_value .one_or_none .\
74- return_value = mock_record
75+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
76+ mock_record
77+ )
7578
7679 result = view_tracker_service .can_increment_view ("owner/repo" , "user123" )
7780 assert result is True
@@ -89,8 +92,9 @@ def test_record_view_anonymous_skips_tracking(
8992 def test_record_view_creates_new_record (self , view_tracker_service , mock_session ):
9093 """First view should create a new tracker record."""
9194 # Mock no existing record
92- mock_session .query .return_value .filter_by .return_value .one_or_none .\
93- return_value = None
95+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
96+ None
97+ )
9498
9599 view_tracker_service .record_view ("owner/repo" , "user123" )
96100
@@ -105,24 +109,24 @@ def test_record_view_updates_existing_record(
105109 # Mock existing record
106110 mock_record = MagicMock (spec = RoadmapViewTracker )
107111 mock_record .viewed_at = datetime .now (timezone .utc ) - timedelta (hours = 25 )
108- mock_session .query .return_value .filter_by .return_value .one_or_none .\
109- return_value = mock_record
112+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
113+ mock_record
114+ )
110115
111116 view_tracker_service .record_view ("owner/repo" , "user123" )
112117
113118 # Verify record was updated (viewed_at should be updated)
114- assert mock_record .viewed_at != datetime .now (timezone .utc ) - timedelta (
115- hours = 25
116- )
119+ assert mock_record .viewed_at != datetime .now (timezone .utc ) - timedelta (hours = 25 )
117120 mock_session .commit .assert_called ()
118121
119122 def test_increment_view_if_eligible_returns_true_when_eligible (
120123 self , view_tracker_service , mock_session
121124 ):
122125 """Should return True and record view when eligible (new user)."""
123126 # Mock no existing record (eligible) with with_for_update chain
124- mock_session .query .return_value .filter_by .return_value .\
125- with_for_update .return_value .one_or_none .return_value = None
127+ mock_session .query .return_value .filter_by .return_value .with_for_update .return_value .one_or_none .return_value = ( # noqa E501
128+ None
129+ )
126130
127131 result = view_tracker_service .increment_view_if_eligible (
128132 "owner/repo" , "user123"
@@ -139,8 +143,9 @@ def test_increment_view_if_eligible_returns_false_when_not_eligible(
139143 # Mock existing record from 1 hour ago (not eligible)
140144 mock_record = MagicMock (spec = RoadmapViewTracker )
141145 mock_record .viewed_at = datetime .now (timezone .utc ) - timedelta (hours = 1 )
142- mock_session .query .return_value .filter_by .return_value .\
143- with_for_update .return_value .one_or_none .return_value = mock_record
146+ mock_session .query .return_value .filter_by .return_value .with_for_update .return_value .one_or_none .return_value = ( # noqa E501
147+ mock_record
148+ )
144149
145150 result = view_tracker_service .increment_view_if_eligible (
146151 "owner/repo" , "user123"
@@ -153,8 +158,9 @@ def test_increment_view_if_eligible_handles_race_condition(
153158 ):
154159 """Should return False when IntegrityError occurs (race condition)."""
155160 # Mock no existing record initially
156- mock_session .query .return_value .filter_by .return_value .\
157- with_for_update .return_value .one_or_none .return_value = None
161+ mock_session .query .return_value .filter_by .return_value .with_for_update .return_value .one_or_none .return_value = ( # noqa E501
162+ None
163+ )
158164 # Mock IntegrityError on commit (another request created the record)
159165 mock_session .commit .side_effect = IntegrityError (
160166 "duplicate key" , params = None , orig = Exception ("duplicate" )
@@ -174,8 +180,9 @@ def test_increment_view_if_eligible_updates_after_cooldown(
174180 # Mock existing record from 25 hours ago (eligible after 24h cooldown)
175181 mock_record = MagicMock (spec = RoadmapViewTracker )
176182 mock_record .viewed_at = datetime .now (timezone .utc ) - timedelta (hours = 25 )
177- mock_session .query .return_value .filter_by .return_value .\
178- with_for_update .return_value .one_or_none .return_value = mock_record
183+ mock_session .query .return_value .filter_by .return_value .with_for_update .return_value .one_or_none .return_value = ( # noqa E501
184+ mock_record
185+ )
179186
180187 result = view_tracker_service .increment_view_if_eligible (
181188 "owner/repo" , "user123"
@@ -185,16 +192,14 @@ def test_increment_view_if_eligible_updates_after_cooldown(
185192 # Verify timestamp was updated
186193 assert mock_record .viewed_at > datetime .now (timezone .utc ) - timedelta (
187194 seconds = 5
188- )
195+ ) # noqa E501
189196 mock_session .commit .assert_called_once ()
190197
191198 def test_increment_view_if_eligible_anonymous_returns_true (
192199 self , view_tracker_service , mock_session
193200 ):
194201 """Should return True for anonymous users without database access."""
195- result = view_tracker_service .increment_view_if_eligible (
196- "owner/repo" , None
197- )
202+ result = view_tracker_service .increment_view_if_eligible ("owner/repo" , None )
198203
199204 assert result is True
200205 # Verify no database queries were made
@@ -204,17 +209,16 @@ def test_increment_view_if_eligible_anonymous_returns_true(
204209class TestRoadmapResultStoreIncrement :
205210 """Test suite for view count increment in RoadmapResultStore."""
206211
207- def test_increment_view_count_updates_counter (
208- self , result_store , mock_session
209- ):
212+ def test_increment_view_count_updates_counter (self , result_store , mock_session ):
210213 """Should increment view count for existing roadmap."""
211214 # Mock existing roadmap
212215 mock_roadmap = MagicMock (spec = GeneratedRoadmap )
213216 mock_roadmap .view_count = 5
214217 mock_roadmap .repo_summary = {"full_name" : "owner/repo" }
215218 mock_roadmap ._sa_instance_state = MagicMock () # Add SQLAlchemy state
216- mock_session .query .return_value .filter_by .return_value .one_or_none .\
217- return_value = mock_roadmap
219+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
220+ mock_roadmap
221+ )
218222
219223 result_store .increment_view_count ("owner/repo" )
220224
@@ -231,8 +235,9 @@ def test_increment_view_count_handles_none_view_count(
231235 mock_roadmap .view_count = None
232236 mock_roadmap .repo_summary = {"full_name" : "owner/repo" }
233237 mock_roadmap ._sa_instance_state = MagicMock () # Add SQLAlchemy state
234- mock_session .query .return_value .filter_by .return_value .one_or_none .\
235- return_value = mock_roadmap
238+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
239+ mock_roadmap
240+ )
236241
237242 result_store .increment_view_count ("owner/repo" )
238243
@@ -243,8 +248,9 @@ def test_increment_view_count_handles_none_view_count(
243248 def test_increment_view_count_nonexistent_roadmap (self , result_store , mock_session ):
244249 """Should handle gracefully when roadmap doesn't exist."""
245250 # Mock no existing roadmap
246- mock_session .query .return_value .filter_by .return_value .one_or_none .\
247- return_value = None
251+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
252+ None
253+ )
248254
249255 # Should not raise an error
250256 result_store .increment_view_count ("owner/nonexistent" )
@@ -496,8 +502,9 @@ def test_view_tracking_workflow_anonymous(
496502 mock_roadmap .view_count = 10
497503 mock_roadmap .repo_summary = {"full_name" : "owner/repo" , "view_count" : 10 }
498504 mock_roadmap ._sa_instance_state = MagicMock () # Add SQLAlchemy state
499- mock_session .query .return_value .filter_by .return_value .one_or_none .\
500- return_value = mock_roadmap
505+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
506+ mock_roadmap
507+ )
501508
502509 # 1. Check if view can be incremented (should be True for anonymous)
503510 can_increment = view_tracker_service .can_increment_view ("owner/repo" , None )
@@ -524,8 +531,9 @@ def test_view_tracking_workflow_authenticated_new_user(
524531 spec = GeneratedRoadmap , view_count = 10 , repo_summary = {"view_count" : 10 }
525532 ), # Third call: roadmap exists
526533 ]
527- mock_session .query .return_value .filter_by .return_value .one_or_none .\
528- side_effect = side_effect_values
534+ mock_session .query .return_value .filter_by .return_value .one_or_none .side_effect = ( # noqa E501
535+ side_effect_values
536+ )
529537
530538 # 1. Check if view can be incremented (should be True for new user)
531539 can_increment = view_tracker_service .can_increment_view ("owner/repo" , "user123" )
@@ -544,8 +552,9 @@ def test_view_tracking_workflow_authenticated_repeat_within_cooldown(
544552 # Setup: Existing view record from 1 hour ago
545553 mock_record = MagicMock (spec = RoadmapViewTracker )
546554 mock_record .viewed_at = datetime .now (timezone .utc ) - timedelta (hours = 1 )
547- mock_session .query .return_value .filter_by .return_value .one_or_none .\
548- return_value = mock_record
555+ mock_session .query .return_value .filter_by .return_value .one_or_none .return_value = ( # noqa E501
556+ mock_record
557+ )
549558
550559 # Check if view can be incremented (should be False within cooldown)
551560 can_increment = view_tracker_service .can_increment_view ("owner/repo" , "user123" )
0 commit comments