gcr 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 654092852f61d95659958cc4b7bdc57863d92474
+  data.tar.gz: 1475ecc16fb24ca8c8b69e63ff5ff63fcfc3cbea
+SHA512:
+  metadata.gz: 14a2b562b9238808de93eea48fbb2ced1cf8a5ba0d52b1c8bdb009f1bcf38179c5387d29a57be9c1ab4efca4efd99e99ed4f60b7c86e3909d681703fce8f5f95
+  data.tar.gz: 8b8a4c32de38ba8dffa4801db17076617e8763080825e4b7e4942531490adbcde99f9a69f4fe99389836ed502969f32d3992ce4883f7d9dd09d02ac2fbc6bdb8

data/lib/gcr.rb

@@ -0,0 +1,85 @@
+module GCR
+  Error = Class.new(StandardError)
+  ConfigError = Class.new(Error)
+  RunningError = Class.new(Error)
+  NoRecording = Class.new(Error)
+
+  # Specify where GCR should store cassettes.
+  #
+  # path - The String path to a directory.
+  #
+  # Returns nothing.
+  def cassette_dir=(path)
+    raise RunningError, "cannot configure GCR within #with_cassette block" if @running
+    @cassette_dir = path
+  end
+
+  # Where GCR stores cassettes.
+  #
+  # Returns a String path to a directory. Raises ConfigError if not configured.
+  def cassette_dir
+    @cassette_dir || (raise ConfigError, "no cassette dir configured")
+  end
+
+  # Specify the stub to intercept calls to.
+  #
+  # stub - A GRPC::ClientStub instance.
+  #
+  # Returns nothing.
+  def stub=(stub)
+    raise RunningError, "cannot configure GCR within #with_cassette block" if @running
+    @stub = stub
+  end
+
+  # The stub that is being mocked.
+  #
+  # Returns a A GRPC::ClientStub instance. Raises ConfigError if not configured.
+  def stub
+    @stub || (raise ConfigError, "no cassette dir configured")
+  end
+
+  def cassette
+    @cassette
+  end
+
+  # If a cassette with the given name exists, play that cassette for the
+  # provided block. Otherwise, record a cassette with the provided block.
+  #
+  # Returns nothing.
+  def with_cassette(name, &blk)
+    @cassette = Cassette.new(name)
+    if @cassette.exist?
+      @cassette.play(&blk)
+    else
+      @cassette.record(&blk)
+    end
+  ensure
+    @cassette = nil
+  end
+
+  def serialize_request(route, req, marshal, unmarshal, metadata)
+    JSON.dump(
+      "route" => route,
+      "req"   => Base64.strict_encode64(req.to_proto)
+    )
+  end
+
+  def serialize_response(resp)
+    JSON.dump(
+      "type"  => resp.class.name,
+      "buf" => Base64.strict_encode64(resp.to_proto)
+    )
+  end
+
+  def deserialize_response(str)
+    hsh = JSON.parse(str)
+    klass = Google::Protobuf::DescriptorPool.generated_pool.lookup(hsh["type"]).msgclass
+    klass.decode(Base64.strict_decode64(hsh["buf"]))
+  end
+
+  extend self
+end
+
+require "json"
+require "base64"
+require "gcr/cassette"

data/lib/gcr/cassette.rb

@@ -0,0 +1,105 @@
+class GCR::Cassette
+  attr_reader :data
+
+  # Delete all recorded cassettes.
+  #
+  # Returns nothing.
+  def self.delete_all
+    Dir[File.join(GCR.cassette_dir, "*.json")].each do |path|
+      File.unlink(path)
+    end
+  end
+
+  # Initialize a new cassette.
+  #
+  # name - The String name of the recording, from which the path is derived.
+  #
+  # Returns nothing.
+  def initialize(name)
+    @path = File.join(GCR.cassette_dir, "#{name}.json")
+    @data = {}
+  end
+
+  # Does this cassette exist?
+  #
+  # Returns boolean.
+  def exist?
+    File.exist?(@path)
+  end
+
+  # Load this cassette.
+  #
+  # Returns nothing.
+  def load
+    @data = JSON.parse(File.read(@path))
+  end
+
+  # Persist this cassette.
+  #
+  # Returns nothing.
+  def save
+    File.open(@path, "w") do |f|
+      f.write(JSON.pretty_generate(data))
+    end
+  end
+
+  # Record all GRPC calls made while calling the provided block.
+  #
+  # Returns nothing.
+  def record(&blk)
+    install_record_hook
+    blk.call
+    save
+  ensure
+    remove_hook
+  end
+
+  # Play recorded GRPC responses.
+  #
+  # Returns nothing.
+  def play(&blk)
+    load
+    install_play_hook
+    blk.call
+  ensure
+    remove_hook
+  end
+
+  private
+
+  def install_record_hook
+    GCR.stub.class.class_eval do
+      alias_method :orig_request_response, :request_response
+
+      def request_response(*args)
+        orig_request_response(*args).tap do |resp|
+          key = GCR.serialize_request(*args)
+          GCR.cassette.data[key] ||= []
+          GCR.cassette.data[key] << GCR.serialize_response(resp)
+        end
+      end
+    end
+  end
+
+  def install_play_hook
+    GCR.stub.class.class_eval do
+      alias_method :orig_request_response, :request_response
+
+      def request_response(*args)
+        key = GCR.serialize_request(*args)
+        GCR.cassette.data[key] ||= []
+        if resp = GCR.cassette.data[key].shift
+          GCR.deserialize_response(resp)
+        else
+          raise GCR::NoRecording
+        end
+      end
+    end
+  end
+
+  def remove_hook
+    GCR.stub.class.class_eval do
+      alias_method :request_response, :orig_request_response
+    end
+  end
+end

metadata

@@ -0,0 +1,105 @@
+--- !ruby/object:Gem::Specification
+name: gcr
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- mastahyeti
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-05-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: google-protobuf
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.3'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.3'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.3.0
+- !ruby/object:Gem::Dependency
+  name: grpc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.5
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.5
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.5.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.5.0
+description: 
+email: 
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- "./lib/gcr.rb"
+- "./lib/gcr/cassette.rb"
+homepage: https://github.com/mastahyeti/gcr
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2
+signing_key: 
+specification_version: 4
+summary: GRPC test helpers
+test_files: []