Skip to content

Commit 2356f36

Browse files
committed
add test
1 parent aaeb7f1 commit 2356f36

File tree

2 files changed

+32
-2
lines changed

2 files changed

+32
-2
lines changed

tests/unit/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,10 +157,10 @@ def identity_in_segment(
157157
@pytest.fixture
158158
def context_in_segment(
159159
identity_in_segment: IdentityModel,
160-
environment: EnvironmentModel,
160+
environment_with_segment_override: EnvironmentModel,
161161
) -> EvaluationContext:
162162
return map_environment_identity_to_context(
163-
environment=environment,
163+
environment=environment_with_segment_override,
164164
identity=identity_in_segment,
165165
override_traits=None,
166166
)

tests/unit/segments/test_segments_evaluator.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from flag_engine.segments.evaluator import (
2020
_matches_context_value,
2121
context_matches_condition,
22+
get_evaluation_result,
2223
get_flag_result_from_feature_context,
2324
get_identity_segments,
2425
is_context_in_segment,
@@ -637,6 +638,35 @@ def test_segment_condition_matches_context_value_for_modulo(
637638
assert result == expected_result
638639

639640

641+
def test_get_evaluation_result__returns_expected(
642+
context_in_segment: EvaluationContext,
643+
) -> None:
644+
# When
645+
result = get_evaluation_result(context_in_segment)
646+
647+
# Then
648+
assert result == {
649+
"context": context_in_segment,
650+
"flags": [
651+
{
652+
"enabled": False,
653+
"feature_key": "1",
654+
"name": "feature_1",
655+
"reason": "TARGETING_MATCH; segment=my_segment",
656+
"value": "segment_override",
657+
},
658+
{
659+
"enabled": False,
660+
"feature_key": "2",
661+
"name": "feature_2",
662+
"reason": "DEFAULT",
663+
"value": None,
664+
},
665+
],
666+
"segments": [{"key": "1", "name": "my_segment"}],
667+
}
668+
669+
640670
@pytest.mark.parametrize(
641671
"percentage_value, expected_result",
642672
(

0 commit comments

Comments
 (0)