Fixing test I didn't touch

preset-io · Oct 2, 2023 · 3f40ab2 · 3f40ab2
1 parent d040d2a
commit 3f40ab2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit_tests/db_engine_specs/test_clickhouse.py b/tests/unit_tests/db_engine_specs/test_clickhouse.py
@@ -16,6 +16,7 @@
 # under the License.
 
 from datetime import datetime
+from http.client import HTTPConnection
 from typing import Any, Optional
 from unittest.mock import Mock
 
@@ -63,7 +64,7 @@ def test_execute_connection_error() -> None:
 
     cursor = Mock()
     cursor.execute.side_effect = NewConnectionError(
-        "Dummypool", "Exception with sensitive data"
+        HTTPConnection("localhost", 8080), "Exception with sensitive data"
     )
     with pytest.raises(SupersetDBAPIDatabaseError) as ex:
         ClickHouseEngineSpec.execute(cursor, "SELECT col1 from table1")