fixed broken test with mock host string

Signed-off-by: Nikhil Suri <[email protected]>

Author: nikhilsuri-db

tests/unit/test_client.py

@@ -652,6 +652,7 @@ def _create_mock_connection(self, mock_session_class):
         mock_session = Mock()
         mock_session.is_open = True
         mock_session.guid_hex = "test-session-id"
+        mock_session.host = "foo"  # Match server_hostname from DUMMY_CONNECTION_ARGS
         mock_session.get_autocommit.return_value = True
         mock_session_class.return_value = mock_session
 
@@ -927,6 +928,7 @@ def test_fetch_autocommit_from_server_queries_server(self, mock_session_class):
         mock_session = Mock()
         mock_session.is_open = True
         mock_session.guid_hex = "test-session-id"
+        mock_session.host = "foo"
         mock_session_class.return_value = mock_session
 
         conn = client.Connection(
@@ -959,6 +961,7 @@ def test_fetch_autocommit_from_server_handles_false_value(self, mock_session_cla
         mock_session = Mock()
         mock_session.is_open = True
         mock_session.guid_hex = "test-session-id"
+        mock_session.host = "foo"
         mock_session_class.return_value = mock_session
 
         conn = client.Connection(
@@ -986,6 +989,7 @@ def test_fetch_autocommit_from_server_raises_on_no_result(self, mock_session_cla
         mock_session = Mock()
         mock_session.is_open = True
         mock_session.guid_hex = "test-session-id"
+        mock_session.host = "foo"
         mock_session_class.return_value = mock_session
 
         conn = client.Connection(
@@ -1015,6 +1019,7 @@ def test_commit_is_noop_when_ignore_transactions_true(self, mock_session_class):
         mock_session = Mock()
         mock_session.is_open = True
         mock_session.guid_hex = "test-session-id"
+        mock_session.host = "foo"
         mock_session_class.return_value = mock_session
 
         # Create connection with ignore_transactions=True (default)
@@ -1043,6 +1048,7 @@ def test_rollback_raises_not_supported_when_ignore_transactions_true(
         mock_session = Mock()
         mock_session.is_open = True
         mock_session.guid_hex = "test-session-id"
+        mock_session.host = "foo"
         mock_session_class.return_value = mock_session
 
         # Create connection with ignore_transactions=True (default)
@@ -1068,6 +1074,7 @@ def test_autocommit_setter_is_noop_when_ignore_transactions_true(
         mock_session = Mock()
         mock_session.is_open = True
         mock_session.guid_hex = "test-session-id"
+        mock_session.host = "foo"
         mock_session_class.return_value = mock_session
 
         # Create connection with ignore_transactions=True (default)