stub_a 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 022265f955db24a8dd7a3bb3465243121379ecd8
+  data.tar.gz: 86be306baf3ec972fdf4163e7ced3920e7bc4407
+SHA512:
+  metadata.gz: aaff64bee34e4ce474b09fe009188c83af14b39128cd44bbaddc55d8055e4e71a7aeb789d097ae45acc8d685f0890571e8603da53178a7223c4a19a0a1a5c639
+  data.tar.gz: d156071d484adaf0b2be902a1eb7903c33c9b9d2fd0271371e4b472aca42387d2a3b5120703c3256da55ba17357fbd64b0c80ea0bc402a77f5af9023fcadc4ea

data/lib/stub_a.rb

@@ -0,0 +1,154 @@
+class StubA
+  def initialize(object)
+    @target_class = object if object.is_a? Class
+    @target_class ||= class << object; self; end
+  end
+
+  [:before,
+   :after,
+  ].each do |hook_type|
+    define_method hook_type, ->(method, &block) {
+      make_hook_points(method)
+
+      method_name = hook_method_name(method, hook_type)
+      hook = block if block
+      hook ||= default_hook(method, hook_type)
+      @target_class.__send__ :define_method, method_name, hook
+      self
+    }
+  end
+
+  def stub(method, &block)
+    raise ArgumentError, "No block" unless block
+    make_hook_points(method)
+
+    method_name = hook_method_name(method, :stub)
+    @target_class.__send__ :define_method, method_name, block
+    self
+  end
+
+  def restore(method=nil, *types)
+    if method.nil?
+      hooks = defined_hooks
+      method = hooks.keys.first if hooks.keys.size == 1
+    end
+    raise "Method name to be restored is required." if method.nil?
+
+    available_types = METHOD_TYPES - [:origin]
+    raise "Type error" if (types - available_types).size > 0
+    types = available_types if types.empty?
+    types.uniq!
+
+    types.each do |type|
+      method_name = hook_method_name(method, type)
+      next unless @target_class.method_defined? method_name
+      @target_class.__send__ :remove_method, method_name
+    end
+
+    hooks = defined_hooks
+    if hooks[method.to_s].size == 1 and
+      hooks[method.to_s].first == 'origin'
+      origin = hook_method_name(method, :origin)
+      @target_class.__send__ :alias_method, method, origin
+      @target_class.__send__ :remove_method, origin
+    end
+  end
+
+  def restore_all
+    defined_hooks.each do |method, types|
+      types.each do |type|
+        name = hook_method_name(method, type)
+        next unless @target_class.method_defined? name
+        @target_class.__send__ :alias_method, method, name if type == 'origin'
+        @target_class.__send__ :remove_method, name
+      end
+    end
+  end
+  
+  private
+  METHOD_TYPES = [
+    :before,
+    :after,
+    :stub,
+    :origin,
+  ]
+  private_constant :METHOD_TYPES
+
+  def hook_method_name(method, type)
+    "===>(stub_a) #{type}-#{method}"
+  end
+
+  def hook_method_names(method)
+    METHOD_TYPES.each_with_object({}) do |t, h|
+      h[t] = hook_method_name(method, t)
+    end
+  end
+
+  def make_hook_points(method_name)
+    methods = hook_method_names(method_name)
+    return if @target_class.method_defined? methods[:origin]
+
+    @target_class.__send__ :alias_method, methods[:origin], method_name
+    @target_class.__send__ :define_method, method_name, ->(*args, &block) {
+      if respond_to? methods[:before]
+        __send__ methods[:before], ARGS[:before].new(method_name, args), &block
+      end
+
+      if respond_to? methods[:stub]
+        origin = __send__ :method, methods[:origin]
+        val = __send__ methods[:stub], ARGS[:stub].new(origin, args), &block
+      else
+        val = __send__ methods[:origin], *args, &block
+      end
+
+      if respond_to? methods[:after]
+        __send__ methods[:after], ARGS[:after].new(method_name, val)
+      end
+
+      val
+    }
+  end
+
+  def defined_hooks
+    re = /\A===>\(stub_a\) (before|after|stub|origin)-(.+)\z/
+    @target_class.instance_methods.each_with_object({}) {|name, h|
+      if name.to_s =~ re
+        h[$2] ||= []
+        h[$2] << $1
+      end
+    }
+  end
+
+  def default_hook(method, type)
+    case type.to_sym
+    when :before
+      default_before_hook(method)
+    when :after
+      default_after_hook(method)
+    end
+  end
+
+  def default_before_hook(method)
+    ->(method, *args, &block) {
+      puts "[#{Time.now}] call `#{method}' (#{self.inspect})"
+      puts "args: #{args.inspect}" if args.size > 0
+    }
+  end
+
+  def default_after_hook(method)
+    before_method = hook_method_name(method, :before)
+    ->(method, val) {
+      unless respond_to? before_method
+        puts "[#{Time.now}] call `#{method}' (#{self.inspect})"
+      end
+      puts "return: #{val.inspect}"
+    }
+  end
+
+  ARGS = {
+    before: Struct.new(:method_name, :args),
+    stub: Struct.new(:method, :args),
+    after: Struct.new(:method_name, :return_value),
+  }
+  private_constant :ARGS
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: stub_a
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- 60ml
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-02-07 00:00:00.000000000 Z
+dependencies: []
+description: a tool to manipulate objects with hooks and stubs
+email: bd.4975.3110@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/stub_a.rb
+homepage: https://github.com/60ml/stub_a
+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.4.5
+signing_key: 
+specification_version: 4
+summary: a tool to manipulate objects with hooks and stubs
+test_files: []