imitate 0.1.1 → 0.1.2

data/VERSION

@@ -1 +1 @@
-0.1.1
+0.1.2

data/imitate.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = "imitate"
-  s.version = "0.1.1"
+  s.version = "0.1.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Petr Janda"]

data/lib/imitate.rb

@@ -4,32 +4,65 @@ module Imitate
 
   class << self
 
+    #
+    # Create class with a given name. If name contains any package, which
+    # doesn't exist yet, its created along the way.
+    #
+    # Example:
+    # mock_class('ActiveRecord::Base')
+    # mock_class('Foo') do
+    #   def bar
+    #     'bar'
+    #   end
+    # end
+    #
+    # Foo.new.bar #= 'bar'
+    #
+    #
+    # @param [String] Name of the class to be created.
+    # @param [Proc] Optional block to be instance evaluated on new class.
+    #
     def mock_class(name, &block)
-      tokens = name.split('::')
-      parent = Object
-
-      while tokens.length > 1
-        parent = create_const(Module, parent, tokens.shift)
-      end
-
-      create_const(Class, parent, tokens.shift, &block)
+      process_name(name) {|parent, token, tokens|
+        if tokens.length > 0
+          create_const(Module, parent, token)
+        else
+          create_const(Class, parent, token, &block)
+        end
+      }
     end
 
+    #
+    # Unmock previously mocked class. It doesn't affect classes created other
+    # way then using Imitate.
+    #
+    # @param [String] Name of the class to be removed.
+    #
     def unmock(name)
-      tokens = name.split('::')
-      parent = Object
-
-      while token = tokens.shift
-        parent = self.send(was_mocked?(parent, token) ? :remove_const : :get_const, parent, token)
-      end
+      process_name(name) {|parent, token|
+        self.send(was_mocked?(parent, token) ? :remove_const : :get_const, parent, token)
+      }
     end
 
+    #
+    # Unmock all previously mocked classes. It doesn't affect classes created other
+    # way then using Imitate.
+    #
     def unmock_all
       MOCKED.each {|mock| remove_const(mock[:parent], mock[:name])}
     end
 
     private 
 
+    def process_name(name)
+      tokens = name.split('::')
+      parent = Object
+
+      while token = tokens.shift
+        parent = yield(parent, token, tokens)
+      end
+    end
+
     def was_mocked?(parent, name)
       MOCKED.index({:parent => parent, :name => name})
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: imitate
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-05-03 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &2157117920 !ruby/object:Gem::Requirement
+  requirement: &2159204420 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,10 +21,10 @@ dependencies:
         version: 2.8.0
   type: :development
   prerelease: false
-  version_requirements: *2157117920
+  version_requirements: *2159204420
 - !ruby/object:Gem::Dependency
   name: rdoc
-  requirement: &2157117320 !ruby/object:Gem::Requirement
+  requirement: &2159203920 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -32,10 +32,10 @@ dependencies:
         version: '3.12'
   type: :development
   prerelease: false
-  version_requirements: *2157117320
+  version_requirements: *2159203920
 - !ruby/object:Gem::Dependency
   name: bundler
-  requirement: &2157116720 !ruby/object:Gem::Requirement
+  requirement: &2159203400 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -43,10 +43,10 @@ dependencies:
         version: 1.0.0
   type: :development
   prerelease: false
-  version_requirements: *2157116720
+  version_requirements: *2159203400
 - !ruby/object:Gem::Dependency
   name: jeweler
-  requirement: &2157116120 !ruby/object:Gem::Requirement
+  requirement: &2159202880 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -54,7 +54,7 @@ dependencies:
         version: 1.8.3
   type: :development
   prerelease: false
-  version_requirements: *2157116120
+  version_requirements: *2159202880
 description: Simple class mock
 email: petrjanda@me.com
 executables: []
@@ -90,7 +90,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -1839012386666507265
+      hash: 2996185596802046611
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: