Sutto-reversible_data 0.0.1

data/lib/reversible_data/shoulda_macros.rb

@@ -0,0 +1,14 @@
+module ReversibleData
+  module ShouldaMacros
+    
+    def with_tables(*args, &blk)
+      table_manager = ReversibleData.manager_for(*args)
+      context '' do
+        setup { table_manager.up! }
+        context('', &blk)
+        teardown { table_manager.down! }
+      end
+    end
+    
+  end
+end

data/lib/reversible_data/table.rb

@@ -0,0 +1,65 @@
+module ReversibleData
+  class Table
+    
+    cattr_accessor :known_models
+    self.known_models = {}
+    
+    attr_accessor :table_name, :model_name, :options
+    
+    def initialize(table_name, model_name = nil, opts = {}, &blk)
+      @table_name       = table_name.to_s.tableize.to_sym
+      @model_name       = (model_name || @table_name).to_s.classify
+      @migrator         = blk
+      @model_definition = nil
+      @options          = opts
+      self.known_models[table_name.to_sym] = self
+    end
+    
+    def drop_table
+      return if @options[:skip_model]
+      connection.drop_table(@table_name) if connection.table_exists?(@table_name)
+    end
+    
+    def create_table(autodrop = true)
+      return if @options[:skip_table]
+      drop_table if autodrop
+      return if connection.table_exists?(@table_name)
+      connection.create_table(table_name, &blk)
+    end
+    
+    def remove_model
+      return if @options[:skip_model]
+      if Object.const_defined?(@model_name)
+        silence_warnings { Object.send(:remove_const, @model_name) }
+      end
+    end
+    
+    def create_model(autoremove = true)
+      return if @options[:skip_model]
+      remove_model if autoremove
+      return if Object.const_defined?(@model_name)
+      @model = Class.new(ActiveRecord::Base)
+      @model.class_eval(&@model_definition) unless @model_definition.nil?
+      Object.const_set(@model_name, @model)
+    end
+    
+    def define_model(&blk)
+      @model_definition = blk unless blk.nil?
+    end
+    
+    def clear_model_definition!
+      @model_definition = nil
+    end
+    
+    def destroy
+      self.known_models[@table_name] = nil
+    end
+    
+    protected
+    
+    def connection
+      ActiveRecord::Base.connection
+    end
+    
+  end
+end

data/lib/reversible_data/table_manager.rb

@@ -0,0 +1,33 @@
+module ReversibleData
+  class TableManager
+    
+    def initialize(*names)
+      @names = names.map { |n| n.to_s.tableize.to_sym }
+    end
+    
+    def up!
+      @names.each do |name|
+        unless (table = table_for(name)).nil?
+          table.create_table
+          table.create_model
+        end
+      end
+    end
+    
+    def down!
+      @names.each do |name|
+        unless (table = table_for(name)).nil?
+          table.drop_table
+          table.remove_model
+        end
+      end
+    end
+    
+    protected
+    
+    def table_for(name)
+      Table.known_models[name.to_sym]
+    end
+    
+  end
+end

data/lib/reversible_data.rb

@@ -0,0 +1,24 @@
+$:.unshift File.dirname(__FILE__)
+
+require 'active_support'
+require 'active_record'
+
+module ReversibleData
+  
+  autoload :Table,         'reversible_data/table'
+  autoload :TableManager,  'reversible_data/table_manager'
+  autoload :ShouldaMacros, 'reversible_data/shoulda_macros'
+  
+  def self.add(name, model_name = nil, &blk)
+    Table.new(name, model_name, &blk)
+  end
+  
+  def self.manager_for(*tables)
+    TableManager.new(*tables)
+  end
+  
+  def self.in_memory!
+    ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :dbfile => ":memory:")
+  end
+  
+end

metadata

@@ -0,0 +1,58 @@
+--- !ruby/object:Gem::Specification 
+name: Sutto-reversible_data
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Darcy Laycock
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-08-05 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: sutto@sutto.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/reversible_data
+- lib/reversible_data/shoulda_macros.rb
+- lib/reversible_data/table.rb
+- lib/reversible_data/table_manager.rb
+- lib/reversible_data.rb
+has_rdoc: false
+homepage: http://sutto.net/
+licenses: 
+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.3.5
+signing_key: 
+specification_version: 3
+summary: Reversible Data provides migration-like functionality for tests etc - All with temporary models.
+test_files: []
+