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

chore(dao): Organize DAOs according to SIP-92 (#24331)

Browse Source 
Co-authored-by: JUST.in DO IT <justin.park@airbnb.com>
This commit is contained in:
John Bodley
2023-06-18 18:32:32 -07:00
committed by GitHub
parent 1bc7d63cab
commit 3e76736874
149 changed files with 426 additions and 554 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unit_tests/dao/queries_test.py
View File

@@ -51,7 +51,7 @@ def test_query_dao_save_metadata(session: Session) -> None:
session.add(db)
session.add(query_obj)
from superset.queries.dao import QueryDAO
from superset.daos.query import QueryDAO
query = session.query(Query).one()
QueryDAO.save_metadata(query=query, payload={"columns": []})
@@ -105,7 +105,7 @@ def test_query_dao_get_queries_changed_after(session: Session) -> None:
session.add(old_query_obj)
session.add(updated_query_obj)
from superset.queries.dao import QueryDAO
from superset.daos.query import QueryDAO
timestamp = datetime.timestamp(now - timedelta(days=2)) * 1000
result = QueryDAO.get_queries_changed_after(timestamp)
@@ -146,7 +146,7 @@ def test_query_dao_stop_query_not_found(
mocker.patch("superset.sql_lab.cancel_query", return_value=False)
from superset.queries.dao import QueryDAO
from superset.daos.query import QueryDAO
with pytest.raises(QueryNotFoundException):
QueryDAO.stop_query("foo2")
@@ -186,7 +186,7 @@ def test_query_dao_stop_query_not_running(
session.add(db)
session.add(query_obj)
from superset.queries.dao import QueryDAO
from superset.daos.query import QueryDAO
QueryDAO.stop_query(query_obj.client_id)
query = session.query(Query).one()
@@ -226,7 +226,7 @@ def test_query_dao_stop_query_failed(
mocker.patch("superset.sql_lab.cancel_query", return_value=False)
from superset.queries.dao import QueryDAO
from superset.daos.query import QueryDAO
with pytest.raises(SupersetCancelQueryException):
QueryDAO.stop_query(query_obj.client_id)
@@ -266,7 +266,7 @@ def test_query_dao_stop_query(mocker: MockFixture, app: Any, session: Session) -
mocker.patch("superset.sql_lab.cancel_query", return_value=True)
from superset.queries.dao import QueryDAO
from superset.daos.query import QueryDAO
QueryDAO.stop_query(query_obj.client_id)
query = session.query(Query).one()