fix: added-none-check-in-context-matches-condition

Author: Zaimwa9

flag_engine/segments/evaluator.py

@@ -7,7 +7,10 @@
 import semver
 
 from flag_engine.context.types import EvaluationContext
-from flag_engine.identities.traits.types import ContextValue
+from flag_engine.identities.traits.types import (
+    ContextValue,
+    map_any_value_to_trait_value,
+)
 from flag_engine.segments import constants
 from flag_engine.segments.models import (
     SegmentConditionModel,
@@ -100,7 +103,11 @@ def context_matches_condition(
     if condition.operator == constants.IS_SET:
         return context_value is not None
 
-    return _matches_context_value(condition, context_value) if context_value else False
+    return (
+        _matches_context_value(condition, context_value)
+        if context_value is not None
+        else False
+    )
 
 
 def _get_trait(context: EvaluationContext, trait_key: str) -> ContextValue:

tests/unit/segments/test_segments_evaluator.py

@@ -6,7 +6,11 @@
 
 from flag_engine.context.types import EvaluationContext
 from flag_engine.segments import constants
-from flag_engine.segments.evaluator import _matches_context_value, is_context_in_segment
+from flag_engine.segments.evaluator import (
+    _matches_context_value,
+    context_matches_condition,
+    is_context_in_segment,
+)
 from flag_engine.segments.models import (
     SegmentConditionModel,
     SegmentModel,
@@ -479,6 +483,63 @@ def test_segment_condition_matches_context_value_for_semver(
     assert result == expected_result
 
 
+@pytest.mark.parametrize(
+    "context,condition,segment_key,expected_result",
+    (
+        (
+            {"identity": {"traits": {trait_key_1: False}}},
+            SegmentConditionModel(
+                operator=constants.EQUAL,
+                property_=trait_key_1,
+                value="false",
+            ),
+            "segment_key",
+            True,
+        ),
+        (
+            {"identity": {"traits": {trait_key_1: True}}},
+            SegmentConditionModel(
+                operator=constants.EQUAL,
+                property_=trait_key_1,
+                value="true",
+            ),
+            "segment_key",
+            True,
+        ),
+        (
+            {"identity": {"traits": {trait_key_1: 12}}},
+            SegmentConditionModel(
+                operator=constants.EQUAL,
+                property_=trait_key_1,
+                value="12",
+            ),
+            "segment_key",
+            True,
+        ),
+        (
+            {"identity": {"traits": {trait_key_1: None}}},
+            SegmentConditionModel(
+                operator=constants.IS_SET,
+                property_=trait_key_1,
+                value="false",
+            ),
+            "segment_key",
+            False,
+        ),
+    ),
+)
+def test_context_matches_condition_evaluates_with_correct_casting(
+    context: EvaluationContext,
+    condition: SegmentConditionModel,
+    segment_key: str,
+    expected_result: bool,
+) -> None:
+    # Given / When
+    result = context_matches_condition(context, condition, segment_key)
+    # Then
+    assert result == expected_result
+
+
 @pytest.mark.parametrize(
     "trait_value, condition_value, expected_result",
     [