tshield 0.11.15.0 → 0.11.20.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e6ed543aa8bab1fff2c4d75d7479b7af0b3e2f06432330a697b6808048466228
-  data.tar.gz: d14d318efb407d1a927cb2c5ac745081531c307f011d873965793c77ec0fd1b6
+  metadata.gz: 1880df41008713100967696d1bf1695b1e830851877f0ed5340f1f3f75cabd5e
+  data.tar.gz: 2a540ed72557ad9f9ba32073a5e98ed9b22812c1d0b96feaec00eeed35494386
 SHA512:
-  metadata.gz: 7a46edf78463d75e523fd842397bae66e749dc58cd31d5fab8500dab706028850963d8e84f366706e0aef7807f45283d74ad785379f38b9b144d40b801681155
-  data.tar.gz: 2eafe43d7acd092e04ff7b7dc1f2fcdd6dc7de2789a7f10e441bda3b72e838e13df86b8193ec7f3cbbd30a7f43a5880ce4c1629faf313ee3db3363695858aee0
+  metadata.gz: 7187004fa9e86545dcf81ebf70ea3059e04a0a8dcfe0143be3714ead2df3b9fc7d3e981641514f38192d69e57f826c85b14c46965f5d8171e3813e1bd9188574
+  data.tar.gz: d1e5e07ff88b722a944d78e22f1d3089214fb1ae9bed74c7bd42943700fc63bc096fc7916368513c83e660cb7936044dffd52e0f5c5b927d2618364a1411074e

data/Gemfile

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
 
 source 'https://rubygems.org'
-gem 'sinatra-cross_origin'
 gemspec

data/README.md

@@ -13,9 +13,10 @@ TShield is an open source proxy for mocks API responses.
 *   REST
 *   SOAP
 *   Session manager to separate multiple scenarios (success, error, sucess variation, ...)
+*   gRPC [EXPERIMENTAL]
 *   Lightweight
 *   MIT license
