diff --git a/tests/profiles/vertica/test_vertica_user_pass.py b/tests/profiles/vertica/test_vertica_user_pass.py index bc93014a3..953a3c553 100644 --- a/tests/profiles/vertica/test_vertica_user_pass.py +++ b/tests/profiles/vertica/test_vertica_user_pass.py @@ -22,7 +22,7 @@ def mock_vertica_conn(): # type: ignore host="my_host", login="my_user", password="my_password", - port=5432, + port=5433, schema="my_schema", extra='{"database": "my_database"}', ) @@ -112,6 +112,14 @@ def test_profile_mapping_selected( assert isinstance(profile_mapping, VerticaUserPasswordProfileMapping) +def test_mock_profile() -> None: + """ + Tests that the mock profile port value get set correctly. + """ + profile = VerticaUserPasswordProfileMapping("mock_conn_id") + assert profile.mock_profile.get("port") == 5433 + + def test_profile_mapping_keeps_custom_port(mock_vertica_conn_custom_port: Connection) -> None: profile = VerticaUserPasswordProfileMapping(mock_vertica_conn_custom_port.conn_id, {"schema": "my_schema"}) assert profile.profile["port"] == 7472