Skip to content

Commit ce41c16

Browse files
committed
fix/unit-tests: Remove invalid flag type check in private method _create_provider_evaluation
Signed-off-by: Andrew Helsby <[email protected]> Signed-off-by: Andrew Helsby <[email protected]>
1 parent d12011c commit ce41c16

File tree

2 files changed

+3
-20
lines changed

2 files changed

+3
-20
lines changed

open_feature/open_feature_client.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from numbers import Number
44

55
from open_feature.evaluation_context.evaluation_context import EvaluationContext
6-
from open_feature.exception.exceptions import GeneralError, OpenFeatureError
6+
from open_feature.exception.exceptions import OpenFeatureError
77
from open_feature.flag_evaluation.error_code import ErrorCode
88
from open_feature.flag_evaluation.flag_evaluation_details import FlagEvaluationDetails
99
from open_feature.flag_evaluation.flag_type import FlagType
@@ -274,9 +274,6 @@ def _create_provider_evaluation(
274274
FlagType.NUMBER: self.provider.get_number_details,
275275
FlagType.OBJECT: self.provider.get_object_details,
276276
FlagType.STRING: self.provider.get_string_details,
277-
}.get(flag_type)
278-
279-
if not get_details_callable:
280-
raise GeneralError(error_message="Unknown flag type")
277+
}[flag_type]
281278

282279
return get_details_callable(*args)

tests/test_open_feature_client.py

+1-15
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,10 @@
33

44
import pytest
55

6-
from open_feature.exception.exceptions import GeneralError, OpenFeatureError
6+
from open_feature.exception.exceptions import OpenFeatureError
77
from open_feature.flag_evaluation.error_code import ErrorCode
88
from open_feature.flag_evaluation.reason import Reason
99
from open_feature.hooks.hook import Hook
10-
from open_feature.open_feature_client import OpenFeatureClient
1110

1211

1312
@pytest.mark.parametrize(
@@ -72,19 +71,6 @@ def test_should_get_flag_detail_based_on_method_type(
7271
assert isinstance(flag.value, flag_type)
7372

7473

75-
def test_should_raise_exception_when_invalid_flag_type_provided():
76-
# Given
77-
# When
78-
with pytest.raises(GeneralError) as ge:
79-
OpenFeatureClient("No provider", "1.0")._create_provider_evaluation(
80-
flag_type=None, flag_key="Key", default_value=True
81-
)
82-
# Then
83-
assert ge.value
84-
assert ge.value.error_message == "Unknown flag type"
85-
assert ge.value.error_code == ErrorCode.GENERAL
86-
87-
8874
def test_should_handle_a_generic_exception_thrown_by_a_provider(no_op_provider_client):
8975
# Given
9076
exception_hook = MagicMock(spec=Hook)

0 commit comments

Comments
 (0)