44
55from posthog .client import Client
66from posthog .test .test_utils import FAKE_TEST_API_KEY
7- from posthog .types import FeatureFlag , FeatureFlagResult , FlagMetadata , FlagReason
7+ from posthog .types import (
8+ FeatureFlag ,
9+ FeatureFlagError ,
10+ FeatureFlagResult ,
11+ FlagMetadata ,
12+ FlagReason ,
13+ )
814
915
1016class TestFeatureFlagResult (unittest .TestCase ):
@@ -450,7 +456,7 @@ def test_get_feature_flag_result_unknown_flag(self, patch_capture, patch_flags):
450456 "$feature_flag_response" : None ,
451457 "locally_evaluated" : False ,
452458 "$feature/no-person-flag" : None ,
453- "$feature_flag_error" : "flag_missing" ,
459+ "$feature_flag_error" : FeatureFlagError . FLAG_MISSING ,
454460 },
455461 groups = {},
456462 disable_geoip = None ,
@@ -496,7 +502,7 @@ def test_get_feature_flag_result_with_errors_while_computing_flags(
496502 "$feature_flag_reason" : "Matched condition set 1" ,
497503 "$feature_flag_id" : 1 ,
498504 "$feature_flag_version" : 1 ,
499- "$feature_flag_error" : "errors_while_computing_flags" ,
505+ "$feature_flag_error" : FeatureFlagError . ERRORS_WHILE_COMPUTING ,
500506 },
501507 groups = {},
502508 disable_geoip = None ,
@@ -538,7 +544,7 @@ def test_get_feature_flag_result_flag_not_in_response(
538544 "locally_evaluated" : False ,
539545 "$feature/missing-flag" : None ,
540546 "$feature_flag_request_id" : "test-request-id-456" ,
541- "$feature_flag_error" : "flag_missing" ,
547+ "$feature_flag_error" : FeatureFlagError . FLAG_MISSING ,
542548 },
543549 groups = {},
544550 disable_geoip = None ,
@@ -574,7 +580,7 @@ def test_get_feature_flag_result_errors_computing_and_flag_missing(
574580 "locally_evaluated" : False ,
575581 "$feature/missing-flag" : None ,
576582 "$feature_flag_request_id" : "test-request-id-999" ,
577- "$feature_flag_error" : "errors_while_computing_flags,flag_missing " ,
583+ "$feature_flag_error" : f" { FeatureFlagError . ERRORS_WHILE_COMPUTING } , { FeatureFlagError . FLAG_MISSING } " ,
578584 },
579585 groups = {},
580586 disable_geoip = None ,
@@ -597,7 +603,7 @@ def test_get_feature_flag_result_unknown_error(self, patch_capture, patch_flags)
597603 "$feature_flag_response" : None ,
598604 "locally_evaluated" : False ,
599605 "$feature/my-flag" : None ,
600- "$feature_flag_error" : "unknown_error" ,
606+ "$feature_flag_error" : FeatureFlagError . UNKNOWN_ERROR ,
601607 },
602608 groups = {},
603609 disable_geoip = None ,
@@ -622,7 +628,7 @@ def test_get_feature_flag_result_timeout_error(self, patch_capture, patch_flags)
622628 "$feature_flag_response" : None ,
623629 "locally_evaluated" : False ,
624630 "$feature/my-flag" : None ,
625- "$feature_flag_error" : "timeout" ,
631+ "$feature_flag_error" : FeatureFlagError . TIMEOUT ,
626632 },
627633 groups = {},
628634 disable_geoip = None ,
@@ -647,7 +653,7 @@ def test_get_feature_flag_result_connection_error(self, patch_capture, patch_fla
647653 "$feature_flag_response" : None ,
648654 "locally_evaluated" : False ,
649655 "$feature/my-flag" : None ,
650- "$feature_flag_error" : "connection_error" ,
656+ "$feature_flag_error" : FeatureFlagError . CONNECTION_ERROR ,
651657 },
652658 groups = {},
653659 disable_geoip = None ,
@@ -672,7 +678,7 @@ def test_get_feature_flag_result_api_error(self, patch_capture, patch_flags):
672678 "$feature_flag_response" : None ,
673679 "locally_evaluated" : False ,
674680 "$feature/my-flag" : None ,
675- "$feature_flag_error" : "api_error_500" ,
681+ "$feature_flag_error" : FeatureFlagError . api_error ( 500 ) ,
676682 },
677683 groups = {},
678684 disable_geoip = None ,
@@ -697,7 +703,7 @@ def test_get_feature_flag_result_quota_limited(self, patch_capture, patch_flags)
697703 "$feature_flag_response" : None ,
698704 "locally_evaluated" : False ,
699705 "$feature/my-flag" : None ,
700- "$feature_flag_error" : "quota_limited" ,
706+ "$feature_flag_error" : FeatureFlagError . QUOTA_LIMITED ,
701707 },
702708 groups = {},
703709 disable_geoip = None ,
@@ -769,7 +775,7 @@ def test_timeout_error_returns_stale_cached_value(self, patch_capture, patch_fla
769775 "locally_evaluated" : False ,
770776 "$feature/my-flag" : "cached-variant" ,
771777 "$feature_flag_payload" : {"from" : "cache" },
772- "$feature_flag_error" : "timeout" ,
778+ "$feature_flag_error" : FeatureFlagError . TIMEOUT ,
773779 },
774780 groups = {},
775781 disable_geoip = None ,
@@ -806,7 +812,7 @@ def test_connection_error_returns_stale_cached_value(
806812 "$feature_flag_response" : True ,
807813 "locally_evaluated" : False ,
808814 "$feature/my-flag" : True ,
809- "$feature_flag_error" : "connection_error" ,
815+ "$feature_flag_error" : FeatureFlagError . CONNECTION_ERROR ,
810816 },
811817 groups = {},
812818 disable_geoip = None ,
@@ -842,7 +848,7 @@ def test_api_error_returns_stale_cached_value(self, patch_capture, patch_flags):
842848 "$feature_flag_response" : "control" ,
843849 "locally_evaluated" : False ,
844850 "$feature/my-flag" : "control" ,
845- "$feature_flag_error" : "api_error_503" ,
851+ "$feature_flag_error" : FeatureFlagError . api_error ( 503 ) ,
846852 },
847853 groups = {},
848854 disable_geoip = None ,
@@ -872,7 +878,7 @@ def test_error_without_cache_returns_none(self, patch_capture, patch_flags):
872878 "$feature_flag_response" : None ,
873879 "locally_evaluated" : False ,
874880 "$feature/my-flag" : None ,
875- "$feature_flag_error" : "timeout" ,
881+ "$feature_flag_error" : FeatureFlagError . TIMEOUT ,
876882 },
877883 groups = {},
878884 disable_geoip = None ,
0 commit comments