mirror of https://github.com/apache/superset.git
test: add rls jinja extra cache key test (#10810)
This commit is contained in:
parent
1d76c5906e
commit
3ae80d3b98
|
@ -1020,7 +1020,7 @@ class TestRowLevelSecurity(SupersetTestCase):
|
||||||
.filter(SqlaTable.table_name.in_(["energy_usage", "unicode_test"]))
|
.filter(SqlaTable.table_name.in_(["energy_usage", "unicode_test"]))
|
||||||
.all()
|
.all()
|
||||||
)
|
)
|
||||||
self.rls_entry.clause = "value > 1"
|
self.rls_entry.clause = "value > {{ cache_key_wrapper(1) }}"
|
||||||
self.rls_entry.roles.append(
|
self.rls_entry.roles.append(
|
||||||
security_manager.find_role("Gamma")
|
security_manager.find_role("Gamma")
|
||||||
) # db.session.query(Role).filter_by(name="Gamma").first())
|
) # db.session.query(Role).filter_by(name="Gamma").first())
|
||||||
|
@ -1052,7 +1052,8 @@ class TestRowLevelSecurity(SupersetTestCase):
|
||||||
extras={},
|
extras={},
|
||||||
)
|
)
|
||||||
sql = tbl.get_query_str(query_obj)
|
sql = tbl.get_query_str(query_obj)
|
||||||
self.assertIn("value > 1", sql)
|
assert tbl.get_extra_cache_keys(query_obj) == [1]
|
||||||
|
assert "value > 1" in sql
|
||||||
|
|
||||||
def test_rls_filter_doesnt_alter_query(self):
|
def test_rls_filter_doesnt_alter_query(self):
|
||||||
g.user = self.get_user(
|
g.user = self.get_user(
|
||||||
|
@ -1071,7 +1072,8 @@ class TestRowLevelSecurity(SupersetTestCase):
|
||||||
extras={},
|
extras={},
|
||||||
)
|
)
|
||||||
sql = tbl.get_query_str(query_obj)
|
sql = tbl.get_query_str(query_obj)
|
||||||
self.assertNotIn("value > 1", sql)
|
assert tbl.get_extra_cache_keys(query_obj) == []
|
||||||
|
assert "value > 1" not in sql
|
||||||
|
|
||||||
def test_multiple_table_filter_alters_another_tables_query(self):
|
def test_multiple_table_filter_alters_another_tables_query(self):
|
||||||
g.user = self.get_user(
|
g.user = self.get_user(
|
||||||
|
@ -1090,4 +1092,5 @@ class TestRowLevelSecurity(SupersetTestCase):
|
||||||
extras={},
|
extras={},
|
||||||
)
|
)
|
||||||
sql = tbl.get_query_str(query_obj)
|
sql = tbl.get_query_str(query_obj)
|
||||||
self.assertIn("value > 1", sql)
|
assert tbl.get_extra_cache_keys(query_obj) == [1]
|
||||||
|
assert "value > 1" in sql
|
||||||
|
|
Loading…
Reference in New Issue