def test_get_all_sqlatables_datasources(app_context: None,
session_with_data: Session) -> None:
from superset.dao.datasource.dao import DatasourceDAO
result = DatasourceDAO.get_all_sqlatables_datasources(
session=session_with_data)
assert len(result) == 1
def test_get_datasource_query(app_context: None,
session_with_data: Session) -> None:
from superset.dao.datasource.dao import DatasourceDAO
from superset.models.sql_lab import Query
result = DatasourceDAO.get_datasource(datasource_type=DatasourceType.QUERY,
datasource_id=1,
session=session_with_data)
assert result.id == 1
assert isinstance(result, Query)
def test_get_datasource_sl_table(app_context: None,
session_with_data: Session) -> None:
from superset.dao.datasource.dao import DatasourceDAO
from superset.tables.models import Table
# todo(hugh): This will break once we remove the dual write
# update the datsource_id=1 and this will pass again
result = DatasourceDAO.get_datasource(datasource_type=DatasourceType.TABLE,
datasource_id=2,
session=session_with_data)
assert result.id == 2
assert isinstance(result, Table)
def test_get_datasource_sqlatable(app_context: None,
session_with_data: Session) -> None:
from superset.connectors.sqla.models import SqlaTable
from superset.dao.datasource.dao import DatasourceDAO
result = DatasourceDAO.get_datasource(
datasource_type=DatasourceType.SQLATABLE,
datasource_id=1,
session=session_with_data,
)
assert 1 == result.id
assert "my_sqla_table" == result.table_name
assert isinstance(result, SqlaTable)