-    
+
 ## Table of Contents
 
 *   [Basic Usage](#basic-usage)
@@ -26,7 +27,7 @@ TShield is an open source proxy for mocks API responses.
 *   [Features](#features)
 *   [Examples](#examples)
 *   [Contributing](#contributing)
-    
+
 ## Basic Usage
 ### Install
 
@@ -37,7 +38,7 @@ TShield is an open source proxy for mocks API responses.
 To run server execute this command
 
     tshield
-    
+
 Default port is `4567`
 
 #### Command Line Options
@@ -102,7 +103,7 @@ To register stub into a session create an object with following attributes:
 *   **session**: name of session.
 *   **stubs**: an array with objects described above.
 
-### Example of matching configuration
+### Example of HTTP matching configuration
 
 ```json
 [
@@ -163,7 +164,7 @@ To register stub into a session create an object with following attributes:
 ]
 ```
 
-## Config options for VCR
+## Config options for HTTP VCR
 ```yaml
 request:
   timeout: 8
@@ -227,7 +228,7 @@ You can use TShield sessions to separate multiple scenarios for your mocks
 By default TShield save request/response into 
 
     requests/<<domain_name>>/<<resource_with_param>>/<<http_verb>>/<<index_based.content and json>>
-    
+
 If you start a session a folder with de **session_name** will be placed between **"requests/"** and **"<<domain_name>>"**
 
 ### Start TShield session
@@ -249,6 +250,45 @@ _DELETE_ to http://localhost:4567/sessions
 curl -X DELETE \
   http://localhost:4567/sessions
 ```
+### Append secondary TShield session
+**Append session. Secondary sessions will used only for read content in VCR mode, writes will be do in the main session. Append only works if exists a current session setted.**
+
+_POST_ to http://localhost:4567/sessions?name=<<same_name>>
+
+```
+curl -X POST \
+  'http://localhost:4567/sessions/append?name=my_valid'
+```
+
+## [Experimental] Config options for gRPC
+
+```yaml
+grpc:
+  port: 5678
+  proto_dir: 'proto'
+  services:
+    'helloworld_services_pb': 
+      module: 'Helloworld::Greeter'
+      hostname: '0.0.0.0:50051'
+```
+
+
+### Not Implemented Yet 
+
+- Matching
+
+### Configuration
+
+First, generate ruby files from proto files. Use `grpc_tools_ruby_protoc`
+present in the gem `grpc-tools`. Example:
+
+`grpc_tools_ruby_protoc -I proto --ruby_out=proto --grpc_out=proto proto/<INPUT>.proto`
+
+Call example in component_tests using [grpcurl](https://github.com/fullstorydev/grpcurl):
+
+`grpcurl -plaintext -import-path component_tests/proto -proto helloworld.proto  -d '{"name": "teste"}' localhost:5678 helloworld.Greeter/SayHello`
+
+### Using in VCR mode
 
 ## Custom controllers
 

data/Rakefile

@@ -32,5 +32,5 @@ end
 
 task :server do
   $LOAD_PATH.unshift File.dirname('./lib/tshield.rb')
-  exec 'bin/tshield'
+  Thread.new { exec 'bin/tshield' }
 end

data/bin/tshield

@@ -5,14 +5,6 @@ require 'tshield/options'
 TShield::Options.init
 
 require 'tshield'
-tshield = Thread.new { TShield::Server.run! }
 
-configuration = TShield::Configuration.load_configuration
-(configuration.tcp_servers || []).each do |tcp_server|
-  puts "initializing #{tcp_server['name']}"
-  require "./servers/#{tcp_server['file']}"
-  klass = Object.const_get(tcp_server['name'])
-  Thread.new { klass.new.listen(tcp_server['port']) }
-end
-
-tshield.join
+Thread.new { TShield::Grpc.run! }
+TShield::Server.run!

data/config/tshield.yml

@@ -0,0 +1,19 @@
+---
+grpc:
+  port: 5678
+  proto_dir: 'proto'
+  services:
+    'helloworld_services_pb': 
+      module: 'Helloworld::Greeter'
+      hostname: '0.0.0.0:50051'
+request:
+  timeout: 10
+domains:
+  'http://localhost:9090':
+    name: 'components'
+    skip_query_params:
+      - b
+    paths:
+      - /fake
+      - /users
+      - /resources

data/lib/tshield.rb

@@ -2,8 +2,8 @@
 
 require 'tshield/extensions/string_extensions'
 require 'tshield/options'
-require 'tshield/simple_tcp_server'
 require 'tshield/server'
+require 'tshield/grpc'
 
 # TShield: API mocks for development and testing
 module TShield

data/lib/tshield/configuration.rb

@@ -103,6 +103,11 @@ module TShield
       session_path || '/sessions'
     end
 
+    def grpc
+      defaults = { 'port' => 5678, 'proto_dir' => 'proto', 'services' => {} }
+      defaults.merge(@grpc || {})
+    end
+
     def self.get_url_for_domain_by_path(path, config)
       config['paths'].select { |pattern| path =~ Regexp.new(pattern) }[0]
     end

data/lib/tshield/controllers/helpers/session_helpers.rb

@@ -10,6 +10,11 @@ module TShield
           session[:name] if session
         end
 
+        def self.secondary_sessions(request)
+          session = TShield::Sessions.current(request.ip)
+          session[:secondary_sessions] if session
+        end
+
         def self.current_session_call(request, callid, method)
           session = TShield::Sessions.current(request.ip)
           session ? session[:counter].current(callid, method) : 0

data/lib/tshield/controllers/requests.rb

@@ -43,6 +43,7 @@ module TShield
           request_content_type = request.content_type
 
           session_name = TShield::Controllers::Helpers::SessionHelpers.current_session_name(request)
+          secondary_sessions = TShield::Controllers::Helpers::SessionHelpers.secondary_sessions(request)
           session_call = TShield::Controllers::Helpers::SessionHelpers
                          .current_session_call(request, callid, method)
 
@@ -51,6 +52,7 @@ module TShield
             headers: Helpers.build_headers(request),
             raw_query: request.env['QUERY_STRING'],
             session: session_name,
+            secondary_sessions: secondary_sessions,
             call: session_call,
             ip: request.ip
           }
@@ -67,7 +69,7 @@ module TShield
           unless api_response
             add_headers(options, path)
 
-            api_response ||= TShield::RequestVCR.new(path, options.clone).response
+            api_response ||= TShield::RequestVCR.new(path, options.clone).vcr_response
             api_response.headers.reject! do |key, _v|
               configuration.get_excluded_headers(domain(path)).include?(key)
             end

data/lib/tshield/controllers/sessions.rb

@@ -23,6 +23,9 @@ module TShield
       end
 
       def self.register_post(app, session_path)
+        app.post "#{session_path}/append" do
+          TShield::Sessions.append(request.ip, params[:name]).to_json
+        end
         app.post session_path do
           TShield::Sessions.start(request.ip, params[:name]).to_json
         end

data/lib/tshield/errors.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+class AppendSessionWithoutMainSessionError < RuntimeError
+end

data/lib/tshield/extensions/string_extensions.rb

@@ -5,6 +5,14 @@ module StringExtensions
   def to_rack_name
     "HTTP_#{upcase.tr('-', '_')}"
   end
+
+  def underscore
+    gsub(/::/, '/')
+      .gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
+      .gsub(/([a-z\d])([A-Z])/, '\1_\2')
+      .tr('-', '_')
+      .downcase
+  end
 end
 
 String.include StringExtensions

data/lib/tshield/grpc.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: false
+
+require 'grpc'
+
+require 'tshield/configuration'
+require 'tshield/grpc/vcr'
+
+module TShield
+  module Grpc
+    module RequestHandler
+      include TShield::Grpc::VCR
+      def handler(method_name, request, parameters)
+        options = self.class.options
+        handler_in_vcr_mode(method_name, request, parameters, options)
+      end
+    end
+    def self.run!
+      @configuration = TShield::Configuration.singleton.grpc
+
+      lib_dir = File.join(Dir.pwd, @configuration['proto_dir'])
+      $LOAD_PATH.unshift(lib_dir) unless $LOAD_PATH.include?(lib_dir)
+
+      TShield.logger.info("loading proto files from #{lib_dir}")
+
+      bind = "0.0.0.0:#{@configuration['port']}"
+      TShield.logger.info("Starting gRPC server in #{bind}")
+
+      server = GRPC::RpcServer.new
+      server.add_http2_port(bind, :this_port_is_insecure)
+
+      services = load_services(@configuration['services'])
+      services.each do |class_service|
+        class_service.include RequestHandler
+        server.handle(class_service)
+      end
+
+      server.run_till_terminated_or_interrupted([1, 'int', 'SIGQUIT']) unless services.empty?
+    end
+
+    def self.load_services(services)
+      handlers = []
+      number_of_handlers = 0
+      services.each do |file, options|
+        require file
+
+        base = Object.const_get("#{options['module']}::Service")
+        number_of_handlers += 1
+
+        implementation = build_handler(base, base.rpc_descs, number_of_handlers, options)
+        handlers << implementation
+      end
+      handlers
+    end
+
+    def self.build_handler(base, descriptions, number_of_handlers, options)
+      handler = Class.new(base) do
+        class << self
+          attr_writer :options
+          attr_reader :options
+        end
+        descriptions.each do |service_name, description|
+          puts description
+          method_name = service_name.to_s.underscore.to_sym
+          define_method(method_name) do |request, parameters|
+            handler(__method__, request, parameters)
+          end
+        end
+      end
+      handler.options = options
+      TShield::Grpc.const_set "GrpcService#{number_of_handlers}", handler
+    end
+  end
+end

data/lib/tshield/grpc/vcr.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+require 'tshield/sessions'
+
+module TShield
+  module Grpc
+    module VCR
+      def handler_in_vcr_mode(method_name, request, parameters, options)
+        parameters.peer =~ /ipv6:\[(.+?)\]|ipv4:(.+?):/
+        peer = Regexp.last_match(1) || Regexp.last_match(2)
+
+        TShield.logger.info("request from #{parameters.peer}")
+        @session = TShield::Sessions.current(peer)
+
+        TShield.logger.info("grpc using session #{@session || 'default'}")
+        module_name = options['module']
+
+        path = create_destiny(module_name, method_name, request)
+        response = saved_response(path)
+        if response
+          TShield.logger.info("returning saved response for request #{request.to_json} saved into #{hexdigest(request)}")
+          return response
+        end
+
+        TShield.logger.info("calling server to get response for #{request.to_json}")
+        client_class = Object.const_get("#{module_name}::Stub")
+        client_instance = client_class.new(options['hostname'], :this_channel_is_insecure)
+        response = client_instance.send(method_name, request)
+        save_request_and_response(path, request, response)
+        response
+      end
+
+      def saved_response(path)
+        response_file = File.join(path, 'response')
+        return false unless File.exist? response_file
+
+        content = JSON.parse File.open(response_file).read
+        response_class = File.open(File.join(path, 'response_class')).read.strip
+        Kernel.const_get(response_class).new(content)
+      end
+
+      def save_request_and_response(path, request, response)
+        save_request(path, request)
+        save_response(path, response)
+      end
+
+      def save_request(path, request)
+        file = File.open(File.join(path, 'original_request'), 'w')
+        file.puts request.to_json
+        file.close
+      end
+
+      def save_response(path, response)
+        file = File.open(File.join(path, 'response'), 'w')
+        file.puts response.to_json
+        file.close
+
+        response_class = File.open(File.join(path, 'response_class'), 'w')
+        response_class.puts response.class.to_s
+        response_class.close
+      end
+
+      def complete_path(module_name, method_name, request)
+        @session_name = (@session || {})[:name]
+        path = ['requests', 'grpc', @session_name, module_name, method_name.to_s, hexdigest(request)].compact
+        path
+      end
+
+      def create_destiny(module_name, method_name, request)
+        current_path = []
+
+        path = complete_path(module_name, method_name, request)
+        TShield.logger.info("using path #{path}")
+        path.each do |path|
+          current_path << path
+          destiny = File.join current_path
+          Dir.mkdir destiny unless File.exist? destiny
+        end
+        path
+      end
+
+      def hexdigest(request)
+        Digest::SHA1.hexdigest request.to_json
+      end
+    end
+  end
+end

data/lib/tshield/matching/filters.rb

@@ -8,6 +8,14 @@ module TShield
         result = filter_stubs(stubs[@options[:session]] || {})
         return result if result
 
+        find_in_secondary_sessions(stubs, @options[:secondary_sessions] || [])
+      end
+
+      def find_in_secondary_sessions(stubs, sessions)
+        sessions.each do |session|
+          result = filter_stubs(stubs[session] || {})
+          return result if result
+        end
         filter_stubs(stubs[DEFAULT_SESSION] || {}) unless @options[:session] == DEFAULT_SESSION
       end
 

data/lib/tshield/request.rb

@@ -13,8 +13,8 @@ module TShield
 
     protected
 
-    def session_destiny(request_path)
-      session = @options[:session]
+    def session_destiny(request_path, current_session = nil)
+      session = current_session || @options[:session]
       return request_path unless session
 
       request_path = File.join(request_path, session)
@@ -22,19 +22,19 @@ module TShield
       request_path
     end
 
-    def content_destiny
-      "#{destiny}.content"
+    def content_destiny(current_session = nil)
+      "#{destiny(current_session)}.content"
     end
 
-    def headers_destiny
-      "#{destiny}.json"
+    def headers_destiny(current_session = nil)
+      "#{destiny(current_session)}.json"
     end
 
-    def destiny
+    def destiny(current_session = nil)
       request_path = File.join('requests')
       Dir.mkdir(request_path) unless File.exist?(request_path)
 
-      request_path = session_destiny(request_path)
+      request_path = session_destiny(request_path, current_session)
 
       name_path = File.join(request_path, name)
       Dir.mkdir(name_path) unless File.exist?(name_path)

data/lib/tshield/request_vcr.rb

@@ -2,11 +2,11 @@
 
 require 'httparty'
 require 'json'
-require 'byebug'
 
 require 'digest/sha1'
 
 require 'tshield/configuration'
+require 'tshield/logger'
 require 'tshield/options'
 require 'tshield/request'
 require 'tshield/response'
@@ -14,6 +14,8 @@ require 'tshield/response'
 module TShield
   # Module to write and read saved responses
   class RequestVCR < TShield::Request
+    attr_reader :vcr_response
+
     def initialize(path, options = {})
       super()
       @path = path
@@ -35,27 +37,31 @@ module TShield
         _method, @url, @options = filter.new.filter(method, @url, @options)
       end
 
-      if exists
-        response.original = false
-        resp = response
+      in_session = find_in_sessions
+      if in_session
+        # TODO: create concept of global session in vcr
+        in_session = nil if in_session == 'global'
+        @vcr_response = response(in_session)
+        @vcr_response.original = false
       else
+        TShield.logger.info("calling original service for request with options #{@options}")
         raw = HTTParty.send(method.to_s, @url, @options)
 
         original_response = save(raw)
         original_response.original = true
-        resp = original_response
+        @vcr_response = original_response
       end
 
       configuration.get_after_filters(domain).each do |filter|
-        resp = filter.new.filter(resp)
+        @vcr_response = filter.new.filter(@vcr_response)
       end
-      resp
     end
 
-    def response
-      @response ||= TShield::Response.new(saved_content['body'],
-                                          saved_content['headers'] || [],
-                                          saved_content['status'] || 200)
+    def response(session)
+      response_content = saved_content(session)
+      TShield::Response.new(response_content['body'],
+                            response_content['headers'] || [],
+                            response_content['status'] || 200)
     end
 
     private
@@ -89,20 +95,27 @@ module TShield
       TShield::Response.new(raw_response.body, headers, raw_response.code)
     end
 
-    def saved_content
-      return @saved_content if @saved_content
-
-      @saved_content = JSON.parse(File.open(headers_destiny).read)
-      @saved_content['body'] = File.open(content_destiny).read unless @saved_content['body']
-      @saved_content
+    def saved_content(session)
+      content = JSON.parse(File.open(headers_destiny(session)).read)
+      content['body'] = File.open(content_destiny(session)).read unless content['body']
+      content
     end
 
-    def file_exists
-      File.exist?(content_destiny)
+    def file_exists(session)
+      File.exist?(content_destiny(session))
     end
 
-    def exists
-      file_exists && configuration.cache_request?(domain)
+    def find_in_sessions
+      in_session = nil
+
+      ([@options[:session]] + (@options[:secondary_sessions] || [])).each do |session|
+        if file_exists(session) && configuration.cache_request?(domain)
+          in_session = (session || 'global')
+          break
+        end
+        TShield.logger.info("saved response not found in #{session}")
+      end
+      in_session
     end
 
     def key

data/lib/tshield/server.rb

@@ -57,7 +57,6 @@ module TShield
 
     def self.run!
       register_resources
-      require 'byebug'
       super
     end
   end

data/lib/tshield/sessions.rb

@@ -1,7 +1,8 @@
 # frozen_string_literal: true
 
-require 'byebug'
+require 'tshield/logger'
 require 'tshield/counter'
+require 'tshield/errors'
 
 module TShield
   # Manage sessions
@@ -9,17 +10,34 @@ module TShield
   # Start and stop session for ip
   module Sessions
     def self.start(ip, name)
-      sessions[normalize_ip(ip)] = { name: name, counter: TShield::Counter.new }
+      TShield.logger.info("starting session #{name} for ip #{normalize_ip(ip)}")
+      sessions[normalize_ip(ip)] = {
+        name: name,
+        counter: TShield::Counter.new,
+        secondary_sessions: []
+      }
     end
 
     def self.stop(ip)
+      TShield.logger.info("stoping session for ip #{normalize_ip(ip)}")
       sessions[normalize_ip(ip)] = nil
     end
 
     def self.current(ip)
+      TShield.logger.info("fetching session for ip #{normalize_ip(ip)}")
       sessions[normalize_ip(ip)]
     end
 
+    def self.append(ip, name)
+      TShield.logger.info("appeding session #{name} for ip #{normalize_ip(ip)}")
+
+      current_session = sessions[normalize_ip(ip)]
+      raise AppendSessionWithoutMainSessionError, "not found main session for #{ip}" unless current_session
+
+      current_session[:secondary_sessions] << name
+      current_session
+    end
+
     def self.sessions
       @sessions ||= {}
     end

data/lib/tshield/version.rb

@@ -5,7 +5,7 @@ module TShield
   class Version
     MAJOR = 0
     MINOR = 11
-    PATCH = 15
+    PATCH = 20
     PRE = 0
 
     class << self

data/spec/spec_helper.rb

@@ -15,9 +15,4 @@ require 'webmock/rspec'
 require 'tshield/extensions/string_extensions'
 
 RSpec.configure do |config|
-  config.before(:each) do
-    allow(File).to receive(:join).and_return(
-      'spec/tshield/fixtures/config/tshield.yml'
-    )
-  end
 end

data/spec/tshield/configuration_spec.rb

@@ -35,6 +35,12 @@ describe TShield::Configuration do
         )
       end
 
+      context 'on grpc configuration' do
+        it 'recover server port' do
+          expect(@configuration.grpc['port']).to(eq(5678))
+        end
+      end
+
       context 'on load filters' do
         it 'recover filters for a domain' do
           expect(@configuration.get_filters('example.org')).to eq([ExampleFilter])
@@ -72,4 +78,17 @@ describe TShield::Configuration do
       expect { TShield::Configuration.singleton }.to raise_error RuntimeError
     end
   end
+
+  context 'on config exists without grpc entry' do
+    before :each do
+      options_instance = double
+      allow(options_instance).to receive(:configuration_file)
+        .and_return('spec/tshield/fixtures/config/tshield-without-grpc.yml')
+      allow(TShield::Options).to receive(:instance).and_return(options_instance)
+      @configuration = TShield::Configuration.singleton
+    end
+    it 'should set default value for port' do
+      expect(@configuration.grpc).to eql('port' => 5678, 'proto_dir' => 'proto', 'services' => {})
+    end
+  end
 end

data/spec/tshield/fixtures/config/tshield-without-grpc.yml

@@ -0,0 +1,17 @@
+---
+request:
+  timeout: 0
+domains:
+  'example.org':
+    name: 'example.org'
+    filters:
+      - 'ExampleFilter'
+    paths:
+      - '/api/one'
+      - '/api/two'
+    skip_query_params:
+      - 'a'
+  'example.com':
+    name: 'example.com'
+    paths:
+      - '/api/three'

data/spec/tshield/fixtures/config/tshield.yml

@@ -1,4 +1,6 @@
 ---
+grpc:
+  port: 5678
 request:
   timeout: 0
 domains:

data/spec/tshield/fixtures/matching/example.json

@@ -99,5 +99,17 @@
         "body": "body content in session"
       }
     }]
+  },
+  {
+    "session": "second-session",
+    "stubs": [{
+      "method": "GET",
+      "path": "/matching/second-example",
+      "response": {
+        "status": 200,
+        "headers": {},
+        "body": "body content in second-session"
+      }
+    }]
   }
 ]

data/spec/tshield/fixtures/proto/test_services_pb.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module TestServices
+  class Service
+    def self.rpc_descs
+      { 'ServiceMethod' => {} }
+    end
+  end
+
+  class Stub
+    def initialize(attributes, options = {}); end
+  end
+end

data/spec/tshield/grpc_spec.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+require 'tshield/grpc'
+
+describe TShield::Grpc do
+  context 'on load services' do
+    before :each do
+      lib_dir = File.join(__dir__, 'fixtures/proto')
+      $LOAD_PATH.unshift(lib_dir) unless $LOAD_PATH.include?(lib_dir)
+
+      @services = {
+        'test_services_pb' => { 'module' => 'TestServices', 'hostname' => '0.0.0.0:5678' }
+      }
+    end
+
+    it 'should implement a service from options' do
+      implementation = TShield::Grpc.load_services(@services).first
+      instance = implementation.new
+      expect(instance.respond_to?(:service_method)).to be_truthy
+    end
+  end
+end

data/spec/tshield/request_matching_spec.rb

