Sutto-reversible_data 0.0.4 → 0.0.5

data/lib/reversible_data.rb

@@ -17,12 +17,35 @@ module ReversibleData
     Table.known_models[name]
   end
   
+  def self.remove(name)
+    table = self[name]
+    table.down!
+    table.destroy
+  end
+  
+  def self.remove_all
+    Table.known_models.each_value do |t|
+      t.down!
+      t.destroy
+    end
+  end
+  
   def self.manager_for(*tables)
     TableManager.new(*tables)
   end
   
   def self.in_memory!
-    ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :dbfile => ":memory:")
+    if connected?
+      ActiveRecord::Base.connection.disconnect!
+      ActiveRecord::Base.remove_connection
+    end
+    ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => ":memory:")
+  end
+  
+  def self.connected?
+    ActiveRecord::Base.connection.active?
+  rescue ActiveRecord::ConnectionNotEstablished
+    false
   end
   
 end

data/lib/reversible_data/table.rb

@@ -4,40 +4,60 @@ module ReversibleData
     cattr_accessor :known_models
     self.known_models = {}
     
-    attr_accessor :table_name, :model_name, :options
+    attr_accessor :name, :table_name, :model_name, :options
     
-    def initialize(table_name, opts = {}, &blk)
-      @table_name       = table_name.to_s.tableize.to_sym
-      @model_name       = (opts[:class_name] || @table_name).to_s.classify
+    def initialize(name, opts = {}, &blk)
+      @name             = name.to_s.underscore.to_sym
+      @table_name       = (opts[:table_name] || @name.to_s.tableize).to_s
+      @model_name       = (opts[:class_name] || @name.to_s.classify).to_s
       @migrator         = blk
       @model_definition = nil
-      default_options   = {:skip_model => Object.const_defined?(@model_name), :skip_table => connection.table_exists?(@table_name)}
+      default_options   = {:skip_model => Object.const_defined?(@model_name),
+                           :skip_table => (connected? && connection.table_exists?(@table_name))}
       @options          = opts.reverse_merge(default_options)
       @blueprint        = nil
-      self.known_models[table_name.to_sym] = self
+      self.known_models[@name] = self
+    end
+    
+    def skip_model?
+      !!@options[:skip_model]
+    end
+    
+    def skip_table?
+      !!@options[:skip_table]
+    end
+    
+    def up!
+      create_table
+      create_model
+    end
+    
+    def down!
+      remove_model
+      drop_table
     end
     
     def drop_table
-      return if @options[:skip_table]
+      return if skip_table? || !connected?
       connection.drop_table(@table_name) if connection.table_exists?(@table_name)
     end
     
     def create_table(autodrop = true)
-      return if @options[:skip_table]
+      return if skip_table? || !connected?
       drop_table if autodrop
       return if connection.table_exists?(@table_name)
       connection.create_table(table_name, &@migrator)
     end
     
     def remove_model
-      return if @options[:skip_model]
+      return if 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]
+      return if skip_model?
       remove_model if autoremove
       return if Object.const_defined?(@model_name)
       @model = Class.new(ActiveRecord::Base)
@@ -59,7 +79,7 @@ module ReversibleData
     end
     
     def destroy
-      self.known_models[@table_name] = nil
+      self.known_models.delete(@name)
     end
     
     protected
@@ -68,5 +88,9 @@ module ReversibleData
       ActiveRecord::Base.connection
     end
     
+    def connected?
+      ReversibleData.connected?
+    end
+    
   end
 end

data/lib/reversible_data/table_manager.rb

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

data/test/reversible_data_test.rb

@@ -0,0 +1,47 @@
+require 'test_helper'
+
+class ReversibleDataTest < Test::Unit::TestCase
+  
+  should 'have add as a shortcut to Table.new' do
+    mock(ReversibleData::Table).new(:awesome, {})
+    ReversibleData.add(:awesome)
+    mock(ReversibleData::Table).new(:rocket, :table_name => "awesomesauces")
+    ReversibleData.add(:rocket, :table_name => "awesomesauces")
+  end
+  
+  should 'have [] as a shortcut to find' do
+    mock(ReversibleData::Table.known_models)[:awesome]
+    ReversibleData[:awesome]
+  end
+  
+  should 'have remove to remove a specific table' do
+    ReversibleData.add(:awesome) { |t| t.string :name }
+    assert !ReversibleData[:awesome].nil?
+    ReversibleData.remove(:awesome)
+    assert ReversibleData[:awesome].nil?
+  end
+  
+  should 'let you remove all definitions with remove_all' do
+    ReversibleData.add(:awesome) { |t| t.string :name }
+    ReversibleData.add(:ninjas)  { |t| t.string :name }
+    assert !ReversibleData[:awesome].nil?
+    assert !ReversibleData[:ninjas].nil?
+    ReversibleData.remove_all
+    assert ReversibleData[:awesome].nil?
+    assert ReversibleData[:ninjas].nil?
+  end
+  
+  should 'let you create a Table Manager with manager_for' do
+    ReversibleData.add(:awesome) { |t| t.string :name }
+    ReversibleData.add(:ninjas)  { |t| t.string :name }
+    mock(ReversibleData::TableManager).new(:awesome, :ninjas)
+    ReversibleData.manager_for(:awesome, :ninjas)
+  end
+  
+  should 'let you establish an in-memory connection' do
+    assert !ReversibleData.connected?
+    ReversibleData.in_memory!
+    assert ReversibleData.connected?
+  end
+  
+end

data/test/table_manager_test.rb

@@ -0,0 +1,28 @@
+require 'test_helper'
+
+class TableManagerTest < Test::Unit::TestCase
+  context 'table managers in general' do
+    
+    setup do
+      @table_a = ReversibleData.add(:a) { |t| t.string :name }
+      @table_b = ReversibleData.add(:b) { |t| t.string :name }
+      @table_c = ReversibleData.add(:c) { |t| t.string :name }
+      @manager = ReversibleData::TableManager.new(:a, :b, :c)
+    end
+  
+    should 'let you create a manager for a set of tables' do
+      assert_equal [:a, :b, :c], @manager.managed_tables
+    end
+  
+    should 'let you run up! on all managed tables' do
+      [@table_a, @table_b, @table_c].each { |t| mock(t).up! }
+      @manager.up!
+    end
+  
+    should 'let you run down! on all managed tables' do
+      [@table_a, @table_b, @table_c].each { |t| mock(t).down! }
+      @manager.down!
+    end
+    
+  end
+end

data/test/table_test.rb

@@ -0,0 +1,113 @@
+require 'test_helper'
+
+class TableTest < Test::Unit::TestCase
+  context '' do
+    
+    setup    { ReversibleData.in_memory! }
+  
+    teardown { ReversibleData.remove_all }
+  
+    should 'automatically add it to a list of known models' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      assert_equal table, ReversibleData::Table.known_models[:user]
+    end
+  
+    should 'default the table name' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      assert_equal "users", table.table_name
+      table = ReversibleData::Table.new(:users) { |t| t.string :name }
+      assert_equal "users", table.table_name
+    end
+  
+    should 'default the model class name' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      assert_equal "User", table.model_name
+      table = ReversibleData::Table.new(:users) { |t| t.string :name }
+      assert_equal "User", table.model_name
+    end
+  
+    should 'let you override the table name' do
+      table = ReversibleData::Table.new(:user, :table_name => "awesomesauce") { |t| t.string :name }
+      assert_equal "awesomesauce", table.table_name
+    end
+  
+    should 'let you override the model class name' do
+      table = ReversibleData::Table.new(:user, :class_name => "Ninja") { |t| t.string :name }
+      assert_equal "Ninja", table.model_name
+    end
+  
+    should 'let you skip creating the model' do
+      table = ReversibleData::Table.new(:user, :skip_model => true) { |t| t.string :name }
+      table.create_model
+      assert !defined?(User)
+    end
+  
+    should 'let you skip creating the table' do
+      table = ReversibleData::Table.new(:user, :skip_table => true) { |t| t.string :name }
+      table.create_table
+      assert !ActiveRecord::Base.connection.table_exists?(:users)
+    end
+  
+    should 'let you create the table' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.create_table
+      assert ActiveRecord::Base.connection.table_exists?(:users)
+    end
+  
+    should 'let you drop the table' do
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.create_table
+      table.drop_table
+      assert !ActiveRecord::Base.connection.table_exists?(:users)
+    end
+  
+    should 'let you create the model' do
+      assert !defined?(::User)
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.create_model
+      assert defined?(::User)
+    end
+  
+    should 'let you remove the model' do
+      assert !defined?(::User)
+      table = ReversibleData::Table.new(:user) { |t| t.string :name }
+      table.create_model
+      table.remove_model
+      assert !defined?(::User)
+    end
+    
+    should 'let you call up! to create the table and the model'
+    
+    should 'let you call down! to clean up'
+  
+    should 'default to skipping model if the constant exists' do
+      ::Awesome = true
+      table = ReversibleData::Table.new(:awesome)
+      assert table.skip_model?
+      Object.send(:remove_const, :Awesome)
+    end
+  
+    should 'default to skipping the table if connected and the table exists' do
+      ActiveRecord::Base.connection.create_table(:ninjas) { |t| t.string :name }
+      table = ReversibleData::Table.new(:ninja)
+      assert table.skip_table?
+      ActiveRecord::Base.connection.drop_table(:ninjas)
+    end
+  
+    should 'let you define a model block' do
+      table = ReversibleData::Table.new(:ninja)
+      table.define_model do
+        cattr_accessor :ninjariffic
+        self.ninjariffic = "felafel"
+      end
+      table.create_model
+      assert Ninja.respond_to?(:ninjariffic)
+      assert Ninja.respond_to?(:ninjariffic=)
+      assert_equal "felafel", Ninja.ninjariffic
+      table.remove_model
+    end
+  
+    should 'let you specify a blueprint for machinist'
+  
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+require 'rr'
+require 'redgreen'
+require 'reversible_data'
+
+class Test::Unit::TestCase
+  include RR::Adapters::TestUnit
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: Sutto-reversible_data
 version: !ruby/object:Gem::Version 
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors: 
 - Darcy Laycock
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-08-05 00:00:00 -07:00
+date: 2009-08-06 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
@@ -27,6 +27,10 @@ files:
 - lib/reversible_data/table.rb
 - lib/reversible_data/table_manager.rb
 - lib/reversible_data.rb
+- test/reversible_data_test.rb
+- test/table_manager_test.rb
+- test/table_test.rb
+- test/test_helper.rb
 has_rdoc: false
 homepage: http://sutto.net/
 licenses: