grpc_mock 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fd11a8a9fe51756f230dd787fe6996643c68a7d4
-  data.tar.gz: 2f3567d682c7d6e934ba7a8ca4c1f64cc98f6fc9
+  metadata.gz: 5a3a30164b826b389b7ab94f98d006695d0fbbe0
+  data.tar.gz: 52a58e8da81742811f8fc9f63e6f1b4eed6e5ed9
 SHA512:
-  metadata.gz: 88dd0416c9169efe35bcb329eb6bb238458081adc70924f93e8d420b612f5aad88cc1aabd799ad010a672a990d03c584ba2aed3726c725963f0d3129ebf2e8ff
-  data.tar.gz: a1660786ac25573a5fb282b29fd31966cc9eedf0ee73a242f3df224a5109ff9088b49e723869ac4b3cab1bf1f954d7a9df8a1b3d398c2e002f80012dad44f75e
+  metadata.gz: 054c34b5b45030d9671da3bb41952f77105f43f6554dbe4ab0a3812d90fa578c58e172020290ff3b5be07907cec4906abbcad4f3af14a5478f585d0db2d1df33
+  data.tar.gz: f733ca899fc9319e175cd953c9117c063a1759844f39faabfea697f2ba2eb39202cdd0c4f3cff327d06704c0e6a2ac2368a9b19f8a78b457c64b78ff2601fdc3

data/README.md

@@ -1,4 +1,4 @@
-# GrpcMock
+# GrpcMock [![Build Status](https://travis-ci.org/ganmacs/grpc_mock.svg?branch=master)](https://travis-ci.org/ganmacs/grpc_mock)
 
 Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/grpc_mock`. To experiment with that code, run `bin/console` for an interactive prompt.
 

data/lib/grpc_mock.rb

@@ -1,5 +1,5 @@
 require 'grpc_mock/version'
-require 'grpc_mock/configure'
+require 'grpc_mock/configuration'
 require 'grpc_mock/adapter'
 
 module GrpcMock
@@ -25,7 +25,7 @@ module GrpcMock
     end
 
     def config
-      @config ||= Configure.new
+      @config ||= Configuration.new
     end
   end
 end

data/lib/grpc_mock/{configure.rb → configuration.rb}

@@ -1,5 +1,5 @@
 module GrpcMock
-  class Configure
+  class Configuration
     attr_accessor :allow_net_connect
 
     def initialize

data/lib/grpc_mock/grpc_stub_adapter.rb

@@ -3,89 +3,78 @@ require 'grpc_mock/errors'
 
 module GrpcMock
   class GrpcStubAdapter
-    # To make hook point for GRCP::ClientStub
+    # To make hook point for GRPC::ClientStub
     # https://github.com/grpc/grpc/blob/bec3b5ada2c5e5d782dff0b7b5018df646b65cb0/src/ruby/lib/grpc/generic/service.rb#L150-L186
-    ADAPTER_CLASS = Class.new(GRPC::ClientStub) do
-      alias_method :original_request_response, :request_response
-      alias_method :original_client_streamer, :client_streamer
-      alias_method :original_server_streamer, :server_streamer
-      alias_method :original_bidi_streamer, :bidi_streamer
-    end
-    GRPC.send(:remove_const, :ClientStub)
-    GRPC.send(:const_set, :ClientStub, ADAPTER_CLASS)
+    class AdapterClass < GRPC::ClientStub
+      def request_response(method, *args)
+        unless GrpcMock::GrpcStubAdapter.enabled?
+          return super
+        end
 
-    HOOKED_CLASS = Module.new do
-      def request_response(method, req, marshal, unmarshal, **opt)
         if GrpcMock.config.allow_net_connect
-          original_request_response(method, req, marshal, unmarshal, **opt)
+          super
         else
           raise NetConnectNotAllowedError, method
         end
       end
 
-      def client_streamer(method, requests, marshal, unmarshal, **opt)
+      def client_streamer(method, *args)
+        unless GrpcMock::GrpcStubAdapter.enabled?
+          return super
+        end
+
         if GrpcMock.config.allow_net_connect
-          original_client_streamer(method, requests, marshal, unmarshal, **opt)
+          super
         else
           raise NetConnectNotAllowedError, method
         end
       end
 
-      def server_streamer(method, req, marshal, unmarshal, **opt)
+      def server_streamer(method, *args)
+        unless GrpcMock::GrpcStubAdapter.enabled?
+          return super
+        end
+
         if GrpcMock.config.allow_net_connect
-          original_server_streamer(method, req, marshal, unmarshal, **opt)
+          super
         else
           raise NetConnectNotAllowedError, method
         end
       end
 
-      def bidi_streamer(method, requests, marshal, unmarshal, **opt)
+      def bidi_streamer(method, *args)
+        unless GrpcMock::GrpcStubAdapter.enabled?
+          return super
+        end
+
         if GrpcMock.config.allow_net_connect
-          original_bidi_streamer(method, requests, marshal, unmarshal, **opt)
+          super
         else
           raise NetConnectNotAllowedError, method
         end
       end
     end
+    GRPC.send(:remove_const, :ClientStub)
+    GRPC.send(:const_set, :ClientStub, AdapterClass)
 
-    # This class is only used for disabling grpc_stub_adapter hook
-    THROUGHT_CLASS = Module.new do
-      def request_response(method, req, marshal, unmarshal, **opt)
-        original_request_response(method, req, marshal, unmarshal, **opt)
-      end
-
-      def client_streamer(method, requests, marshal, unmarshal, **opt)
-        original_client_streamer(method, requests, marshal, unmarshal, **opt)
-      end
+    def self.disable!
+      @enabled = false
+    end
 
-      def server_streamer(method, req, marshal, unmarshal, **opt)
-        original_server_streamer(method, req, marshal, unmarshal, **opt)
-      end
+    def self.enable!
+      @enabled = true
+    end
 
-      def bidi_streamer(method, requests, marshal, unmarshal, **opt)
-        original_bidi_streamer(method, requests, marshal, unmarshal, **opt)
-      end
+    def self.enabled?
+      @enabled
     end
 
     def enable!
-      clean_up
-      ADAPTER_CLASS.prepend(HOOKED_CLASS)
+      GrpcMock::GrpcStubAdapter.enable!
     end
 
     def disable!
-      clean_up
-      ADAPTER_CLASS.prepend(THROUGHT_CLASS)
-    end
-
-    private
-
-    def clean_up
-      ADAPTER_CLASS.class_eval do
-        undef_method(:request_response)
-        undef_method(:client_streamer)
-        undef_method(:server_streamer)
-        undef_method(:bidi_streamer)
-      end
+      GrpcMock::GrpcStubAdapter.disable!
     end
   end
 end

data/lib/grpc_mock/version.rb

@@ -1,3 +1,3 @@
 module GrpcMock
-  VERSION = '0.1.1'.freeze
+  VERSION = '0.1.2'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: grpc_mock
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - ganmacs
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-07-09 00:00:00.000000000 Z
+date: 2018-07-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -129,7 +129,7 @@ files:
 - grpc_mock.gemspec
 - lib/grpc_mock.rb
 - lib/grpc_mock/adapter.rb
-- lib/grpc_mock/configure.rb
+- lib/grpc_mock/configuration.rb
 - lib/grpc_mock/errors.rb
 - lib/grpc_mock/grpc_stub_adapter.rb
 - lib/grpc_mock/rspec.rb