mirror of
https://github.com/apache/superset.git
synced 2026-04-19 16:14:52 +00:00
chore: remove sqlparse (#33564)
This commit is contained in:
@@ -30,7 +30,7 @@ from sqlalchemy.dialects import sqlite
|
||||
from sqlalchemy.engine.url import make_url, URL
|
||||
from sqlalchemy.sql import sqltypes
|
||||
|
||||
from superset.sql_parse import Table
|
||||
from superset.sql.parse import Table
|
||||
from superset.superset_typing import ResultSetColumnType, SQLAColumnType
|
||||
from superset.utils.core import GenericDataType
|
||||
from tests.unit_tests.db_engine_specs.utils import assert_column_spec
|
||||
|
||||
@@ -27,7 +27,7 @@ from sqlalchemy.engine.url import make_url
|
||||
from sqlalchemy.sql import sqltypes
|
||||
from sqlalchemy_bigquery import BigQueryDialect
|
||||
|
||||
from superset.sql_parse import Table
|
||||
from superset.sql.parse import Table
|
||||
from superset.superset_typing import ResultSetColumnType
|
||||
from superset.utils import json
|
||||
from tests.unit_tests.db_engine_specs.utils import assert_convert_dttm
|
||||
|
||||
@@ -18,7 +18,7 @@
|
||||
import pytest # noqa: F401
|
||||
from pytest_mock import MockerFixture
|
||||
|
||||
from superset.sql_parse import Table
|
||||
from superset.sql.parse import Table
|
||||
|
||||
|
||||
def test_epoch_to_dttm() -> None:
|
||||
|
||||
@@ -16,7 +16,6 @@
|
||||
# under the License.
|
||||
from datetime import datetime
|
||||
from typing import Any, Optional
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
import pytest
|
||||
from sqlalchemy import column # noqa: F401
|
||||
@@ -93,21 +92,3 @@ def test_opendistro_sqla_column_label(original: str, expected: str) -> None:
|
||||
from superset.db_engine_specs.elasticsearch import OpenDistroEngineSpec
|
||||
|
||||
assert OpenDistroEngineSpec.make_label_compatible(original) == expected
|
||||
|
||||
|
||||
def test_opendistro_strip_comments() -> None:
|
||||
"""
|
||||
DB Eng Specs (opendistro): Test execute sql strip comments
|
||||
"""
|
||||
from superset.db_engine_specs.elasticsearch import OpenDistroEngineSpec
|
||||
|
||||
mock_database = MagicMock()
|
||||
mock_cursor = MagicMock()
|
||||
mock_cursor.execute.return_value = []
|
||||
|
||||
OpenDistroEngineSpec.execute(
|
||||
mock_cursor,
|
||||
"-- some comment \nSELECT 1\n --other comment",
|
||||
mock_database,
|
||||
)
|
||||
mock_cursor.execute.assert_called_once_with("SELECT 1\n")
|
||||
|
||||
@@ -27,7 +27,7 @@ from sqlalchemy.engine.url import make_url
|
||||
|
||||
from superset.errors import ErrorLevel, SupersetError, SupersetErrorType
|
||||
from superset.exceptions import SupersetException
|
||||
from superset.sql_parse import Table
|
||||
from superset.sql.parse import Table
|
||||
from superset.superset_typing import OAuth2ClientConfig
|
||||
from superset.utils import json
|
||||
from superset.utils.oauth2 import decode_oauth2_state
|
||||
|
||||
@@ -24,7 +24,7 @@ from pyhive.sqlalchemy_presto import PrestoDialect
|
||||
from sqlalchemy import column, sql, text, types
|
||||
from sqlalchemy.engine.url import make_url
|
||||
|
||||
from superset.sql_parse import Table
|
||||
from superset.sql.parse import Table
|
||||
from superset.utils.core import GenericDataType
|
||||
from tests.unit_tests.db_engine_specs.utils import (
|
||||
assert_column_spec,
|
||||
|
||||
@@ -44,7 +44,7 @@ from superset.db_engine_specs.exceptions import (
|
||||
SupersetDBAPIOperationalError,
|
||||
SupersetDBAPIProgrammingError,
|
||||
)
|
||||
from superset.sql_parse import Table
|
||||
from superset.sql.parse import Table
|
||||
from superset.superset_typing import (
|
||||
OAuth2ClientConfig,
|
||||
ResultSetColumnType,
|
||||
|
||||
Reference in New Issue
Block a user