QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-18 23:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: proper current_app.config proxy usage (#34345)

Browse Source 
Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
Maxime Beauchemin
2025-07-31 19:27:42 -07:00
committed by GitHub
parent 6c9cda758a
commit cb27d5fe8d
144 changed files with 1428 additions and 1119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/integration_tests/security/row_level_security_tests.py
View File

@@ -23,7 +23,7 @@ import pytest
from flask import g
import prison
from superset import db, security_manager, app # noqa: F401
from superset import db, security_manager
from superset.connectors.sqla.models import RowLevelSecurityFilter, SqlaTable
from superset.security.guest_token import (
GuestTokenResourceType,
@@ -625,15 +625,16 @@ class TestRowLevelSecurityWithRelatedAPI(SupersetTestCase):
def _base_filter(query):
return query.filter_by(name="Alpha")
with mock.patch.dict(
"superset.views.filters.current_app.config",
{"EXTRA_RELATED_QUERY_FILTERS": {"role": _base_filter}},
):
original_conf = self.app.config.get("EXTRA_RELATED_QUERY_FILTERS", {}).copy()
try:
self.app.config["EXTRA_RELATED_QUERY_FILTERS"] = {"role": _base_filter}
rv = self.client.get("/api/v1/rowlevelsecurity/related/roles") # noqa: F541
assert rv.status_code == 200
response = json.loads(rv.data.decode("utf-8"))
response_roles = [result["text"] for result in response["result"]]
assert response_roles == ["Alpha"]
finally:
self.app.config["EXTRA_RELATED_QUERY_FILTERS"] = original_conf
RLS_ALICE_REGEX = re.compile(r"name = 'Alice'")