Making mypy happy

exasol · Mar 11, 2024 · 9d0759f · 9d0759f
1 parent 5db3a6b
commit 9d0759f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/exasol/nb_connector/connections.py b/exasol/nb_connector/connections.py
@@ -138,7 +138,7 @@ def open_sqlalchemy_connection(conf: Secrets):
                                username=conf.get(CKey.db_user),
                                password=conf.get(CKey.db_password),
                                host=conf.get(CKey.db_host_name),
-                               port=int(conf.get(CKey.db_port)),
+                               port=int(getattr(conf, CKey.db_port.name)),
                                database=conf.get(CKey.db_schema),
                                query=query_params
                                )

diff --git a/test/unit/test_connections.py b/test/unit/test_connections.py
@@ -107,6 +107,7 @@ def test_open_pyexasol_connection_error(mock_connect, conf):
 
 @unittest.mock.patch("sqlalchemy.create_engine")
 def test_open_sqlalchemy_connection(mock_create_engine, conf):
+    setattr(conf, CKey.db_port.name, conf.get(CKey.db_port))
     open_sqlalchemy_connection(conf)
     mock_create_engine.assert_called_once_with(
         make_url(f"exa+websocket://{conf.get(CKey.db_user)}:{conf.get(CKey.db_password)}@{get_external_host(conf)}")
@@ -117,6 +118,7 @@ def test_open_sqlalchemy_connection(mock_create_engine, conf):
 def test_open_sqlalchemy_connection_ssl(mock_create_engine, conf):
     conf.save(CKey.db_encryption, "True")
     conf.save(CKey.cert_vld, "False")
+    setattr(conf, CKey.db_port.name, conf.get(CKey.db_port))
 
     open_sqlalchemy_connection(conf)
     mock_create_engine.assert_called_once_with(