mongoid-bolt 1.0.0

data/README.md

@@ -0,0 +1,45 @@
+NAME
+----
+  mongoid-bolt
+
+INSTALL
+-------
+  gem install mongoid-bolt
+
+SYNOPSIS
+--------
+
+````ruby
+
+  require 'mongoid-bolt'
+
+  Bolt.for(:shared_resource) do
+    ioslated!
+  end
+
+
+  class A
+    include Mongoid::Document
+    include Mongoid::Bolt
+  end
+
+  a = A.new
+
+  a.lock!
+
+  a.unlock!
+
+  a.lock do
+    isolated!
+  end
+
+
+
+````
+
+DESCRIPTION
+-----------
+
+mongoid-bolt is a concrete lock implementation and mixin.
+
+it is process safe and atomic in a mongoid cluster.

data/Rakefile

@@ -0,0 +1,446 @@
+This.name =
+  "Mongoid::Bolt"
+
+This.synopsis =
+  "a mongoid 3/moped compatible lock implementation and mixin"
+
+This.rubyforge_project = 'codeforpeople'
+This.author            = "Ara T. Howard"
+This.email             = "ara.t.howard@gmail.com"
+This.homepage          = "https://github.com/ahoward/#{ This.lib }"
+
+This.setup!
+
+
+
+task :default do
+  puts((Rake::Task.tasks.map{|task| task.name.gsub(/::/,':')} - ['default']).sort)
+end
+
+task :test do
+  This.run_tests!
+end
+
+namespace :test do
+  task(:unit){ This.run_tests!(:unit) }
+  task(:functional){ This.run_tests!(:functional) }
+  task(:integration){ This.run_tests!(:integration) }
+end
+
+def This.run_tests!(which = nil)
+  which ||= '**'
+  test_dir = File.join(This.dir, "test")
+  test_glob ||= File.join(test_dir, "#{ which }/**_test.rb")
+  test_rbs = Dir.glob(test_glob).sort
+        
+  div = ('=' * 119)
+  line = ('-' * 119)
+
+  test_rbs.each_with_index do |test_rb, index|
+    testno = index + 1
+    command = "#{ File.basename(This.ruby) } -I ./lib -I ./test/lib #{ test_rb }"
+
+    puts
+    This.say(div, :color => :cyan, :bold => true)
+    This.say("@#{ testno } => ", :bold => true, :method => :print)
+    This.say(command, :color => :cyan, :bold => true)
+    This.say(line, :color => :cyan, :bold => true)
+
+    system(command)
+
+    This.say(line, :color => :cyan, :bold => true)
+
+    status = $?.exitstatus
+
+    if status.zero? 
+      This.say("@#{ testno } <= ", :bold => true, :color => :white, :method => :print)
+      This.say("SUCCESS", :color => :green, :bold => true)
+    else
+      This.say("@#{ testno } <= ", :bold => true, :color => :white, :method => :print)
+      This.say("FAILURE", :color => :red, :bold => true)
+    end
+    This.say(line, :color => :cyan, :bold => true)
+
+    exit(status) unless status.zero?
+  end
+end
+
+
+task :gemspec do
+  ignore_extensions = ['git', 'svn', 'tmp', /sw./, 'bak', 'gem']
+  ignore_directories = ['pkg', 'db']
+  ignore_files = ['test/log', 'test/db.yml', 'a.rb', 'b.rb'] + Dir['db/*'] + %w'db'
+
+  shiteless = 
+    lambda do |list|
+      list.delete_if do |entry|
+        next unless test(?e, entry)
+        extension = File.basename(entry).split(%r/[.]/).last
+        ignore_extensions.any?{|ext| ext === extension}
+      end
+      list.delete_if do |entry|
+        next unless test(?d, entry)
+        dirname = File.expand_path(entry)
+        ignore_directories.any?{|dir| File.expand_path(dir) == dirname}
+      end
+      list.delete_if do |entry|
+        next unless test(?f, entry)
+        filename = File.expand_path(entry)
+        ignore_files.any?{|file| File.expand_path(file) == filename}
+      end
+    end
+
+  lib         = This.lib
+  object      = This.object
+  version     = This.version
+  files       = shiteless[Dir::glob("**/**")]
+  executables = shiteless[Dir::glob("bin/*")].map{|exe| File.basename(exe)}
+  #has_rdoc    = true #File.exist?('doc')
+  test_files  = test(?e, "test/#{ lib }.rb") ? "test/#{ lib }.rb" : nil
+  summary     = This.summary || This.synopsis || "#{ lib } kicks the ass"
+  description = This.description || summary
+
+  if This.extensions.nil?
+    This.extensions = []
+    extensions = This.extensions
+    %w( Makefile configure extconf.rb ).each do |ext|
+      extensions << ext if File.exists?(ext)
+    end
+  end
+  extensions = [extensions].flatten.compact
+
+# TODO
+  if This.dependencies.nil?
+    dependencies = []
+  else
+    case This.dependencies
+      when Hash
+        dependencies = This.dependencies.values
+      when Array
+        dependencies = This.dependencies
+    end
+  end
+
+  template = 
+    if test(?e, 'gemspec.erb')
+      This.template_for{ IO.read('gemspec.erb') }
+    else
+      This.template_for {
+        <<-__
+          ## <%= lib %>.gemspec
+          #
+
+          Gem::Specification::new do |spec|
+            spec.name = <%= lib.inspect %>
+            spec.version = <%= version.inspect %>
+            spec.platform = Gem::Platform::RUBY
+            spec.summary = <%= lib.inspect %>
+            spec.description = <%= description.inspect %>
+
+            spec.files =\n<%= files.sort.pretty_inspect %>
+            spec.executables = <%= executables.inspect %>
+            
+            spec.require_path = "lib"
+
+            spec.test_files = <%= test_files.inspect %>
+
+            <% dependencies.each do |lib_version| %>
+              spec.add_dependency(*<%= Array(lib_version).flatten.inspect %>)
+            <% end %>
+
+            spec.extensions.push(*<%= extensions.inspect %>)
+
+            spec.rubyforge_project = <%= This.rubyforge_project.inspect %>
+            spec.author = <%= This.author.inspect %>
+            spec.email = <%= This.email.inspect %>
+            spec.homepage = <%= This.homepage.inspect %>
+          end
+        __
+      }
+    end
+
+  FileUtils.mkdir_p(This.pkgdir)
+  gemspec = "#{ lib }.gemspec"
+  open(gemspec, "w"){|fd| fd.puts(template)}
+  This.gemspec = gemspec
+end
+
+task :gem => [:clean, :gemspec] do
+  FileUtils.mkdir_p(This.pkgdir)
+  before = Dir['*.gem']
+  cmd = "gem build #{ This.gemspec }"
+  `#{ cmd }`
+  after = Dir['*.gem']
+  gem = ((after - before).first || after.first) or abort('no gem!')
+  FileUtils.mv(gem, This.pkgdir)
+  This.gem = File.join(This.pkgdir, File.basename(gem))
+end
+
+task :readme do
+  samples = ''
+  prompt = '~ > '
+  lib = This.lib
+  version = This.version
+
+  Dir['sample*/*'].sort.each do |sample|
+    samples << "\n" << "  <========< #{ sample } >========>" << "\n\n"
+
+    cmd = "cat #{ sample }"
+    samples << This.util.indent(prompt + cmd, 2) << "\n\n"
+    samples << This.util.indent(`#{ cmd }`, 4) << "\n"
+
+    cmd = "ruby #{ sample }"
+    samples << This.util.indent(prompt + cmd, 2) << "\n\n"
+
+    cmd = "ruby -e'STDOUT.sync=true; exec %(ruby -I ./lib #{ sample })'"
+    samples << This.util.indent(`#{ cmd } 2>&1`, 4) << "\n"
+  end
+
+  template = 
+    if test(?e, 'readme.erb')
+      This.template_for{ IO.read('readme.erb') }
+    else
+      This.template_for {
+        <<-__
+          NAME
+            #{ lib }
+
+          DESCRIPTION
+
+          INSTALL
+            gem install #{ lib }
+
+          SAMPLES
+            #{ samples }
+        __
+      }
+    end
+
+  open("README", "w"){|fd| fd.puts template}
+end
+
+
+task :clean do
+  Dir[File.join(This.pkgdir, '**/**')].each{|entry| FileUtils.rm_rf(entry)}
+end
+
+
+task :release => [:clean, :gemspec, :gem] do
+  gems = Dir[File.join(This.pkgdir, '*.gem')].flatten
+  raise "which one? : #{ gems.inspect }" if gems.size > 1
+  raise "no gems?" if gems.size < 1
+
+  cmd = "gem push #{ This.gem }"
+  puts cmd
+  puts
+  system(cmd)
+  abort("cmd(#{ cmd }) failed with (#{ $?.inspect })") unless $?.exitstatus.zero?
+
+  #cmd = "rubyforge login && rubyforge add_release #{ This.rubyforge_project } #{ This.lib } #{ This.version } #{ This.gem }"
+  #puts cmd
+  #puts
+  #system(cmd)
+  #abort("cmd(#{ cmd }) failed with (#{ $?.inspect })") unless $?.exitstatus.zero?
+end
+
+
+
+
+
+BEGIN {
+# support for this rakefile
+#
+  $VERBOSE = nil
+
+  require 'erb'
+  require 'fileutils'
+  require 'rbconfig'
+  require 'pp'
+
+# cache a bunch of stuff about this rakefile/environment
+#
+
+  This =
+    Class.new(Hash) do
+
+      def method_missing(method, *args, &block)
+        if method.to_s =~ /=/
+          key = method.to_s.chomp('=')
+          value = block ? block : args.shift
+          self[key] = value
+        else
+          key = method.to_s
+          if block
+            value = block
+            self[key] = value
+          else
+            value = self[key]
+
+            if value.respond_to?(:call)
+              self[key] = value.call()
+            else
+              value
+            end
+          end
+        end
+      end
+
+      def inspect
+        expand!
+        PP.pp(self, '')
+      end
+
+      def expand!
+        keys.each do |key|
+          value = self[key]
+          if value.respond_to?(:call)
+            self[key] = value.call()
+          end
+        end
+      end
+
+    end.new()
+
+  This.file = File.expand_path(__FILE__)
+  This.dir = File.dirname(This.file)
+  This.pkgdir = File.join(This.dir, 'pkg')
+
+# defaults 
+#
+  This.lib do
+    File.basename(Dir.pwd)
+  end
+
+  def This.setup!
+    begin
+      require "./lib/#{ This.lib }"
+    rescue LoadError
+      abort("could not load #{ This.lib }")
+    end
+  end
+
+  This.name do
+    This.name = This.lib.capitalize
+  end
+
+  This.object do
+    begin
+      This.object = eval(This.name)
+    rescue Object
+      abort("could not determine object from #{ This.name }")
+    end
+  end
+
+  This.version do
+    This.object.send(:version)
+  end
+
+  This.dependencies do
+    if This.object.respond_to?(:dependencies)
+      This.object.dependencies
+    end
+  end
+
+  This.ruby do
+    c = Config::CONFIG
+    bindir = c["bindir"] || c['BINDIR']
+    ruby_install_name = c['ruby_install_name'] || c['RUBY_INSTALL_NAME'] || 'ruby'
+    ruby_ext = c['EXEEXT'] || ''
+    File.join(bindir, (ruby_install_name + ruby_ext))
+  end
+
+# some utils
+#
+  This.util = Module.new do
+    def indent(s, n = 2)
+      s = unindent(s)
+      ws = ' ' * n
+      s.gsub(%r/^/, ws)
+    end
+
+    def unindent(s)
+      indent = nil
+      s.each_line do |line|
+        next if line =~ %r/^\s*$/
+        indent = line[%r/^\s*/] and break
+      end
+      indent ? s.gsub(%r/^#{ indent }/, "") : s
+    end
+
+    extend self
+  end
+
+# template support
+#
+  This.template = Class.new do
+    def initialize(&block)
+      @block = block
+      @template = block.call.to_s
+    end
+
+    def expand(b=nil)
+      ERB.new(This.util.unindent(@template)).result((b||@block).binding)
+    end
+
+    alias_method 'to_s', 'expand'
+  end
+
+  def This.template_for(*args, &block)
+    This.template.new(*args, &block)
+  end
+
+# colored console output support
+#
+  This.ansi = {
+    :clear      => "\e[0m",
+    :reset      => "\e[0m",
+    :erase_line => "\e[K",
+    :erase_char => "\e[P",
+    :bold       => "\e[1m",
+    :dark       => "\e[2m",
+    :underline  => "\e[4m",
+    :underscore => "\e[4m",
+    :blink      => "\e[5m",
+    :reverse    => "\e[7m",
+    :concealed  => "\e[8m",
+    :black      => "\e[30m",
+    :red        => "\e[31m",
+    :green      => "\e[32m",
+    :yellow     => "\e[33m",
+    :blue       => "\e[34m",
+    :magenta    => "\e[35m",
+    :cyan       => "\e[36m",
+    :white      => "\e[37m",
+    :on_black   => "\e[40m",
+    :on_red     => "\e[41m",
+    :on_green   => "\e[42m",
+    :on_yellow  => "\e[43m",
+    :on_blue    => "\e[44m",
+    :on_magenta => "\e[45m",
+    :on_cyan    => "\e[46m",
+    :on_white   => "\e[47m"
+  }
+
+  def This.say(something, *args)
+    options = args.last.is_a?(Hash) ? args.pop : {}
+    options[:color] = args.shift.to_s.to_sym unless args.empty?
+    keys = options.keys
+    keys.each{|key| options[key.to_s.to_sym] = options.delete(key)}
+
+    color = options[:color]
+    bold = options.has_key?(:bold)
+
+    parts = [something]
+    parts.unshift(This.ansi[color]) if color
+    parts.unshift(This.ansi[:bold]) if bold
+    parts.push(This.ansi[:clear]) if parts.size > 1
+
+    method = options[:method] || :puts
+
+    Kernel.send(method, parts.join)
+  end
+
+# always run out of the project dir
+#
+  Dir.chdir(This.dir)
+}

data/lib/mongoid-bolt.rb

@@ -0,0 +1,349 @@
+# encoding: utf-8
+
+##
+#
+module Mongoid
+  class Bolt
+    const_set :Version, '1.0.0'
+
+    class << Bolt
+      def version
+        const_get :Version
+      end
+
+      def dependencies
+        {
+          'mongoid'         => [ 'mongoid'         , ' >= 3.0.1' ] ,
+        }
+      end
+
+      def libdir(*args, &block)
+        @libdir ||= File.expand_path(__FILE__).sub(/\.rb$/,'')
+        args.empty? ? @libdir : File.join(@libdir, *args)
+      ensure
+        if block
+          begin
+            $LOAD_PATH.unshift(@libdir)
+            block.call()
+          ensure
+            $LOAD_PATH.shift()
+          end
+        end
+      end
+
+      def load(*libs)
+        libs = libs.join(' ').scan(/[^\s+]+/)
+        libdir{ libs.each{|lib| Kernel.load(lib) } }
+      end
+    end
+
+    begin
+      require 'rubygems'
+    rescue LoadError
+      nil
+    end
+
+    if defined?(gem)
+      dependencies.each do |lib, dependency|
+        gem(*dependency)
+        require(lib)
+      end
+    end
+  end
+end
+
+##
+#
+  module Mongoid
+    class Bolt
+      module Ability
+        Code = proc do
+        ## embedded lock class and associations
+        #
+          target_class = self
+
+          class << target_class
+            attr_accessor :lock_class
+          end
+
+          const_set(:Bolt, Class.new)
+          lock_class = const_get(:Bolt)
+
+          target_class.lock_class = lock_class
+
+          lock_class.class_eval do
+            define_method(:target_class){ target_class }
+            define_method(:lock_class){ lock_class }
+
+            include Mongoid::Document
+            include Mongoid::Timestamps
+
+            field(:hostname, :default => proc{ ::Mongoid::Bolt.hostname })
+            field(:ppid, :default => proc{ ::Mongoid::Bolt.ppid })
+            field(:pid, :default => proc{ ::Mongoid::Bolt.pid })
+
+            attr_accessor :stolen
+            alias_method :stolen?, :stolen
+
+            def initialize(*args, &block)
+              super
+            ensure
+              now = Time.now
+              self.created_at ||= now
+              self.updated_at ||= now
+              @locked = false
+            end
+
+            def localhost?
+              ::Mongoid::Bolt.hostname == hostname
+            end
+
+            def alive?
+              return true unless localhost?
+              ::Mongoid::Bolt.alive?(ppid, pid)
+            end
+
+            def relock!
+              reload
+
+              conditions = {
+                '_lock._id'      => id,
+                '_lock.hostname' => hostname,
+                '_lock.ppid'     => ppid,
+                '_lock.pid'      => pid
+              }
+
+              update = {
+                '$set' => {
+                  '_lock.hostname'   => ::Mongoid::Bolt.hostname,
+                  '_lock.ppid'       => ::Mongoid::Bolt.ppid,
+                  '_lock.pid'        => ::Mongoid::Bolt.pid,
+                  '_lock.updated_at' => Time.now.utc
+                }
+              }
+
+              result =
+                  target_class.
+                    with(safe: true).
+                      where(conditions).
+                        find_and_modify(update, new: false)
+
+            ensure
+              reload
+            end
+
+            def steal!
+              self.stolen = !!relock!
+            end
+
+            def stale?
+              localhost? and not alive?
+            end
+
+            def owner?
+              ::Mongoid::Bolt.identifier == identifier
+            end
+
+            def identifier
+              {:hostname => hostname, :ppid => ppid, :pid => pid}
+            end
+          end
+
+          target_association_name = "_" + target_class.name.underscore.split(%r{/}).last
+          
+          lock_class.class_eval do
+            embedded_in(target_association_name, :class_name => "::#{ target_class.name }")
+          end
+
+          embeds_one(:_lock, :class_name => "::#{ lock_class.name }")
+
+        ## locking methods
+        #
+          def target_class.lock!(conditions = {}, update = {})
+            conditions.to_options!
+            update.to_options!
+
+            conditions[:_lock] = nil
+
+            update[:$set] = {:_lock => lock_class.new.attributes}
+
+            with(safe: true).
+              where(conditions).
+                find_and_modify(update, new: true)
+          end
+
+          def lock!(conditions = {})
+            conditions.to_options!
+
+            begin
+              if _lock and _lock.stale? and _lock.steal!
+                return _lock
+              end
+            rescue
+              nil
+            end
+
+            conditions[:_id] = id
+
+            if self.class.lock!(conditions)
+              reload
+
+              begin
+                @locked = _lock && _lock.owner?
+              rescue
+                nil
+              end
+            else
+              false
+            end
+          end
+
+          def unlock!
+            unlocked = false
+
+            if _lock
+              begin
+                _lock.destroy if _lock.owner?
+                @locked = false
+                unlocked = true
+              rescue
+                nil
+              end
+
+              reload
+            end
+
+            unlocked
+          end
+
+          def relock!
+            raise(::Mongoid::Bolt::Error, "#{ name } is not locked!") unless @locked
+
+            _lock.relock!
+          end
+
+          def locked?
+            begin
+              _lock and _lock.owner?
+            rescue
+              nil
+            end
+          end
+
+          def lock(options = {}, &block)
+            options.to_options!
+
+            return block.call(_lock) if locked?
+
+            loop do
+              if lock!
+                return _lock unless block
+
+                begin
+                  return block.call(_lock)
+                ensure
+                  unlock!
+                end
+              else
+                if options[:blocking] == false
+                  if block
+                    raise(::Mongoid::Bolt::Error, name)
+                  else
+                    return(false)
+                  end
+                end
+
+                if options[:waiting]
+                  options[:waiting].call(reload._lock)
+                end
+
+                sleep(rand)
+              end
+            end
+          end
+        end
+
+        def Ability.included(other)
+          super
+        ensure
+          other.module_eval(&Code)
+        end
+      end
+
+      def Bolt.ability
+        Ability
+      end
+
+    ##
+    #
+      include Mongoid::Document
+      include Mongoid::Timestamps
+
+    ##
+    #
+      class Error < ::StandardError; end
+
+    ##
+    #
+      def Bolt.for(name, options = {}, &block)
+        name = name.to_s
+        conditions = {:name => name}
+
+        attributes = conditions.dup
+        attributes[:created_at] || attributes[:updated_at] = Time.now.utc
+
+        lock =
+          begin
+            where(conditions).first or create!(attributes)
+          rescue Object => e
+            sleep(rand)
+            where(conditions).first or create!(attributes)
+          end
+
+        block ? lock.lock(options, &block) : lock
+      end
+
+    ##
+    #
+      field(:name)
+
+      validates_presence_of(:name)
+      validates_uniqueness_of(:name)
+
+      index({:name => 1}, {:unique => true})
+
+    ##
+    #
+      def Bolt.hostname
+        Socket.gethostname
+      end
+
+      def Bolt.ppid
+        Process.ppid
+      end
+
+      def Bolt.pid
+        Process.pid
+      end
+
+      def Bolt.identifier
+        {:hostname => hostname, :ppid => ppid, :pid => pid}
+      end
+
+      def Bolt.alive?(*pids)
+        pids.flatten.compact.all? do |pid|
+          begin
+            Process.kill(0, Integer(pid))
+            true
+          rescue Object
+            false
+          end
+        end
+      end
+
+    ##
+    #
+      include Bolt.ability
+    end
+  end
+

data/mongoid-bolt.gemspec

@@ -0,0 +1,38 @@
+## mongoid-bolt.gemspec
+#
+
+Gem::Specification::new do |spec|
+  spec.name = "mongoid-bolt"
+  spec.version = "1.0.0"
+  spec.platform = Gem::Platform::RUBY
+  spec.summary = "mongoid-bolt"
+  spec.description = "a mongoid 3/moped compatible lock implementation and mixin"
+
+  spec.files =
+["README.md",
+ "Rakefile",
+ "lib",
+ "lib/mongoid-bolt.rb",
+ "mongoid-bolt.gemspec",
+ "test",
+ "test/helper.rb",
+ "test/mongoid-bolt_test.rb",
+ "test/testing.rb"]
+
+  spec.executables = []
+  
+  spec.require_path = "lib"
+
+  spec.test_files = nil
+
+  
+    spec.add_dependency(*["mongoid", " >= 3.0.1"])
+  
+
+  spec.extensions.push(*[])
+
+  spec.rubyforge_project = "codeforpeople"
+  spec.author = "Ara T. Howard"
+  spec.email = "ara.t.howard@gmail.com"
+  spec.homepage = "https://github.com/ahoward/mongoid-bolt"
+end

data/test/helper.rb

@@ -0,0 +1,7 @@
+# -*- encoding : utf-8 -*-
+require_relative 'testing'
+require_relative '../lib/mongoid-bolt.rb'
+
+Mongoid.configure do |config|
+  config.connect_to('mongoid-bolt_test')
+end

data/test/mongoid-bolt_test.rb

@@ -0,0 +1,70 @@
+require_relative 'helper'
+
+Testing Mongoid::Bolt do
+##
+#
+  Bolt = Mongoid::Bolt
+
+  class A
+    include Mongoid::Document
+    include Bolt::Ability
+  end
+        
+  setup do
+    Bolt.destroy_all
+    A.destroy_all
+  end
+
+  test 'that locks can be acquired by name' do
+    lock = assert{ Bolt.for(:shared_resource) } 
+  end
+
+  test 'that locks can be locked' do
+    lock = assert{ Bolt.for(:shared_resource) } 
+    assert{ lock.lock! } 
+  end
+
+  test 'that locks cannot be locked twice' do
+    lock = assert{ Bolt.for(:shared_resource) } 
+    assert{ lock.lock! } 
+    assert{ !lock.lock! } 
+  end
+
+  test 'that locks can be re-locked' do
+    lock = assert{ Bolt.for(:shared_resource) } 
+    assert{ lock.lock! } 
+    a = lock._lock.updated_at
+    sleep(0.042)
+    assert{ lock.relock! } 
+    b = lock._lock.updated_at
+    assert{ b > a }
+  end
+
+  test 'that locks can be un-locked' do
+    lock = assert{ Bolt.for(:shared_resource) } 
+    assert{ lock.lock! } 
+    assert{ lock.unlock! } 
+  end
+
+  test 'that locks know when they are locked' do
+    lock = assert{ Bolt.for(:shared_resource) } 
+    assert{ !lock.locked? }
+    assert{ lock.lock! }
+    assert{ lock.locked? }
+  end
+
+  test 'that #lock takes a block' do
+    lock = assert{ Bolt.for(:shared_resource) } 
+    assert do
+      assert{ !lock.locked? }
+      lock.lock{ assert{ lock.locked? } }
+      assert{ !lock.locked? }
+    end
+  end
+
+  test 'that other classes can mix-in lockability' do
+    locked = false
+    assert{ A.create.lock{ locked = true } }
+    assert{ locked }
+  end
+end

data/test/testing.rb

@@ -0,0 +1,196 @@
+# -*- encoding : utf-8 -*-
+require 'test/unit'
+
+testdir = File.expand_path(File.dirname(__FILE__))
+rootdir = File.dirname(testdir)
+libdir = File.join(rootdir, 'lib')
+
+STDOUT.sync = true
+
+$:.unshift(testdir) unless $:.include?(testdir)
+$:.unshift(libdir) unless $:.include?(libdir)
+$:.unshift(rootdir) unless $:.include?(rootdir)
+
+class Testing
+  class Slug < ::String
+    def Slug.for(*args)
+      string = args.flatten.compact.join('-')
+      words = string.to_s.scan(%r/\w+/)
+      words.map!{|word| word.gsub %r/[^0-9a-zA-Z_-]/, ''}
+      words.delete_if{|word| word.nil? or word.strip.empty?}
+      new(words.join('-').downcase)
+    end
+  end
+  
+  class Context
+    attr_accessor :name
+
+    def initialize(name, *args)
+      @name = name
+    end
+
+    def to_s
+      Slug.for(name)
+    end
+  end
+end
+
+def Testing(*args, &block)
+  Class.new(::Test::Unit::TestCase) do
+
+  ## class methods
+  #
+    class << self
+      def contexts
+        @contexts ||= []
+      end
+
+      def context(*args, &block)
+        return contexts.last if(args.empty? and block.nil?)
+
+        context = Testing::Context.new(*args)
+        contexts.push(context)
+
+        begin
+          block.call(context)
+        ensure
+          contexts.pop
+        end
+      end
+
+      def slug_for(*args)
+        string = [context, args].flatten.compact.join('-')
+        words = string.to_s.scan(%r/\w+/)
+        words.map!{|word| word.gsub %r/[^0-9a-zA-Z_-]/, ''}
+        words.delete_if{|word| word.nil? or word.strip.empty?}
+        words.join('-').downcase.sub(/_$/, '')
+      end
+
+      def name() const_get(:Name) end
+
+      def testno()
+        '%05d' % (@testno ||= 0)
+      ensure
+        @testno += 1
+      end
+
+      def testing(*args, &block)
+        method = ["test", testno, slug_for(*args)].delete_if{|part| part.empty?}.join('_')
+        define_method(method, &block)
+      end
+
+      def test(*args, &block)
+        testing(*args, &block)
+      end
+
+      def setup(&block)
+        define_method(:setup, &block) if block
+      end
+
+      def teardown(&block)
+        define_method(:teardown, &block) if block
+      end
+
+      def prepare(&block)
+        @prepare ||= []
+        @prepare.push(block) if block
+        @prepare
+      end
+
+      def cleanup(&block)
+        @cleanup ||= []
+        @cleanup.push(block) if block
+        @cleanup
+      end
+    end
+
+  ## configure the subclass!
+  #
+    const_set(:Testno, '0')
+    slug = slug_for(*args).gsub(%r/-/,'_')
+    name = ['TESTING', '%03d' % const_get(:Testno), slug].delete_if{|part| part.empty?}.join('_')
+    name = name.upcase!
+    const_set(:Name, name)
+    const_set(:Missing, Object.new.freeze)
+
+  ## instance methods
+  #
+    alias_method('__assert__', 'assert')
+
+    def assert(*args, &block)
+      if args.size == 1 and args.first.is_a?(Hash)
+        options = args.first
+        expected = getopt(:expected, options){ missing }
+        actual = getopt(:actual, options){ missing }
+        if expected == missing and actual == missing
+          actual, expected, *ignored = options.to_a.flatten
+        end
+        expected = expected.call() if expected.respond_to?(:call)
+        actual = actual.call() if actual.respond_to?(:call)
+        assert_equal(expected, actual)
+      end
+
+      if block
+        label = "assert(#{ args.join(' ') })"
+        result = nil
+        assert_nothing_raised{ result = block.call }
+        __assert__(result, label)
+        result
+      else
+        result = args.shift
+        label = "assert(#{ args.join(' ') })"
+        __assert__(result, label)
+        result
+      end
+    end
+
+    def missing
+      self.class.const_get(:Missing)
+    end
+
+    def getopt(opt, hash, options = nil, &block)
+      [opt.to_s, opt.to_s.to_sym].each do |key|
+        return hash[key] if hash.has_key?(key)
+      end
+      default =
+        if block
+          block.call
+        else
+          options.is_a?(Hash) ? options[:default] : nil
+        end
+      return default
+    end
+
+    def subclass_of exception
+      class << exception
+        def ==(other) super or self > other end
+      end
+      exception
+    end
+
+  ##
+  #
+    module_eval(&block)
+
+    self.setup()
+    self.prepare.each{|b| b.call()}
+
+    at_exit{ 
+      self.teardown()
+      self.cleanup.each{|b| b.call()}
+    }
+
+    self
+  end
+end
+
+
+if $0 == __FILE__
+
+  Testing 'Testing' do
+    testing('foo'){ assert true }
+    test{ assert true }
+    p instance_methods.grep(/test/)
+  end
+
+end

metadata

@@ -0,0 +1,67 @@
+--- !ruby/object:Gem::Specification
+name: mongoid-bolt
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Ara T. Howard
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mongoid
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.1
+description: a mongoid 3/moped compatible lock implementation and mixin
+email: ara.t.howard@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- Rakefile
+- lib/mongoid-bolt.rb
+- mongoid-bolt.gemspec
+- test/helper.rb
+- test/mongoid-bolt_test.rb
+- test/testing.rb
+homepage: https://github.com/ahoward/mongoid-bolt
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: codeforpeople
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: mongoid-bolt
+test_files: []