Skip to content

Commit bd2b171

Browse files
refactor(test): normalize upload state defaults in commit factory
Normalize `_upload_state_ids` to two values up front so CommitWithReportFactory always seeds both upload state IDs without conditional branches. Made-with: Cursor
1 parent f52f950 commit bd2b171

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

libs/shared/shared/django_apps/core/tests/factories.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,9 @@ class Meta:
6767
class CommitWithReportFactory(CommitFactory):
6868
@classmethod
6969
def _create(cls, model_class, *args, **kwargs):
70-
upload_state_ids = kwargs.pop("_upload_state_ids", None)
70+
default_state_id = UploadState.PROCESSED.db_id
71+
upload_state_ids = list(kwargs.pop("_upload_state_ids", []))
72+
upload_state_ids = (upload_state_ids + [default_state_id, default_state_id])[:2]
7173
commit = super()._create(
7274
model_class,
7375
_report={
@@ -143,9 +145,7 @@ def _create(cls, model_class, *args, **kwargs):
143145
report=commit_report,
144146
order_number=0,
145147
storage_path="v4/raw/2019-01-10/4434BC2A2EC4FCA57F77B473D83F928C/abf6d4df662c47e32460020ab14abf9303581429/9ccc55a1-8b41-4bb1-a946-ee7a33a7fb56.txt",
146-
state_id=(
147-
upload_state_ids[0] if upload_state_ids else UploadState.PROCESSED.db_id
148-
),
148+
state_id=upload_state_ids[0],
149149
)
150150
UploadLevelTotalsFactory(
151151
report_session=upload1,
@@ -167,11 +167,7 @@ def _create(cls, model_class, *args, **kwargs):
167167
report=commit_report,
168168
order_number=1,
169169
storage_path="v4/raw/2019-01-10/4434BC2A2EC4FCA57F77B473D83F928C/abf6d4df662c47e32460020ab14abf9303581429/9ccc55a1-8b41-4bb1-a946-ee7a33a7fb56.txt",
170-
state_id=(
171-
upload_state_ids[1]
172-
if upload_state_ids and len(upload_state_ids) > 1
173-
else UploadState.PROCESSED.db_id
174-
),
170+
state_id=upload_state_ids[1],
175171
)
176172
UploadLevelTotalsFactory(
177173
report_session=upload2,

0 commit comments

Comments
 (0)