mirror of
https://github.com/apache/superset.git
synced 2026-04-16 22:55:52 +00:00
committed by
GitHub
parent
dea9068647
commit
a9fb853e3e
@@ -1959,13 +1959,11 @@ class TestSecurityManager(SupersetTestCase):
|
||||
|
||||
class TestDatasources(SupersetTestCase):
|
||||
@patch("superset.security.SupersetSecurityManager.can_access_database")
|
||||
@patch("superset.security.SupersetSecurityManager.get_session")
|
||||
def test_get_user_datasources_admin(
|
||||
self, mock_get_session, mock_can_access_database
|
||||
):
|
||||
@patch("superset.security.SupersetSecurityManager.session")
|
||||
def test_get_user_datasources_admin(self, mock_session, mock_can_access_database):
|
||||
Datasource = namedtuple("Datasource", ["database", "schema", "name"])
|
||||
mock_can_access_database.return_value = True
|
||||
mock_get_session.query.return_value.filter.return_value.all.return_value = []
|
||||
mock_session.query.return_value.filter.return_value.all.return_value = []
|
||||
|
||||
with mock.patch.object(
|
||||
SqlaTable, "get_all_datasources"
|
||||
@@ -1984,13 +1982,11 @@ class TestDatasources(SupersetTestCase):
|
||||
]
|
||||
|
||||
@patch("superset.security.SupersetSecurityManager.can_access_database")
|
||||
@patch("superset.security.SupersetSecurityManager.get_session")
|
||||
def test_get_user_datasources_gamma(
|
||||
self, mock_get_session, mock_can_access_database
|
||||
):
|
||||
@patch("superset.security.SupersetSecurityManager.session")
|
||||
def test_get_user_datasources_gamma(self, mock_session, mock_can_access_database):
|
||||
Datasource = namedtuple("Datasource", ["database", "schema", "name"])
|
||||
mock_can_access_database.return_value = False
|
||||
mock_get_session.query.return_value.filter.return_value.all.return_value = []
|
||||
mock_session.query.return_value.filter.return_value.all.return_value = []
|
||||
|
||||
with mock.patch.object(
|
||||
SqlaTable, "get_all_datasources"
|
||||
@@ -2005,14 +2001,14 @@ class TestDatasources(SupersetTestCase):
|
||||
assert datasources == []
|
||||
|
||||
@patch("superset.security.SupersetSecurityManager.can_access_database")
|
||||
@patch("superset.security.SupersetSecurityManager.get_session")
|
||||
@patch("superset.security.SupersetSecurityManager.session")
|
||||
def test_get_user_datasources_gamma_with_schema(
|
||||
self, mock_get_session, mock_can_access_database
|
||||
self, mock_session, mock_can_access_database
|
||||
):
|
||||
Datasource = namedtuple("Datasource", ["database", "schema", "name"])
|
||||
mock_can_access_database.return_value = False
|
||||
|
||||
mock_get_session.query.return_value.filter.return_value.all.return_value = [
|
||||
mock_session.query.return_value.filter.return_value.all.return_value = [
|
||||
Datasource("database1", "schema1", "table1"),
|
||||
Datasource("database1", "schema1", "table2"),
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user