Option to allow localhost and allowlist

lib/grpc_mock/api.rb

@@ -15,8 +15,10 @@ def request_including(values)
       GrpcMock::Matchers::RequestIncludingMatcher.new(values)
     end
 
-    def disable_net_connect!
+    def disable_net_connect!(allow_localhost: false, allow: nil)
       GrpcMock.config.allow_net_connect = false
+      GrpcMock.config.allow_localhost = allow_localhost
+      GrpcMock.config.allow = allow
     end
 
     def allow_net_connect!

lib/grpc_mock/configuration.rb

@@ -2,10 +2,12 @@
 
 module GrpcMock
   class Configuration
-    attr_accessor :allow_net_connect
+    attr_accessor :allow_net_connect, :allow_localhost, :allow
 
     def initialize
       @allow_net_connect = true
+      @allow_localhost = false
+      @allow = nil
     end
   end
 end

lib/grpc_mock/grpc_stub_adapter.rb

@@ -26,7 +26,7 @@ def request_response(method, request, *args, metadata: {}, return_op: false, **k
           else
             mock.evaluate(request, call.single_req_view)
           end
-        elsif GrpcMock.config.allow_net_connect
+        elsif connection_allowed?
           super
         else
           raise NetConnectNotAllowedError, method
@@ -52,7 +52,7 @@ def client_streamer(method, requests, *args, metadata: {}, return_op: false, **k
           else
             mock.evaluate(r, call.multi_req_view)
           end
-        elsif GrpcMock.config.allow_net_connect
+        elsif connection_allowed?
           super
         else
           raise NetConnectNotAllowedError, method
@@ -76,7 +76,7 @@ def server_streamer(method, request, *args, metadata: {}, return_op: false, **kw
           else
             mock.evaluate(request, call.single_req_view)
           end
-        elsif GrpcMock.config.allow_net_connect
+        elsif connection_allowed?
           super
         else
           raise NetConnectNotAllowedError, method
@@ -100,12 +100,48 @@ def bidi_streamer(method, requests, *args, metadata: {}, return_op: false, **kwa
           else
             mock.evaluate(r, nil) # FIXME: provide BidiCall equivalent
           end
-        elsif GrpcMock.config.allow_net_connect
+        elsif connection_allowed?
           super
         else
           raise NetConnectNotAllowedError, method
         end
       end
+
+      def connection_allowed?
+        return true if GrpcMock.config.allow_net_connect
+
+        uri = @host.include?("://") ? URI.parse(@host) : URI.parse("http://#{@host}")
+        return true if GrpcMock.config.allow_localhost && localhost_allowed?(uri)
+
+        net_connection_explicitly_allowed?(GrpcMock.config.allow, uri)
+      end
+
+      def localhost_allowed?(uri)
+        return false unless GrpcMock.config.allow_localhost
+        %w(localhost 127.0.0.1 0.0.0.0 [::1]).include?(uri.host)
+      end
+
+      def net_connection_explicitly_allowed?(allowed, uri)
+        return false unless GrpcMock.config.allow
+
+        case allowed
+        when Array
+          allowed.any? { |allowed_item| net_connection_explicitly_allowed?(allowed_item, uri) }
+        when Regexp
+          (uri.to_s =~ allowed) != nil ||
+          ("#{uri.scheme}://#{uri.host}" =~ allowed) != nil && uri.port == uri.default_port
+        when String
+          allowed == uri.to_s ||
+          allowed == uri.host ||
+          allowed == "#{uri.host}:#{uri.port}" ||
+          allowed == "#{uri.scheme}://#{uri.host}:#{uri.port}" ||
+          allowed == "#{uri.scheme}://#{uri.host}" && uri.port == uri.default_port
+        else
+          if allowed.respond_to?(:call)
+            allowed.call(uri)
+          end
+        end
+      end
     end
 
     def self.disable!

spec/examples/hello/hello_client.rb

@@ -1,23 +1,26 @@
 require_relative './hello_services_pb'
 
 class HelloClient
-  def initialize
-    @client = Hello::Hello::Stub.new('localhost:8000', :this_channel_is_insecure)
+  def initialize(host = 'localhost:8000')
+    @client = Hello::Hello::Stub.new(host, :this_channel_is_insecure)
   end
 
-  def send_message(msg, client_stream: false, server_stream: false, metadata: {})
+  def send_message(msg, client_stream: false, server_stream: false, metadata: {}, timeout: 0.1)
+    options = { metadata: metadata }
+    options[:deadline] = Time.now + timeout if timeout
+
     if client_stream && server_stream
       m = Hello::HelloStreamRequest.new(msg: msg)
-      @client.hello_stream([m].to_enum, metadata: metadata)
+      @client.hello_stream([m].to_enum, **options)
     elsif client_stream
       m = Hello::HelloStreamRequest.new(msg: msg)
-      @client.hello_client_stream([m].to_enum, metadata: metadata)
+      @client.hello_client_stream([m].to_enum, **options)
     elsif server_stream
       m = Hello::HelloRequest.new(msg: msg)
-      @client.hello_server_stream(m, metadata: metadata)
+      @client.hello_server_stream(m, **options)
     else
       m = Hello::HelloRequest.new(msg: msg)
-      @client.hello(m, metadata: metadata)
+      @client.hello(m, **options)
     end
   end
 end