@@ -235,6 +235,18 @@ describe TShield::RequestMatching do
           expect(@response.headers).to eql({})
           expect(@response.status).to eql(201)
         end
+        context 'with secondary session' do
+          it 'should return response object from session settings' do
+            @request_matching = TShield::RequestMatching.new('/matching/second-example',
+                                                             method: 'GET',
+                                                             session: 'a-session',
+                                                             secondary_sessions: ['second-session'])
+            @response = @request_matching.match_request
+            expect(@response.body).to eql('body content in second-session')
+            expect(@response.headers).to eql({})
+            expect(@response.status).to eql(200)
+          end
+        end
       end
     end
   end

data/spec/tshield/sessions_spec.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require 'tshield/sessions'
+require 'spec_helper'
+
+describe TShield::Sessions do
+  context 'on append session' do
+    it 'should raise error if not has a main session' do
+      expect { TShield::Sessions.append 'ip', 'secondary-session' }
+        .to raise_error(AppendSessionWithoutMainSessionError)
+    end
+    it 'should append if has main session' do
+      TShield::Sessions.start 'ip', 'main-session'
+      result = TShield::Sessions.append 'ip', 'secondary-session'
+      expect(result[:secondary_sessions]).to include('secondary-session')
+    end
+  end
+end

data/tshield.gemspec

