test_ssl_retry.py

# Copyright 2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import time
from unittest import mock

import pytest
import requests.exceptions
from google.api_core import exceptions as core_exceptions
from google.cloud import bigquery

def test_insert_rows_json_ssl_error_no_retry(bigquery_client, dataset_id):
    """
    Verify that SSLError during insert_rows_json is NOT retried and
    propagates a descriptive error message immediately.
    """
    table_id = f"{dataset_id}.test_ssl_retry_{int(time.time())}"
    schema = [bigquery.SchemaField("name", "STRING")]
    table = bigquery.Table(table_id, schema=schema)
    bigquery_client.create_table(table)
    
    # We mock the api_request to simulate the GFE abruptly closing the connection
    # which manifests as a requests.exceptions.SSLError.
    original_api_request = bigquery_client._connection.api_request
    call_count = 0

    def mock_api_request(*args, **kwargs):
        nonlocal call_count
        call_count += 1
        raise requests.exceptions.SSLError("EOF occurred in violation of protocol")

    with mock.patch.object(bigquery_client._connection, "api_request", side_effect=mock_api_request):
        # Use a reasonably short deadline for the test, although it should fail on the first attempt anyway.
        retry = bigquery.DEFAULT_RETRY.with_deadline(5.0)
        
        start_time = time.time()
        with pytest.raises(requests.exceptions.SSLError) as excinfo:
            bigquery_client.insert_rows_json(
                table, 
                [{"name": "test"}],
                retry=retry
            )
        duration = time.time() - start_time

    # Verification:
    # 1. It should NOT have retried (total calls should be 1)
    assert call_count == 1
    
    # 2. It should have failed quickly (much less than the 5s deadline)
    assert duration < 2.0
    
    # 3. The error message should contain our descriptive wrapping
    assert "invalid table schema" in str(excinfo.value)
    assert "SSL/Connection error occurred" in str(excinfo.value)

    # Cleanup
    bigquery_client.delete_table(table_id)