fix: set correct schema on config import (#16041)

* fix: set correct schema on config import

* Fix lint

* Fix test

* Fix tests

* Fix another test

* Fix another test

* Fix base test

* Add helper function

* Fix examples

* Fix test

* Fix test

* Fixing more tests
This commit is contained in:
Beto Dealmeida
2021-11-04 11:09:08 -07:00
committed by GitHub
parent 03a2c6ee8a
commit 1fbce88a46
30 changed files with 309 additions and 116 deletions

View File

@@ -38,7 +38,7 @@ from superset.exceptions import SupersetSecurityException
from superset.models.core import Database
from superset.models.slice import Slice
from superset.sql_parse import Table
from superset.utils.core import get_example_database
from superset.utils.core import get_example_database, get_example_default_schema
from superset.views.access_requests import AccessRequestsModelView
from .base_tests import SupersetTestCase
@@ -104,13 +104,14 @@ class TestRolePermission(SupersetTestCase):
"""Testing export role permissions."""
def setUp(self):
schema = get_example_default_schema()
session = db.session
security_manager.add_role(SCHEMA_ACCESS_ROLE)
session.commit()
ds = (
db.session.query(SqlaTable)
.filter_by(table_name="wb_health_population")
.filter_by(table_name="wb_health_population", schema=schema)
.first()
)
ds.schema = "temp_schema"
@@ -133,11 +134,11 @@ class TestRolePermission(SupersetTestCase):
session = db.session
ds = (
session.query(SqlaTable)
.filter_by(table_name="wb_health_population")
.filter_by(table_name="wb_health_population", schema="temp_schema")
.first()
)
schema_perm = ds.schema_perm
ds.schema = None
ds.schema = get_example_default_schema()
ds.schema_perm = None
ds_slices = (
session.query(Slice)