cross-stub.rb in cross-stub-0.2.0

- old
+ new

@@ -1,49 +1,83 @@
 require 'rubygems'
-require 'parse_tree'
+require 'base64'
 require 'ruby2ruby'
-require 'cross-stub/stub_helpers'
-require 'cross-stub/setup_helpers'
-require 'cross-stub/cache_helpers'
-require 'cross-stub/pseudo_class'
+require 'ruby_parser'
+require 'forwardable'
+require 'cross-stub/cache'
+require 'cross-stub/stubber'
+require 'cross-stub/arguments'
+require 'cross-stub/stores'
 
 module CrossStub
 
   class Error < Exception ; end
   class CannotStubInstanceError < Error ; end
+  class ModuleCannotBeInstantiatedError < Error ; end
 
   class << self
 
-    include CacheHelpers
-    include SetupHelpers
-    include StubHelpers
+    extend Forwardable
+    def_delegator :'CrossStub::Cache', :setup
 
-    attr_reader :options
-
-    def setup(opts)
-      @options = opts
-      setup_for_current_process
+    def refresh(opts)
+      Cache.refresh(opts)
+      [[:previous, :unapply], [:current, :apply]].each do |(mode, method)|
+        Cache.get(mode).map do |cache_key, stubs|
+          type, thing = stubbable(cache_key)
+          Stubber.send(method, type, thing, stubs)
+        end
+      end
     end
 
     def clear
-      clear_stubs_for_current_process
+      Cache.get.map do |cache_key, stubs|
+        type, thing = stubbable(cache_key)
+        Stubber.unapply(type, thing, stubs)
+      end
+      Cache.clear
     end
 
-    def apply(*args, &blk)
-      apply_stubs_for_current_process(*args, &blk)
+    def apply(type, thing, cache_key, args, &block)
+      Cache.set(Cache.get.merge(
+        cache_key => Stubber.apply(type, thing, Arguments.parse(args, &block))
+      ))
     end
 
-    def refresh(opts)
-      @options = opts
-      apply_or_unapply_stubs_for_other_process
+    def stubbable(str)
+      [
+        str.end_with?(suffix = '#instance') ? :instance : :class,
+        klassify(str.sub(suffix,''))
+      ]
     end
 
+    def klassify(str)
+      str.split('::').inject(Object){|klass, const| klass.const_get(const) }
+    end
+
   end
 
   module ClassMethods
-    def xstub(*args, &blk)
-      CrossStub.apply(self, args, &blk)
+    def xstub(*args, &block)
+      if args[-1].is_a?(::Hash) && args[-1][:instance]
+        raise ModuleCannotBeInstantiatedError if self.class == Module
+        CrossStub.apply(
+          :instance,                      # stubbing for instance
+          self,                           # the class to action on
+          '%s#instance' % self,           # cache key (storing of stubbing info for other process)
+          args.size>1 ? args[0..-2] : [], # stubbing arguments
+          &block                          # any other more complex stubbing arguments
+        )
+      else
+        CrossStub.apply(
+          :class,    # stubbing for class/module
+          self,      # the class to action on
+          "#{self}", # cache key (storing of stubbing info for other process)
+          args,      # stubbing arguments
+          &block     # any other more complex stubbing arguments
+        )
+      end
     end
   end
 
   module InstanceMethods
     def xstub(*args)
@@ -51,8 +85,13 @@
     end
   end
 
 end
 
-Object.send(:extend, CrossStub::ClassMethods)
-Object.send(:include, CrossStub::InstanceMethods)
-Module.send(:include, CrossStub::ClassMethods)
+Object.class_eval do
+  extend CrossStub::ClassMethods
+  include CrossStub::InstanceMethods
+end
+
+Module.class_eval do
+  include CrossStub::ClassMethods
+end