Updating unit tests

Signed-off-by: Ryan Lettieri <ryanLettieri@microsoft.com>

Author: RyanLettieri

durabletask/internal/shared.py

@@ -24,7 +24,6 @@ def get_default_host_address() -> str:
 
 def get_grpc_channel(
         host_address: Optional[str],
-        metadata: Optional[list[tuple[str, str]]] = None,
         secure_channel: bool = False,
         interceptors: Optional[list[DefaultClientInterceptorImpl]] = None) -> grpc.Channel:
 

tests/test_client.py

@@ -6,31 +6,31 @@
 
 HOST_ADDRESS = 'localhost:50051'
 METADATA = [('key1', 'value1'), ('key2', 'value2')]
-
+INTERCEPTORS = DefaultClientInterceptorImpl(METADATA)
 
 def test_get_grpc_channel_insecure():
     with patch('grpc.insecure_channel') as mock_channel:
-        get_grpc_channel(HOST_ADDRESS, METADATA, False)
+        get_grpc_channel(HOST_ADDRESS, False, interceptors=INTERCEPTORS)
         mock_channel.assert_called_once_with(HOST_ADDRESS)
 
 
 def test_get_grpc_channel_secure():
     with patch('grpc.secure_channel') as mock_channel, patch(
             'grpc.ssl_channel_credentials') as mock_credentials:
-        get_grpc_channel(HOST_ADDRESS, METADATA, True)
+        get_grpc_channel(HOST_ADDRESS, True, interceptors=INTERCEPTORS)
         mock_channel.assert_called_once_with(HOST_ADDRESS, mock_credentials.return_value)
 
 
 def test_get_grpc_channel_default_host_address():
     with patch('grpc.insecure_channel') as mock_channel:
-        get_grpc_channel(None, METADATA, False)
+        get_grpc_channel(None, False, interceptors=INTERCEPTORS)
         mock_channel.assert_called_once_with(get_default_host_address())
 
 
 def test_get_grpc_channel_with_metadata():
     with patch('grpc.insecure_channel') as mock_channel, patch(
             'grpc.intercept_channel') as mock_intercept_channel:
-        get_grpc_channel(HOST_ADDRESS, METADATA, False)
+        get_grpc_channel(HOST_ADDRESS, False, interceptors=INTERCEPTORS)
         mock_channel.assert_called_once_with(HOST_ADDRESS)
         mock_intercept_channel.assert_called_once()
 
@@ -48,41 +48,41 @@ def test_grpc_channel_with_host_name_protocol_stripping():
         host_name = "myserver.com:1234"
 
         prefix = "grpc://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_insecure_channel.assert_called_with(host_name)
 
         prefix = "http://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_insecure_channel.assert_called_with(host_name)
 
         prefix = "HTTP://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_insecure_channel.assert_called_with(host_name)
 
         prefix = "GRPC://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_insecure_channel.assert_called_with(host_name)
 
         prefix = ""
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_insecure_channel.assert_called_with(host_name)
 
         prefix = "grpcs://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_secure_channel.assert_called_with(host_name, ANY)
 
         prefix = "https://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_secure_channel.assert_called_with(host_name, ANY)
 
         prefix = "HTTPS://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_secure_channel.assert_called_with(host_name, ANY)
 
         prefix = "GRPCS://"
-        get_grpc_channel(prefix + host_name, METADATA)
+        get_grpc_channel(prefix + host_name, interceptors=INTERCEPTORS)
         mock_secure_channel.assert_called_with(host_name, ANY)
 
         prefix = ""
-        get_grpc_channel(prefix + host_name, METADATA, True)
+        get_grpc_channel(prefix + host_name, True, interceptors=INTERCEPTORS)
         mock_secure_channel.assert_called_with(host_name, ANY)