@@ -25,6 +25,8 @@ Gem::Specification.new do |s|
   s.required_ruby_version = '>= 2.3'
 
   s.add_dependency('byebug', '~> 11.0', '>= 11.0.1')
+  s.add_dependency('grpc', '~> 1.28', '>= 1.28.0')
+  s.add_dependency('grpc-tools', '~> 1.28', '>= 1.28.0')
   s.add_dependency('httparty', '~> 0.14', '>= 0.14.0')
   s.add_dependency('json', '~> 2.0', '>= 2.0')
   s.add_dependency('puma', '~> 4.3', '>= 4.3.3')

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tshield
 version: !ruby/object:Gem::Version
-  version: 0.11.15.0
+  version: 0.11.20.0
 platform: ruby
 authors:
 - Diego Rubin
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-07-08 00:00:00.000000000 Z
+date: 2020-07-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: byebug
@@ -31,6 +31,46 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 11.0.1
+- !ruby/object:Gem::Dependency
+  name: grpc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.28.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.28'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.28.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.28'
+- !ruby/object:Gem::Dependency
+  name: grpc-tools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.28.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.28'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.28.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.28'
 - !ruby/object:Gem::Dependency
   name: httparty
   requirement: !ruby/object:Gem::Requirement
@@ -376,6 +416,7 @@ files:
 - README.md
 - Rakefile
 - bin/tshield
+- config/tshield.yml
 - lib/tshield.rb
 - lib/tshield/after_filter.rb
 - lib/tshield/before_filter.rb
@@ -385,7 +426,10 @@ files:
 - lib/tshield/controllers/requests.rb
 - lib/tshield/controllers/sessions.rb
 - lib/tshield/counter.rb
+- lib/tshield/errors.rb
 - lib/tshield/extensions/string_extensions.rb
+- lib/tshield/grpc.rb
+- lib/tshield/grpc/vcr.rb
 - lib/tshield/logger.rb
 - lib/tshield/matching/filters.rb
 - lib/tshield/options.rb
@@ -395,18 +439,21 @@ files:
 - lib/tshield/response.rb
 - lib/tshield/server.rb
 - lib/tshield/sessions.rb
-- lib/tshield/simple_tcp_server.rb
 - lib/tshield/version.rb
 - spec/spec_helper.rb
 - spec/tshield/after_filter_spec.rb
 - spec/tshield/configuration_spec.rb
 - spec/tshield/controllers/requests_spec.rb
+- spec/tshield/fixtures/config/tshield-without-grpc.yml
 - spec/tshield/fixtures/config/tshield.yml
 - spec/tshield/fixtures/filters/example_filter.rb
 - spec/tshield/fixtures/matching/example.json
