Skip to content

Commit 9ac4295

Browse files
committed
improve assert style
1 parent 1e7e0e8 commit 9ac4295

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

tests/unit/assessment/test_sequencing.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,14 @@ def test_cluster_from_task_has_children(ws, simple_dependency_resolver, mock_pat
2525
sequencer = MigrationSequencer(ws, AdministratorLocator(ws, finders=[lambda _ws: admin_finder]))
2626
sequencer.register_workflow_task(task, job, graph)
2727
steps = list(sequencer.generate_steps())
28-
step = steps[-1]
29-
assert step.step_id
30-
assert step.object_type == "CLUSTER"
31-
assert step.object_id == "cluster-123"
32-
assert step.object_name == "my-cluster"
33-
assert step.object_owner == "John Doe"
34-
assert step.step_number == 3
35-
assert len(step.required_step_ids) == 2
28+
step = dataclasses.replace(steps[-1], step_id=0)
29+
# we don't know the exact ids of the required steps, se let's zero them
30+
step.required_step_ids = [0 for id in step.required_step_ids]
31+
assert step == MigrationStep(
32+
step_id=0,
33+
step_number = 3,
34+
object_type = "CLUSTER",
35+
object_id = "cluster-123",
36+
object_name = "my-cluster",
37+
object_owner = "John Doe",
38+
required_step_ids = [0, 0])

0 commit comments

Comments
 (0)