hassox-machinist 0.1.7

data/lib/machinist.rb

@@ -0,0 +1,88 @@
+require 'active_support'
+require 'sham'
+require File.join(File.dirname(__FILE__), "adapters", "abstract")
+ 
+module Machinist
+  # :api: private
+  def self.adapters
+    @adapters ||= []
+    @_adapter_size ||= @adapters.size
+    unless @adapters.size == @_adapter_size
+      @adapters = @adapters.uniq.sort_by{|a| a.priority}.reverse
+      @_adapter_size = @adapters.size
+    end
+    @adapters
+  end
+  
+  # :api: private
+  def self.add_adapter(adapter)
+    adapters << adapter
+  end
+  
+  # :api: private
+  def self.adapter_for(klass)
+    adapters.detect{|a| a.use_for_class?(klass)} || raise("No Adapter found for #{klass}")
+  end
+  
+  module Extensions
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+    
+    module ClassMethods
+      def blueprint(&blueprint)
+        @blueprint = blueprint
+        @machinist_adapter = Machinist.adapter_for(self)
+      end
+  
+      def make(attributes = {}, &block)
+        obj = make_unsaved(attributes.merge(:_save_ => true), &block)
+        @machinist_adapter.save(obj)
+      end
+    
+      def make_unsaved(attributes = {})
+        raise "No blueprint for class #{self}" if @blueprint.nil?
+        lathe = Lathe.new(self, attributes)
+        lathe.instance_eval(&@blueprint)
+        returning(lathe.object) do |object|
+          yield object if block_given?
+        end
+      end
+    end
+  end
+  
+  class Lathe
+    def initialize(klass, attributes = {})
+      @save = !!attributes.delete(:_save_)
+      @object = klass.new
+      attributes.each {|key, value| @object.send("#{key}=", value) }
+      @assigned_attributes = attributes.keys.map(&:to_sym)
+    end
+
+    # Undef a couple of methods that are common ActiveRecord attributes.
+    # (Both of these are deprecated in Ruby 1.8 anyway.)
+    undef_method :id
+    undef_method :type
+    
+    attr_reader :object
+
+    def method_missing(symbol, *args, &block)
+      if @assigned_attributes.include?(symbol)
+        @object.send(symbol)
+      else
+        value = if block
+          block.call
+        elsif args.first.is_a?(Hash) || args.empty?
+          meth = @save ? :make : :make_unsaved
+          symbol.to_s.camelize.constantize.send(meth, (args.first || {}))
+        else
+          args.first
+        end
+        @object.send("#{symbol}=", value)
+        @assigned_attributes << symbol
+      end
+    end
+  end
+end
+
+require File.join(File.dirname(__FILE__), 'adapters/active_record') if defined?(ActiveRecord)

data/lib/sham.rb

@@ -0,0 +1,75 @@
+require 'active_support'
+
+class Sham
+  @@shams = {}
+  
+  # Over-ride module's built-in name method, so we can re-use it for
+  # generating names. This is a bit of a no-no, but we get away with
+  # it in this context.
+  def self.name(*args, &block)
+    method_missing(:name, *args, &block)
+  end
+  
+  def self.method_missing(symbol, *args, &block)
+    if block_given?
+      @@shams[symbol] = Sham.new(symbol, args.pop || {}, &block)
+    else
+      sham = @@shams[symbol]
+      raise "No sham defined for #{symbol}" if sham.nil?
+      sham.fetch_value
+    end
+  end
+
+  def self.reset
+    @@shams.values.each(&:reset)
+  end
+  
+  def self.define(&block)
+    definer = Object.new
+    class << definer
+      def method_missing(*args, &block)
+        Sham.send(*args, &block)
+      end
+    end
+    definer.instance_eval(&block)
+  end
+  
+  def initialize(name, options = {}, &block)
+    @name      = name
+    @generator = block
+    @offset    = 0
+    @unique    = options.has_key?(:unique) ? options[:unique] : true
+    generate_values(12)
+  end
+  
+  def reset
+    @offset = 0
+  end
+  
+  def fetch_value
+    # Generate more values if we need them.
+    if @offset >= @values.length
+      generate_values(2 * @values.length)
+      raise "Can't generate more unique values for Sham.#{@name}" if @offset >= @values.length
+    end
+    returning @values[@offset] do
+      @offset += 1
+    end
+  end
+    
+private
+  
+  def generate_values(count)
+    @values = seeded { (1..count).map(&@generator) }
+    @values.uniq! if @unique
+  end
+  
+  def seeded
+    begin
+      srand(1)
+      yield
+    ensure
+      srand
+    end
+  end
+end

metadata

@@ -0,0 +1,54 @@
+--- !ruby/object:Gem::Specification 
+name: hassox-machinist
+version: !ruby/object:Gem::Version 
+  version: 0.1.7
+platform: ruby
+authors: 
+- Pete Yandell
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-12-22 00:00:00 -08:00
+default_executable: 
+dependencies: []
+
+description: 
+email: pete@nothat.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/machinist.rb
+- lib/sham.rb
+has_rdoc: false
+homepage: http://github.com/notahat/machinist
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Fixtures aren't fun. Machinist is.
+test_files: []
+