+- spec/tshield/fixtures/proto/test_services_pb.rb
+- spec/tshield/grpc_spec.rb
 - spec/tshield/options_spec.rb
 - spec/tshield/request_matching_spec.rb
 - spec/tshield/request_vcr_spec.rb
+- spec/tshield/sessions_spec.rb
 - tshield.gemspec
 homepage: https://github.com/diegorubin/tshield
 licenses:
@@ -434,11 +481,15 @@ summary: Proxy for mocks API responses
 test_files:
 - spec/spec_helper.rb
 - spec/tshield/request_matching_spec.rb
+- spec/tshield/grpc_spec.rb
 - spec/tshield/configuration_spec.rb
+- spec/tshield/sessions_spec.rb
 - spec/tshield/request_vcr_spec.rb
 - spec/tshield/controllers/requests_spec.rb
 - spec/tshield/options_spec.rb
 - spec/tshield/after_filter_spec.rb
 - spec/tshield/fixtures/matching/example.json
+- spec/tshield/fixtures/proto/test_services_pb.rb
 - spec/tshield/fixtures/filters/example_filter.rb
 - spec/tshield/fixtures/config/tshield.yml
+- spec/tshield/fixtures/config/tshield-without-grpc.yml

data/lib/tshield/simple_tcp_server.rb

@@ -1,28 +0,0 @@
-# frozen_string_literal: true
-
-require 'socket'
-
-module TShield
-  class SimpleTCPServer
-    def initialize
-      @running = true
-    end
-
-    def on_connect(_client)
-      raise 'should implement method on_connect'
-    end
-
-    def close
-      @running = false
-    end
-
-    def listen(port)
-      puts "listening #{port}"
-      @server = TCPServer.new(port)
-      while @running
-        client = @server.accept
-        on_connect(client)
-      end
-    end
-  end
-end