fix: do_ping takes a connection, not engine (#39013)

This commit is contained in:
Beto Dealmeida
2026-04-15 11:10:24 -04:00
committed by GitHub
parent ddcb9be9a7
commit 84f7b4a973
2 changed files with 5 additions and 4 deletions

View File

@@ -77,7 +77,7 @@ def test_ping_sqlite_exception(mocker: MockerFixture, mock_engine: MockerFixture
assert result is True
mock_dialect.do_ping.assert_has_calls(
[mocker.call(mock_connection), mocker.call(mock_engine)]
[mocker.call(mock_connection), mocker.call(mock_connection)]
)
@@ -85,7 +85,7 @@ def test_ping_runtime_exception(mocker: MockerFixture, mock_engine: MockerFixtur
"""
Test the ``ping`` method when a RuntimeError is raised.
"""
mock_engine, _, mock_dialect = mock_engine
mock_engine, mock_connection, mock_dialect = mock_engine
mock_timeout = mocker.patch("superset.commands.database.utils.timeout")
mock_timeout.side_effect = RuntimeError("timeout")
mock_dialect.do_ping.return_value = True
@@ -93,7 +93,7 @@ def test_ping_runtime_exception(mocker: MockerFixture, mock_engine: MockerFixtur
result = ping(mock_engine)
assert result is True
mock_dialect.do_ping.assert_called_once_with(mock_engine)
mock_dialect.do_ping.assert_called_once_with(mock_connection)
@pytest.fixture