remove certain prefixes from host name

Author: berndverst

durabletask/internal/shared.py

@@ -15,6 +15,9 @@
 # and should be deserialized as a SimpleNamespace
 AUTO_SERIALIZED = "__durabletask_autoobject__"
 
+SECURE_PROTOCOLS = ["https://", "grpcs://"]
+INSECURE_PROTOCOLS = ["http://", "grpc://"]
+
 
 def get_default_host_address() -> str:
     return "localhost:4001"
@@ -27,6 +30,20 @@ def get_grpc_channel(
     if host_address is None:
         host_address = get_default_host_address()
 
+    for protocol in SECURE_PROTOCOLS:
+        if host_address.lower().startswith(protocol):
+            secure_channel = True
+            # remove the protocol from the host name
+            host_address = host_address[len(protocol):]
+            break
+
+    for protocol in INSECURE_PROTOCOLS:
+        if host_address.lower().startswith(protocol):
+            secure_channel = False
+            # remove the protocol from the host name
+            host_address = host_address[len(protocol):]
+            break
+
     if secure_channel:
         channel = grpc.secure_channel(host_address, grpc.ssl_channel_credentials())
     else:

tests/test_client.py

@@ -1,4 +1,4 @@
-from unittest.mock import patch
+from unittest.mock import patch, Mock, ANY
 
 from durabletask.internal.shared import (DefaultClientInterceptorImpl,
                                          get_default_host_address,
@@ -39,3 +39,50 @@ def test_get_grpc_channel_with_metadata():
         assert args[0] == mock_channel.return_value
         assert isinstance(args[1], DefaultClientInterceptorImpl)
         assert args[1]._metadata == METADATA
+
+
+def test_grpc_channel_with_host_name_protocol_stripping():
+    with patch('grpc.insecure_channel') as mock_insecure_channel, patch(
+            'grpc.secure_channel') as mock_secure_channel:
+
+        host_name = "myserver.com:1234"
+
+        prefix = "grpc://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_insecure_channel.assert_called_with(host_name)
+
+        prefix = "http://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_insecure_channel.assert_called_with(host_name)
+
+        prefix = "HTTP://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_insecure_channel.assert_called_with(host_name)
+
+        prefix = "GRPC://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_insecure_channel.assert_called_with(host_name)
+
+        prefix = ""
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_insecure_channel.assert_called_with(host_name)
+
+        prefix = "grpcs://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_secure_channel.assert_called_with(host_name, ANY)
+
+        prefix = "https://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_secure_channel.assert_called_with(host_name, ANY)
+
+        prefix = "HTTPS://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_secure_channel.assert_called_with(host_name, ANY)
+
+        prefix = "GRPCS://"
+        get_grpc_channel(prefix + host_name, METADATA)
+        mock_secure_channel.assert_called_with(host_name, ANY)
+
+        prefix = ""
+        get_grpc_channel(prefix + host_name, METADATA, True)
+        mock_secure_channel.assert_called_with(host_name, ANY)