pt
/
superset
mirror of https://github.com/apache/superset.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: add rls jinja extra cache key test (#10810)

This commit is contained in:
Ville Brofeldt 2020-09-08 07:42:48 +03:00 committed by GitHub
parent 1d76c5906e
commit 3ae80d3b98
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/security_tests.py
View File

@ -1020,7 +1020,7 @@ class TestRowLevelSecurity(SupersetTestCase):
.filter(SqlaTable.table_name.in_(["energy_usage", "unicode_test"]))
.all()
)
self.rls_entry.clause = "value > 1"
self.rls_entry.clause = "value > {{ cache_key_wrapper(1) }}"
self.rls_entry.roles.append(
security_manager.find_role("Gamma")
) # db.session.query(Role).filter_by(name="Gamma").first())
@ -1052,7 +1052,8 @@ class TestRowLevelSecurity(SupersetTestCase):
extras={},
)
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):
g.user = self.get_user(
@ -1071,7 +1072,8 @@ class TestRowLevelSecurity(SupersetTestCase):
extras={},
)
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):
g.user = self.get_user(
@ -1090,4 +1092,5 @@ class TestRowLevelSecurity(SupersetTestCase):
extras={},
)
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