Skip to content

Commit 43f6df8

Browse files
committed
fix: lint
1 parent 8b45bde commit 43f6df8

File tree

1 file changed

+49
-40
lines changed

1 file changed

+49
-40
lines changed

commitly-backend/tests/test_roadmap_views.py

Lines changed: 49 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -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(
204209
class 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

Comments
 (0)