fix tests

Author: rahul2393

tests/unit/gapic/spanner_admin_database_v1/test_database_admin.py

@@ -1136,6 +1136,7 @@ def test_database_admin_client_create_channel_credentials_file(
             options=[
                 ("grpc.max_send_message_length", -1),
                 ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
             ],
         )
 
@@ -23315,6 +23316,7 @@ def test_database_admin_transport_create_channel(transport_class, grpc_helpers):
             options=[
                 ("grpc.max_send_message_length", -1),
                 ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
             ],
         )
 
@@ -23347,6 +23349,7 @@ def test_database_admin_grpc_transport_client_cert_source_for_mtls(transport_cla
             options=[
                 ("grpc.max_send_message_length", -1),
                 ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
             ],
         )
 
@@ -23590,11 +23593,12 @@ def test_database_admin_transport_channel_mtls_with_client_cert_source(transport
                 scopes=None,
                 ssl_credentials=mock_ssl_cred,
                 quota_project_id=None,
-                options=[
-                    ("grpc.max_send_message_length", -1),
-                    ("grpc.max_receive_message_length", -1),
-                ],
-            )
+            options=[
+                ("grpc.max_send_message_length", -1),
+                ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
+            ],
+        )
             assert transport.grpc_channel == mock_grpc_channel
             assert transport._ssl_channel_credentials == mock_ssl_cred
 
@@ -23637,11 +23641,12 @@ def test_database_admin_transport_channel_mtls_with_adc(transport_class):
                 scopes=None,
                 ssl_credentials=mock_ssl_cred,
                 quota_project_id=None,
-                options=[
-                    ("grpc.max_send_message_length", -1),
-                    ("grpc.max_receive_message_length", -1),
-                ],
-            )
+            options=[
+                ("grpc.max_send_message_length", -1),
+                ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
+            ],
+        )
             assert transport.grpc_channel == mock_grpc_channel
 
 

tests/unit/gapic/spanner_admin_instance_v1/test_instance_admin.py

@@ -1125,6 +1125,7 @@ def test_instance_admin_client_create_channel_credentials_file(
             options=[
                 ("grpc.max_send_message_length", -1),
                 ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
             ],
         )
 
@@ -18621,6 +18622,7 @@ def test_instance_admin_transport_create_channel(transport_class, grpc_helpers):
             options=[
                 ("grpc.max_send_message_length", -1),
                 ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
             ],
         )
 
@@ -18653,6 +18655,7 @@ def test_instance_admin_grpc_transport_client_cert_source_for_mtls(transport_cla
             options=[
                 ("grpc.max_send_message_length", -1),
                 ("grpc.max_receive_message_length", -1),
+                ("grpc.keepalive_time_ms", 120000),
             ],
         )
 
@@ -18881,6 +18884,7 @@ def test_instance_admin_transport_channel_mtls_with_client_cert_source(transport
                 options=[
                     ("grpc.max_send_message_length", -1),
                     ("grpc.max_receive_message_length", -1),
+                    ("grpc.keepalive_time_ms", 120000),
                 ],
             )
             assert transport.grpc_channel == mock_grpc_channel
@@ -18928,6 +18932,7 @@ def test_instance_admin_transport_channel_mtls_with_adc(transport_class):
                 options=[
                     ("grpc.max_send_message_length", -1),
                     ("grpc.max_receive_message_length", -1),
+                    ("grpc.keepalive_time_ms", 120000),
                 ],
             )
             assert transport.grpc_channel == mock_grpc_channel