fixturize 0.1.0

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    YTc3ZTM0NzQ4YzlmYjg4ZDlkZjlmMjI2ZGIxNmE3NzU1MTllMDg2OQ==
+  data.tar.gz: !binary |-
+    YTU3NTljYmJlMzI5OTVmYWU2OWIzNzI4YTA0NTRmZmM5MDFhOGE2MQ==
+SHA512:
+  metadata.gz: !binary |-
+    YTU2MzIwODY3MDRiOTY1YzRhNWU5ODJlNGY0NzJkYjM4ODJmNjM2ZjBkZjEw
+    MWVkMjlkMGYyODgzZWE0Yzg1NjRiNjk0MzM0MjkzM2ZjNjMwMGFhMGZjNGIz
+    NzVkYmQyMDA4ZDJkMzVmYTA1ZTU2YmVmZGU0NWYxMmIyMjU5Zjk=
+  data.tar.gz: !binary |-
+    ODBiODIzOTRkOGEyZDc3M2NiYjI2MTZhMWIyMWMyMjc4MjQ1NmU3YzBmZGEz
+    Yzk3YWFkYWUwMjdiNDlmZmQwY2NiYmRhNjM0YjQyMDc2MjAzNGRmODQ4ODU3
+    NDRlZDlhOTVlMDFjN2U0ODFlMmY0YmFkM2VmOWNjODk2MDk1OGY=

data/lib/fixturize.rb

@@ -0,0 +1,200 @@
+require 'mongo'
+require 'yaml'
+require 'set'
+
+class Fixturize
+  METHODS_FOR_INSTRUMENTATION = [
+    :save,
+    :insert,
+    :remove,
+    :update,
+    :drop,
+    :rename,
+  ]
+
+  class << self
+    attr_accessor :database
+    attr_accessor :current_instrumentation
+    attr_writer :database_version
+
+    def database_version
+      @database_version ||= 0
+    end
+
+    def reset_version!
+      @database_version = nil
+    end
+
+    def collections
+      [
+        db_updates_collection_name,
+        db_saved_ivars_collection_name,
+      ]
+    end
+
+    def db_updates_collection_name
+      "mongo_saved_contexts_#{database_version}_"
+    end
+
+    def db_saved_ivars_collection_name
+      "mongo_saved_ivars_#{database_version}_"
+    end
+
+    def saved_contexts_collection
+      if !database
+        raise "Fixturize is not yet setup!  Make sure the database is set!"
+      end
+
+      database.collection(db_updates_collection_name)
+    end
+
+    def saved_ivars_collection
+      if !database
+        raise "Fixturize is not yet setup!  Make sure the database is set!"
+      end
+
+      database.collection(db_saved_ivars_collection_name)
+    end
+
+    def clear_cache!
+      database.collections.each do |c|
+        if c.name == /mongo_saved_/
+          c.drop
+        end
+      end
+    end
+
+    def instrument_database(collection_name, method_name, *args)
+      saved_contexts_collection.insert_aliased_from_mongo_saved_context({
+        :name => current_instrumentation,
+        :collection_name => collection_name.to_s,
+        :method_name => method_name.to_s,
+        :args => YAML.dump(args)
+      })
+    end
+
+    def instrument_ivars(ivars, context)
+      ivars.each do |ivar|
+        obj = context.instance_variable_get(ivar)
+
+        # TODO: Use duck typing?
+        if defined?(MongoMapper) && obj.kind_of?(MongoMapper::Document)
+          saved_ivars_collection.insert({
+            :name => current_instrumentation,
+            :ivar => ivar,
+            :model => obj.class.to_s,
+            :id => obj.id
+          })
+        end
+      end
+    end
+
+    def load_data_from(instrumentation)
+      collection = database.collection(instrumentation['collection_name'])
+      collection.send(instrumentation['method_name'], *YAML.load(instrumentation['args']))
+    end
+
+    def load_ivars_from(instrumentation, target_obj)
+      ivar = instrumentation['ivar']
+      model_str = instrumentation['model']
+      id = instrumentation['id']
+
+      model = Object.const_get(model_str)
+      obj = model.find(id)
+      target_obj.instance_variable_set(ivar, obj)
+    end
+
+    def refresh!(name = nil)
+      if name
+        saved_contexts_collection.remove({ :name => name.to_s })
+      else
+        saved_contexts_collection.drop()
+      end
+    end
+
+    def fixturize(name = nil, &block)
+      if !name && block.respond_to?(:source_location)
+        # is this portable?
+        name = block.source_location.join(":")
+      end
+
+      if !name
+        raise "A name must be given to fixturize"
+      end
+
+      name = name.to_s
+      self.current_instrumentation = name
+      db_instrumentations = saved_contexts_collection.find({ :name => name }).to_a
+      ivar_instrumentations = saved_ivars_collection.find({ :name => name }).to_a
+
+      if db_instrumentations.any? || ivar_instrumentations.any?
+        uninstall!
+
+        db_instrumentations.each do |instrumentation|
+          load_data_from(instrumentation)
+        end
+
+        ivar_instrumentations.each do |instrumentation|
+          load_ivars_from(instrumentation, caller_of_block(block))
+        end
+      else
+        safe_install(&block)
+      end
+    end
+
+  private
+
+    def caller_of_block(block)
+      block.binding.eval("self")
+    end
+
+    def safe_install(&block)
+      install!(&block)
+    ensure
+      self.current_instrumentation = nil
+      uninstall!
+    end
+
+    def install!(&block)
+      METHODS_FOR_INSTRUMENTATION.each do |method_name|
+        Mongo::Collection.class_eval <<-HERE, __FILE__, __LINE__
+          unless instance_methods.include?(:#{method_name}_aliased_from_mongo_saved_context)
+            alias_method :#{method_name}_aliased_from_mongo_saved_context, :#{method_name}
+
+            def #{method_name}(*args, &block)
+              Fixturize.instrument_database(@name, :#{method_name}, *args, &block)
+              #{method_name}_aliased_from_mongo_saved_context(*args, &block)
+            end
+          end
+        HERE
+      end
+
+      block_caller = caller_of_block(block)
+      ivars_before_block = block_caller.instance_variables
+
+      yield.tap do
+        new_ivars = (Set.new(block_caller.instance_variables) - Set.new(ivars_before_block)).to_a
+        instrument_ivars(new_ivars, block_caller)
+      end
+    end
+
+    def uninstall!
+      METHODS_FOR_INSTRUMENTATION.each do |method_name|
+        Mongo::Collection.class_eval <<-HERE, __FILE__, __LINE__
+          if instance_methods.include?(:#{method_name}_aliased_from_mongo_saved_context)
+            alias_method :#{method_name}, :#{method_name}_aliased_from_mongo_saved_context
+            remove_method :#{method_name}_aliased_from_mongo_saved_context
+          end
+        HERE
+      end
+    end
+  end
+end
+
+def fixturize(*args, &block)
+  Fixturize.fixturize(*args, &block)
+end
+
+if defined?(MongoMapper && MongoMapper.database)
+  Fixturize.database = MongoMapper.database
+end

metadata

@@ -0,0 +1,45 @@
+--- !ruby/object:Gem::Specification
+name: fixturize
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Scott Taylor
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-06-06 00:00:00.000000000 Z
+dependencies: []
+description: fixturize your mongo(mapper) tests inline by caching blocks of created
+  objects
+email: scott@railsnewbie.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/fixturize.rb
+homepage: http://github.com/smtlaissezfaire/fixturize
+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.2.2
+signing_key: 
+specification_version: 4
+summary: fixturize your mongo tests inline
+test_files: []