mirror of
https://github.com/apache/superset.git
synced 2026-04-16 22:55:52 +00:00
chore(command): Organize Commands according to SIP-92 (#25850)
This commit is contained in:
@@ -19,11 +19,11 @@
|
||||
import pytest
|
||||
from sqlalchemy.orm.session import Session
|
||||
|
||||
from superset.databases.ssh_tunnel.commands.exceptions import SSHTunnelInvalidError
|
||||
from superset.commands.database.ssh_tunnel.exceptions import SSHTunnelInvalidError
|
||||
|
||||
|
||||
def test_create_ssh_tunnel_command() -> None:
|
||||
from superset.databases.ssh_tunnel.commands.create import CreateSSHTunnelCommand
|
||||
from superset.commands.database.ssh_tunnel.create import CreateSSHTunnelCommand
|
||||
from superset.databases.ssh_tunnel.models import SSHTunnel
|
||||
from superset.models.core import Database
|
||||
|
||||
@@ -44,7 +44,7 @@ def test_create_ssh_tunnel_command() -> None:
|
||||
|
||||
|
||||
def test_create_ssh_tunnel_command_invalid_params() -> None:
|
||||
from superset.databases.ssh_tunnel.commands.create import CreateSSHTunnelCommand
|
||||
from superset.commands.database.ssh_tunnel.create import CreateSSHTunnelCommand
|
||||
from superset.databases.ssh_tunnel.models import SSHTunnel
|
||||
from superset.models.core import Database
|
||||
|
||||
|
||||
@@ -54,8 +54,8 @@ def session_with_data(session: Session) -> Iterator[Session]:
|
||||
def test_delete_ssh_tunnel_command(
|
||||
mocker: MockFixture, session_with_data: Session
|
||||
) -> None:
|
||||
from superset.commands.database.ssh_tunnel.delete import DeleteSSHTunnelCommand
|
||||
from superset.daos.database import DatabaseDAO
|
||||
from superset.databases.ssh_tunnel.commands.delete import DeleteSSHTunnelCommand
|
||||
from superset.databases.ssh_tunnel.models import SSHTunnel
|
||||
|
||||
result = DatabaseDAO.get_ssh_tunnel(1)
|
||||
@@ -64,7 +64,7 @@ def test_delete_ssh_tunnel_command(
|
||||
assert isinstance(result, SSHTunnel)
|
||||
assert 1 == result.database_id
|
||||
mocker.patch(
|
||||
"superset.databases.ssh_tunnel.commands.delete.is_feature_enabled",
|
||||
"superset.commands.database.ssh_tunnel.delete.is_feature_enabled",
|
||||
return_value=True,
|
||||
)
|
||||
DeleteSSHTunnelCommand(1).run()
|
||||
|
||||
@@ -20,7 +20,7 @@ from collections.abc import Iterator
|
||||
import pytest
|
||||
from sqlalchemy.orm.session import Session
|
||||
|
||||
from superset.databases.ssh_tunnel.commands.exceptions import SSHTunnelInvalidError
|
||||
from superset.commands.database.ssh_tunnel.exceptions import SSHTunnelInvalidError
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
@@ -50,8 +50,8 @@ def session_with_data(session: Session) -> Iterator[Session]:
|
||||
|
||||
|
||||
def test_update_shh_tunnel_command(session_with_data: Session) -> None:
|
||||
from superset.commands.database.ssh_tunnel.update import UpdateSSHTunnelCommand
|
||||
from superset.daos.database import DatabaseDAO
|
||||
from superset.databases.ssh_tunnel.commands.update import UpdateSSHTunnelCommand
|
||||
from superset.databases.ssh_tunnel.models import SSHTunnel
|
||||
|
||||
result = DatabaseDAO.get_ssh_tunnel(1)
|
||||
@@ -72,8 +72,8 @@ def test_update_shh_tunnel_command(session_with_data: Session) -> None:
|
||||
|
||||
|
||||
def test_update_shh_tunnel_invalid_params(session_with_data: Session) -> None:
|
||||
from superset.commands.database.ssh_tunnel.update import UpdateSSHTunnelCommand
|
||||
from superset.daos.database import DatabaseDAO
|
||||
from superset.databases.ssh_tunnel.commands.update import UpdateSSHTunnelCommand
|
||||
from superset.databases.ssh_tunnel.models import SSHTunnel
|
||||
|
||||
result = DatabaseDAO.get_ssh_tunnel(1)
|
||||
|
||||
Reference in New Issue
Block a user