mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-04 03:52:56 +00:00
5ec82105cc
## Problem Follow up of #12400 ## Summary of changes We didn't set remote_size_mb to Some when initialized so it never gets computed :( Also added a new API to force refresh the properties. Signed-off-by: Alex Chi Z <chi@neon.tech>
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from fixtures.utils import run_only_on_default_postgres
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import NeonEnvBuilder
|
|
|
|
|
|
@run_only_on_default_postgres("Pageserver-only test only needs to run on one version")
|
|
def test_feature_flag(neon_env_builder: NeonEnvBuilder):
|
|
env = neon_env_builder.init_start()
|
|
env.pageserver.http_client().force_override_feature_flag("test-feature-flag", "true")
|
|
assert env.pageserver.http_client().evaluate_feature_flag_boolean(
|
|
env.initial_tenant, "test-feature-flag"
|
|
)["result"]["Ok"]
|
|
assert (
|
|
env.pageserver.http_client().evaluate_feature_flag_multivariate(
|
|
env.initial_tenant, "test-feature-flag"
|
|
)["result"]["Ok"]
|
|
== "true"
|
|
)
|
|
|
|
env.pageserver.http_client().force_override_feature_flag("test-feature-flag", "false")
|
|
assert (
|
|
env.pageserver.http_client().evaluate_feature_flag_boolean(
|
|
env.initial_tenant, "test-feature-flag"
|
|
)["result"]["Err"]
|
|
== "No condition group is matched"
|
|
)
|
|
assert (
|
|
env.pageserver.http_client().evaluate_feature_flag_multivariate(
|
|
env.initial_tenant, "test-feature-flag"
|
|
)["result"]["Ok"]
|
|
== "false"
|
|
)
|
|
|
|
env.pageserver.http_client().force_override_feature_flag("test-feature-flag", None)
|
|
assert (
|
|
"Err"
|
|
in env.pageserver.http_client().evaluate_feature_flag_boolean(
|
|
env.initial_tenant, "test-feature-flag"
|
|
)["result"]
|
|
)
|
|
assert (
|
|
"Err"
|
|
in env.pageserver.http_client().evaluate_feature_flag_multivariate(
|
|
env.initial_tenant, "test-feature-flag"
|
|
)["result"]
|
|
)
|
|
|
|
env.pageserver.http_client().force_refresh_feature_flag(env.initial_tenant)
|
|
|
|
# Check if the properties exist
|
|
result = env.pageserver.http_client().evaluate_feature_flag_multivariate(
|
|
env.initial_tenant, "test-feature-flag"
|
|
)
|
|
assert "tenant_remote_size_mb" in result["properties"]
|
|
assert "tenant_id" in result["properties"]
|