Update tests/unit/test_job_retry.py

Author: chalmerlowe

tests/unit/test_job_retry.py

@@ -695,66 +695,3 @@ def test_retry_load_job_result(result_retry_param, PROJECT, DS_ID):
     # The number of api_request calls should still be 3
     assert conn.api_request.call_count == 3
 
-
-# @pytest.mark.parametrize(
-#     "result_retry",
-#     [
-#         pytest.param(
-#             {},
-#             id="default retry use case",
-#         ),
-#         pytest.param(
-#             {
-#                 "retry": google.cloud.bigquery.retry.DEFAULT_RETRY.with_timeout(
-#                     timeout=10.0
-#                 )
-#             },
-#             id="custom retry object use case",
-#         ),
-#     ],
-# )
-# def test_retry_load_job_result(result_retry, PROJECT, DS_ID):
-
-# from google.cloud.bigquery.dataset import DatasetReference
-# from google.cloud.bigquery.job.load import LoadJob
-# import google.cloud.bigquery.retry
-
-# client = make_client()
-# conn = client._connection = make_connection(
-#     dict(
-#         status=dict(state="RUNNING"),
-#         jobReference={"jobId": "id_1"},
-#     ),
-#     google.api_core.exceptions.ServiceUnavailable("retry me"),
-#     dict(
-#         status=dict(state="DONE"),
-#         jobReference={"jobId": "id_1"},
-#         statistics={"load": {"outputRows": 1}},
-#     ),
-# )
-
-# table_ref = DatasetReference(project=PROJECT, dataset_id=DS_ID).table("new_table")
-# job = LoadJob("id_1", source_uris=None, destination=table_ref, client=client)
-# result = job.result(**result_retry)
-
-# print(f"Test Case ID: {result_retry}")
-# for call in conn.api_request.mock_calls:
-#     name, args, kwargs = call
-#     print(f"  Call to api_request with kwargs: {kwargs}")
-
-# if result_retry:
-#     custom_retry_used = False
-#     for name, args, kwargs in conn.api_request.mock_calls:
-#         if 'retry' in kwargs:
-#             # Check if the retry object passed has the custom timeout
-#             if kwargs['retry']._timeout == 10.0:
-#                 custom_retry_used = True
-#                 assert kwargs['retry']._deadline == 10.0 # Deadline should also be updated
-#             # Optional: check other properties if needed
-#     assert custom_retry_used, "Custom retry object with timeout 10.0 was not used"
-
-# assert job.state == "DONE"
-# assert result.output_rows == 1
-
-# # We made all the calls we expected to.
-# assert conn.api_request.call_count == 3