mirror of https://github.com/apache/superset.git
chore: convert feature flag tests to unit tests (#17689)
* chore: convert feature flag tests to unit tests * Fix test
This commit is contained in:
parent
aad2abe7f0
commit
b7ae7f976a
|
@ -1,83 +0,0 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing,
|
||||
# software distributed under the License is distributed on an
|
||||
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
# KIND, either express or implied. See the License for the
|
||||
# specific language governing permissions and limitations
|
||||
# under the License.
|
||||
from unittest.mock import patch
|
||||
|
||||
from parameterized import parameterized
|
||||
|
||||
from superset import get_feature_flags, is_feature_enabled
|
||||
from tests.integration_tests.base_tests import SupersetTestCase
|
||||
|
||||
|
||||
def dummy_is_feature_enabled(feature_flag_name: str, default: bool = True) -> bool:
|
||||
return True if feature_flag_name.startswith("True_") else default
|
||||
|
||||
|
||||
class TestFeatureFlag(SupersetTestCase):
|
||||
@patch.dict(
|
||||
"superset.extensions.feature_flag_manager._feature_flags",
|
||||
{"FOO": True},
|
||||
clear=True,
|
||||
)
|
||||
def test_existing_feature_flags(self):
|
||||
self.assertTrue(is_feature_enabled("FOO"))
|
||||
|
||||
@patch.dict(
|
||||
"superset.extensions.feature_flag_manager._feature_flags", {}, clear=True
|
||||
)
|
||||
def test_nonexistent_feature_flags(self):
|
||||
self.assertFalse(is_feature_enabled("FOO"))
|
||||
|
||||
def test_feature_flags(self):
|
||||
self.assertEqual(is_feature_enabled("foo"), "bar")
|
||||
self.assertEqual(is_feature_enabled("super"), "set")
|
||||
|
||||
|
||||
@patch.dict(
|
||||
"superset.extensions.feature_flag_manager._feature_flags",
|
||||
{"True_Flag1": False, "True_Flag2": True, "Flag3": False, "Flag4": True},
|
||||
clear=True,
|
||||
)
|
||||
class TestFeatureFlagBackend(SupersetTestCase):
|
||||
@parameterized.expand(
|
||||
[
|
||||
("True_Flag1", True),
|
||||
("True_Flag2", True),
|
||||
("Flag3", False),
|
||||
("Flag4", True),
|
||||
("True_DoesNotExist", False),
|
||||
]
|
||||
)
|
||||
@patch(
|
||||
"superset.extensions.feature_flag_manager._is_feature_enabled_func",
|
||||
dummy_is_feature_enabled,
|
||||
)
|
||||
def test_feature_flags_override(self, feature_flag_name, expected):
|
||||
self.assertEqual(is_feature_enabled(feature_flag_name), expected)
|
||||
|
||||
@patch(
|
||||
"superset.extensions.feature_flag_manager._is_feature_enabled_func",
|
||||
dummy_is_feature_enabled,
|
||||
)
|
||||
@patch(
|
||||
"superset.extensions.feature_flag_manager._get_feature_flags_func", None,
|
||||
)
|
||||
def test_get_feature_flags(self):
|
||||
feature_flags = get_feature_flags()
|
||||
self.assertEqual(
|
||||
feature_flags,
|
||||
{"True_Flag1": True, "True_Flag2": True, "Flag3": False, "Flag4": True},
|
||||
)
|
|
@ -0,0 +1,65 @@
|
|||
# Licensed to the Apache Software Foundation (ASF) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The ASF licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing,
|
||||
# software distributed under the License is distributed on an
|
||||
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
# KIND, either express or implied. See the License for the
|
||||
# specific language governing permissions and limitations
|
||||
# under the License.
|
||||
from pytest_mock import MockFixture
|
||||
|
||||
from superset import is_feature_enabled
|
||||
|
||||
|
||||
def dummy_is_feature_enabled(feature_flag_name: str, default: bool = True) -> bool:
|
||||
return True if feature_flag_name.startswith("True_") else default
|
||||
|
||||
|
||||
def test_existing_feature_flags(mocker: MockFixture) -> None:
|
||||
"""
|
||||
Test that ``is_feature_enabled`` reads flags correctly.
|
||||
"""
|
||||
mocker.patch.dict(
|
||||
"superset.extensions.feature_flag_manager._feature_flags",
|
||||
{"FOO": True},
|
||||
clear=True,
|
||||
)
|
||||
assert is_feature_enabled("FOO") is True
|
||||
|
||||
|
||||
def test_nonexistent_feature_flags(mocker: MockFixture) -> None:
|
||||
"""
|
||||
Test that ``is_feature_enabled`` returns ``False`` when flag not set.
|
||||
"""
|
||||
mocker.patch.dict(
|
||||
"superset.extensions.feature_flag_manager._feature_flags", {}, clear=True
|
||||
)
|
||||
assert is_feature_enabled("FOO") is False
|
||||
|
||||
|
||||
def test_is_feature_enabled(mocker: MockFixture) -> None:
|
||||
"""
|
||||
Test ``_is_feature_enabled_func``.
|
||||
"""
|
||||
mocker.patch.dict(
|
||||
"superset.extensions.feature_flag_manager._feature_flags",
|
||||
{"True_Flag1": False, "True_Flag2": True, "Flag3": False, "Flag4": True},
|
||||
clear=True,
|
||||
)
|
||||
mocker.patch(
|
||||
"superset.extensions.feature_flag_manager._is_feature_enabled_func",
|
||||
dummy_is_feature_enabled,
|
||||
)
|
||||
|
||||
assert is_feature_enabled("True_Flag1") is True
|
||||
assert is_feature_enabled("True_Flag2") is True
|
||||
assert is_feature_enabled("Flag3") is False
|
||||
assert is_feature_enabled("Flag4") is True
|
Loading…
Reference in New Issue