use helper everywhere for runtime data

Author: joshua-koehler

mixpanel/flags/test_local_feature_flags.py

@@ -196,30 +196,22 @@ async def test_get_variant_value_respects_runtime_evaluation_rule_satisfied(self
         }
         flag = create_test_flag(runtime_evaluation_rule=runtime_eval)
         await self.setup_flags([flag])
-        context = {
-            "distinct_id": DISTINCT_ID,
-            "custom_properties": {
-                "plan": "premium",
-                "region": "US"
-            }
-        }
+        context = self.user_context_with_properties({
+            "plan": "premium",
+        })
         result = self._flags.get_variant_value(TEST_FLAG_KEY, "fallback", context)
         assert result != "fallback"
 
     @respx.mock
     async def test_get_variant_value_respects_runtime_evaluation_rule_not_satisfied(self):
         runtime_eval = {
-            "==": [{"var": "plan"}, "basic"]
+            "==": [{"var": "plan"}, "premium"]
         }
         flag = create_test_flag(runtime_evaluation_rule=runtime_eval)
         await self.setup_flags([flag])
-        context = {
-            "distinct_id": DISTINCT_ID,
-            "custom_properties": {
-                "plan": "premium",
-                "region": "US"
-            }
-        }
+        context = self.user_context_with_properties({
+            "plan": "basic",
+        })
         result = self._flags.get_variant_value(TEST_FLAG_KEY, "fallback", context)
         assert result == "fallback"
 
@@ -244,13 +236,10 @@ async def test_get_variant_value_returns_fallback_when_runtime_evaluation_not_sa
         runtime_eval = {"plan": "premium", "region": "US"}
         flag = create_test_flag(runtime_evaluation_legacy_definition=runtime_eval)
         await self.setup_flags([flag])
-        context = {
-            "distinct_id": DISTINCT_ID,
-            "custom_properties": {
-                "plan": "basic",
-                "region": "US"
-            }
-        }
+        context = self.user_context_with_properties({
+            "plan": "basic",
+            "region": "US"
+        })
         result = self._flags.get_variant_value(TEST_FLAG_KEY, "fallback", context)
         assert result == "fallback"