Skip to content

Commit 5741869

Browse files
committed
Update test
1 parent 0869763 commit 5741869

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

api/tests/unit/environments/test_unit_environments_models.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1255,13 +1255,17 @@ def test_environment_clone_from_non_versioned_environment_with_use_v2_feature_ve
12551255
enable_v2_versioning_for_new_environments: typing.Callable[[], None],
12561256
) -> None:
12571257
# Given
1258+
assert not environment.use_v2_feature_versioning
12581259
enable_v2_versioning_for_new_environments()
12591260

12601261
# When
12611262
new_environment = environment.clone(name="new-environment")
12621263

12631264
# Then
1264-
assert EnvironmentFeatureVersion.objects.filter(
1265-
environment=new_environment, feature=feature
1266-
).exists()
12671265
assert new_environment.use_v2_feature_versioning
1266+
1267+
latest_feature_states = get_environment_flags_queryset(new_environment)
1268+
assert latest_feature_states.count() == 1
1269+
1270+
latest_feature_state = latest_feature_states.get()
1271+
assert latest_feature_state.environment_feature_version is not None

0 commit comments

Comments
 (0)