also_migrate 0.3.0 → 0.3.2

data/.gitignore

@@ -0,0 +1,7 @@
+.DS_Store
+*.gem
+coverage
+pkg
+spec/config/database.yml
+spec/log
+tmp

data/LICENSE

@@ -0,0 +1,18 @@
+Copyright (c) 2010
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of 
+this software and associated documentation files (the "Software"), to deal in 
+the Software without restriction, including without limitation the rights to 
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of 
+the Software, and to permit persons to whom the Software is furnished to do so, 
+subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all 
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS 
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR 
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER 
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN 
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,43 @@
+AlsoMigrate
+===========
+
+Migrate multiple tables with similar schema at once.
+
+Requirements
+------------
+
+<pre>
+gem install also_migrate
+</pre>
+
+Define the model
+----------------
+
+<pre>
+class Article &lt; ActiveRecord::Base
+  also_migrate(
+    :article_archives,
+    :add => [
+      # Parameters to ActiveRecord::ConnectionAdapters::SchemaStatements#add_column
+      [ 'deleted_at', :datetime, {} ]
+    ],
+    :subtract => 'restored_at',
+    :ignore => 'deleted_at',
+    :indexes => 'id'
+  )
+end
+</pre>
+
+Options:
+
+* <code>add</code> Create columns that the original table doesn't have (defaults to none)
+* <code>subtract</code> Exclude columns from the original table (defaults to none)
+* <code>ignore</code> Ignore migrations that apply to certain columns (defaults to none)
+* <code>indexes</code> Only index certain columns (duplicates all indexes by default)
+
+That's it!
+----------
+
+Next time you migrate, <code>article_archives</code> is created if it doesn't exist.
+
+Any new migration applied to <code>articles</code> is automatically applied to <code>article_archives</code>.

data/Rakefile

@@ -0,0 +1,90 @@
+require File.dirname(__FILE__) + '/lib/also_migrate/gems'
+
+AlsoMigrate::Gems.activate %w(rake rspec)
+
+require 'rake'
+require 'spec/rake/spectask'
+
+def gemspec
+  @gemspec ||= begin
+    file = File.expand_path('../also_migrate.gemspec', __FILE__)
+    eval(File.read(file), binding, file)
+  end
+end
+
+if defined?(Spec::Rake::SpecTask)
+  desc "Run specs"
+  Spec::Rake::SpecTask.new do |t|
+    t.spec_files = FileList['spec/**/*_spec.rb']
+    t.spec_opts = %w(-fs --color)
+    t.warning = true
+  end
+  task :spec
+  task :default => :spec
+end
+
+desc "Build gem(s)"
+task :gem do
+  old_gemset = ENV['GEMSET']
+  root = File.expand_path('../', __FILE__)
+  pkg = "#{root}/pkg"
+  system "rm -Rf #{pkg}"
+  AlsoMigrate::Gems.gemset_names.each do |gemset|
+    ENV['GEMSET'] = gemset.to_s
+    system "cd #{root} && gem build also_migrate.gemspec"
+    system "mkdir -p #{pkg} && mv *.gem pkg"
+  end
+  ENV['GEMSET'] = old_gemset
+end
+
+namespace :gem do
+  desc "Install gem(s)"
+  task :install do
+    Rake::Task['gem'].invoke
+    Dir["#{File.dirname(__FILE__)}/pkg/*.gem"].each do |pkg|
+      system "gem install #{pkg} --no-ri --no-rdoc"
+    end
+  end
+  
+  desc "Push gem(s)"
+  task :push do
+    Rake::Task['gem'].invoke
+    Dir["#{File.dirname(__FILE__)}/pkg/*.gem"].each do |pkg|
+      system "gem push #{pkg}"
+    end
+  end
+end
+
+namespace :gems do
+  desc "Install gem dependencies (DEV=0 DOCS=0 GEMSPEC=default SUDO=0)"
+  task :install do
+    dev = ENV['DEV'] == '1'
+    docs = ENV['DOCS'] == '1' ? '' : '--no-ri --no-rdoc'
+    gemset = ENV['GEMSET']
+    sudo = ENV['SUDO'] == '1' ? 'sudo' : ''
+    
+    AlsoMigrate::Gems.gemset = gemset if gemset
+    
+    if dev
+      gems = AlsoMigrate::Gems.gemspec.development_dependencies
+    else
+      gems = AlsoMigrate::Gems.gemspec.dependencies
+    end
+    
+    gems.each do |name|
+      name = name.to_s
+      version = AlsoMigrate::Gems.versions[name]
+      if Gem.source_index.find_name(name, version).empty?
+        version = version ? "-v #{version}" : ''
+        system "#{sudo} gem install #{name} #{version} #{docs}"
+      else
+        puts "already installed: #{name} #{version}"
+      end
+    end
+  end
+end
+
+desc "Validate the gemspec"
+task :gemspec do
+  gemspec.validate
+end

data/also_migrate.gemspec

@@ -0,0 +1,32 @@
+# -*- encoding: utf-8 -*-
+root = File.expand_path('../', __FILE__)
+lib = "#{root}/lib"
+$:.unshift lib unless $:.include?(lib)
+ 
+require 'also_migrate/gems'
+AlsoMigrate::Gems.gemset ||= ENV['GEMSET'] || :default
+
+Gem::Specification.new do |s|
+  AlsoMigrate::Gems.gemspec.hash.each do |key, value|
+    if key == 'name' && AlsoMigrate::Gems.gemset != :default
+      s.name = "#{value}-#{AlsoMigrate::Gems.gemset}"
+    elsif key == 'summary' && AlsoMigrate::Gems.gemset == :solo
+      s.summary = value + " (no dependencies)"
+    elsif !%w(dependencies development_dependencies).include?(key)
+      s.send "#{key}=", value
+    end
+  end
+
+  AlsoMigrate::Gems.dependencies.each do |g|
+    s.add_dependency g.to_s, AlsoMigrate::Gems.versions[g]
+  end
+  
+  AlsoMigrate::Gems.development_dependencies.each do |g|
+    s.add_development_dependency g.to_s, AlsoMigrate::Gems.versions[g]
+  end
+
+  s.executables = `cd #{root} && git ls-files -- {bin}/*`.split("\n").collect { |f| File.basename(f) }
+  s.files = `cd #{root} && git ls-files`.split("\n")
+  s.require_paths = %w(lib)
+  s.test_files = `cd #{root} && git ls-files -- {features,test,spec}/*`.split("\n")
+end

data/config/gemsets.yml

@@ -0,0 +1,4 @@
+also_migrate:
+  active_wrapper: =0.4.2
+  rake: >=0.8.7
+  rspec: ~>1.0

data/config/gemspec.yml

@@ -0,0 +1,12 @@
+name: also_migrate
+version: 0.3.2
+authors:
+  - Winton Welsh
+email: mail@wintoni.us
+homepage: http://github.com/winton/also_migrate
+summary: Migrate multiple tables with similar schema at once.
+description: Migrate multiple tables with similar schema at once.
+dependencies: null
+development_dependencies:
+  - rake
+  - rspec

data/init.rb

@@ -0,0 +1 @@
+require File.dirname(__FILE__) + "/rails/init"

data/lib/also_migrate/base.rb

@@ -0,0 +1,35 @@
+module AlsoMigrate
+  module Base
+    
+    def self.included(base)
+      unless base.respond_to?(:also_migrate)
+        base.extend ClassMethods
+      end
+    end
+    
+    module ClassMethods
+      
+      def also_migrate(*args)
+        options = args.extract_options!
+        @also_migrate_config ||= []
+        @also_migrate_config << {
+          :table_name => self.table_name,
+          :tables => args.collect(&:to_s),
+          :options => {
+            :add => options[:add] ? options[:add] : [],
+            :subtract => [ options[:subtract] ].flatten.compact,
+            :ignore => [ options[:ignore] ].flatten.compact,
+            :indexes => options[:indexes] ? [ options[:indexes] ].flatten : nil
+          }
+        }
+        self.class_eval do
+          class <<self
+            attr_accessor :also_migrate_config
+          end
+        end
+        ::AlsoMigrate.classes ||= []
+        ::AlsoMigrate.classes << self
+      end
+    end
+  end
+end

data/lib/also_migrate/gems.rb

@@ -0,0 +1,154 @@
+unless defined?(AlsoMigrate::Gems)
+  
+  require 'yaml'
+  
+  module AlsoMigrate
+    module Gems
+      class <<self
+        
+        attr_accessor :config
+        attr_reader :gemset, :gemsets, :versions
+        
+        class SimpleStruct
+          attr_reader :hash
+          
+          def initialize(hash)
+            @hash = hash
+            @hash.each do |key, value|
+              self.class.send(:define_method, key) { @hash[key] }
+              self.class.send(:define_method, "#{key}=") { |v| @hash[key] = v }
+            end
+          end
+        end
+        
+        Gems.config = SimpleStruct.new(
+          :gemsets => [ "#{File.expand_path('../../../', __FILE__)}/config/gemsets.yml" ],
+          :gemspec => "#{File.expand_path('../../../', __FILE__)}/config/gemspec.yml",
+          :warn => true
+        )
+        
+        def activate(*gems)
+          begin
+            require 'rubygems' unless defined?(::Gem)
+          rescue LoadError
+            puts "rubygems library could not be required" if @config.warn
+          end
+          
+          self.gemset ||= gemset_from_loaded_specs
+          
+          gems.flatten.collect(&:to_sym).each do |name|
+            version = @versions[name]
+            vendor = File.expand_path("../../../vendor/#{name}/lib", __FILE__)
+            if File.exists?(vendor)
+              $:.unshift vendor
+            elsif defined?(gem)
+              gem name.to_s, version
+            else
+              puts "#{name} #{"(#{version})" if version} failed to activate" if @config.warn
+            end
+          end
+        end
+        
+        def dependencies
+          dependency_filter(@gemspec.dependencies, @gemset)
+        end
+        
+        def development_dependencies
+          dependency_filter(@gemspec.development_dependencies, @gemset)
+        end
+        
+        def gemset=(gemset)
+          if gemset
+            @gemset = gemset.to_sym
+        
+            @gemsets = @config.gemsets.reverse.collect { |config|
+              if config.is_a?(::String)
+                YAML::load(File.read(config)) rescue {}
+              elsif config.is_a?(::Hash)
+                config
+              end
+            }.inject({}) do |hash, config|
+              deep_merge(hash, symbolize_keys(config))
+            end
+            
+            @versions = (@gemsets[gemspec.name.to_sym] || {}).inject({}) do |hash, (key, value)|
+              if !value.is_a?(::Hash) && value
+                hash[key] = value
+              elsif key == @gemset
+                (value || {}).each { |k, v| hash[k] = v }
+              end
+              hash
+            end
+          else
+            @gemset = nil
+            @gemsets = nil
+            @versions = nil
+          end
+        end
+        
+        def gemset_names
+          (
+            [ :default ] +
+            @gemsets[gemspec.name.to_sym].inject([]) { |array, (key, value)|
+              array.push(key) if value.is_a?(::Hash) || value.nil?
+              array
+            }
+          ).uniq
+        end
+        
+        def gemspec(reload=false)
+          if @gemspec && !reload
+            @gemspec
+          else
+            data = YAML::load(File.read(@config.gemspec)) rescue {}
+            @gemspec = SimpleStruct.new(data)
+          end
+        end
+      
+        private
+      
+        def deep_merge(first, second)
+          merger = lambda do |key, v1, v2|
+            Hash === v1 && Hash === v2 ? v1.merge(v2, &merger) : v2
+          end
+          first.merge(second, &merger)
+        end
+        
+        def dependency_filter(dependencies, match)
+          (dependencies || []).inject([]) { |array, value|
+            if value.is_a?(::Hash)
+              array += value[match.to_s] if value[match.to_s]
+            else
+              array << value
+            end
+            array
+          }.uniq.collect(&:to_sym)
+        end
+        
+        def gemset_from_loaded_specs
+          if defined?(Gem)
+            Gem.loaded_specs.each do |name, spec|
+              if name == gemspec.name
+                return :default
+              elsif name[0..gemspec.name.length] == "#{gemspec.name}-"
+                return name[gemspec.name.length+1..-1].to_sym
+              end
+            end
+            :default
+          else
+            :none
+          end
+        end
+      
+        def symbolize_keys(hash)
+          return {} unless hash.is_a?(::Hash)
+          hash.inject({}) do |options, (key, value)|
+            value = symbolize_keys(value) if value.is_a?(::Hash)
+            options[(key.to_sym rescue key) || key] = value
+            options
+          end
+        end
+      end
+    end
+  end
+end

data/lib/also_migrate/migration.rb

@@ -0,0 +1,72 @@
+module AlsoMigrate
+  module Migration
+    
+    def self.included(base)
+      unless base.respond_to?(:method_missing_with_also_migrate)
+        base.extend ClassMethods
+        base.class_eval do
+          class <<self
+            alias_method :method_missing_without_also_migrate, :method_missing
+            alias_method :method_missing, :method_missing_with_also_migrate
+          end
+        end
+      end
+    end
+
+    module ClassMethods
+
+      def method_missing_with_also_migrate(method, *arguments, &block)
+        args = Marshal.load(Marshal.dump(arguments))
+        method_missing_without_also_migrate(method, *arguments, &block)
+
+        supported = [
+          :add_column, :add_index, :add_timestamps, :change_column,
+          :change_column_default, :change_table, :create_table,
+          :drop_table, :remove_column, :remove_columns,
+          :remove_timestamps, :rename_column, :rename_table
+        ]
+
+        if !args.empty? && supported.include?(method)
+          connection = ActiveRecord::Base.connection
+          table_name = ActiveRecord::Migrator.proper_table_name(args[0])
+          
+          # Find models
+          if ::AlsoMigrate.classes
+            ::AlsoMigrate.classes.uniq.each do |klass|
+              if klass.also_migrate_config
+                klass.also_migrate_config.each do |config|
+                  options = config[:options]
+                  tables = config[:tables]
+                  
+                  next unless config[:table_name] == table_name
+              
+                  # Don't change ignored columns
+                  options[:ignore].each do |column|
+                    next if args.include?(column) || args.include?(column.intern)
+                  end
+
+                  # Run migration
+                  config[:tables].each do |table|
+                    if method == :create_table
+                      ActiveRecord::Migrator::AlsoMigrate.create_tables(klass)
+                    elsif method == :add_index && !options[:indexes].nil?
+                      next
+                    elsif connection.table_exists?(table)
+                      args[0] = table
+                      args[1] = table if method == :rename_table
+                      begin
+                        connection.send(method, *args, &block)
+                      rescue Exception => e
+                        puts "(also_migrate warning) #{e.message}"
+                      end
+                    end
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/also_migrate/migrator.rb

@@ -0,0 +1,100 @@
+module AlsoMigrate
+  module Migrator
+    
+    def self.included(base)
+      unless base.respond_to?(:migrate_with_also_migrate)
+        base.send :include, InstanceMethods
+        base.class_eval do
+          alias_method :migrate_without_also_migrate, :migrate
+          alias_method :migrate, :migrate_with_also_migrate
+        end
+      end
+    end
+    
+    module InstanceMethods
+      
+      def migrate_with_also_migrate
+        if ::AlsoMigrate.classes
+          ::AlsoMigrate.classes.uniq.each do |klass|
+            if klass.respond_to?(:also_migrate_config)
+              AlsoMigrate.create_tables(klass)
+            end
+          end
+        end
+      rescue Exception => e
+        puts "AlsoMigrate error: #{e.message}"
+        puts e.backtrace.join("\n")
+      ensure
+        migrate_without_also_migrate
+      end
+      
+      module AlsoMigrate
+        class <<self
+        
+          def connection
+            ActiveRecord::Base.connection
+          end
+        
+          def create_tables(klass)
+            config = klass.also_migrate_config
+            return unless config
+            old_table = klass.table_name
+            config.each do |config|
+              options = config[:options]
+              config[:tables].each do |new_table|
+                if !connection.table_exists?(new_table) && connection.table_exists?(old_table)
+                  columns = connection.columns(old_table).collect(&:name)
+                  columns -= options[:subtract].collect(&:to_s)
+                  columns.collect! { |col| connection.quote_column_name(col) }
+                  indexes = options[:indexes]
+                  if indexes
+                    engine =
+                      if connection.class.to_s.include?('Mysql')
+                        'ENGINE=' + connection.select_one(<<-SQL)['Engine']
+                          SHOW TABLE STATUS
+                          WHERE Name = '#{old_table}'
+                        SQL
+                      end
+                    connection.execute(<<-SQL)
+                      CREATE TABLE #{new_table} #{engine}
+                      AS SELECT #{columns.join(',')}
+                      FROM #{old_table}
+                      WHERE false;
+                    SQL
+                    indexes.each do |column|
+                      connection.add_index(new_table, column)
+                    end
+                  else
+                    connection.execute(<<-SQL)
+                      CREATE TABLE #{new_table}
+                      LIKE #{old_table};
+                    SQL
+                  end
+                end
+                if connection.table_exists?(new_table)
+                  if options[:add] || options[:subtract]
+                    columns = connection.columns(new_table).collect(&:name)
+                  end
+                  if options[:add]
+                    options[:add].each do |column|
+                      unless columns.include?(column[0])
+                        connection.add_column(*([ new_table ] + column))
+                      end
+                    end
+                  end
+                  if options[:subtract]
+                    options[:subtract].each do |column|
+                      if columns.include?(column)
+                        connection.remove_column(new_table, column)
+                      end
+                    end
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/also_migrate.rb

@@ -0,0 +1,17 @@
+require File.dirname(__FILE__) + '/also_migrate/gems'
+
+$:.unshift File.dirname(__FILE__)
+
+require 'also_migrate/base'
+require 'also_migrate/migration'
+require 'also_migrate/migrator'
+
+module AlsoMigrate
+  class <<self
+    attr_accessor :classes
+  end
+end
+
+ActiveRecord::Base.send(:include, AlsoMigrate::Base)
+ActiveRecord::Migrator.send(:include, AlsoMigrate::Migrator)
+ActiveRecord::Migration.send(:include, AlsoMigrate::Migration)

data/rails/init.rb

@@ -0,0 +1 @@
+require File.expand_path(File.dirname(__FILE__) + '/../lib/also_migrate')

data/spec/Rakefile

@@ -0,0 +1,13 @@
+require File.expand_path(File.dirname(__FILE__) + '/../lib/also_migrate/gems')
+
+AlsoMigrate::Gems.require(:spec_rake)
+
+require 'active_wrapper/tasks'
+
+#begin
+  ActiveWrapper::Tasks.new(
+    :base => File.dirname(__FILE__),
+    :env => 'test'
+  )
+# rescue Exception
+# end

data/spec/also_migrate/gems_spec.rb

@@ -0,0 +1,249 @@
+require 'spec_helper'
+
+describe AlsoMigrate::Gems do
+  
+  before(:each) do
+    @old_config = AlsoMigrate::Gems.config
+    
+    AlsoMigrate::Gems.config.gemspec = "#{$root}/spec/fixtures/gemspec.yml"
+    AlsoMigrate::Gems.config.gemsets = [
+      "#{$root}/spec/fixtures/gemsets.yml"
+    ]
+    AlsoMigrate::Gems.config.warn = true
+    
+    AlsoMigrate::Gems.gemspec true
+    AlsoMigrate::Gems.gemset = nil
+  end
+  
+  after(:each) do
+    AlsoMigrate::Gems.config = @old_config
+  end
+  
+  describe :activate do
+    it "should activate gems" do
+      AlsoMigrate::Gems.stub!(:gem)
+      AlsoMigrate::Gems.should_receive(:gem).with('rspec', '=1.3.1')
+      AlsoMigrate::Gems.should_receive(:gem).with('rake', '=0.8.7')
+      AlsoMigrate::Gems.activate :rspec, 'rake'
+    end
+  end
+  
+  describe :gemset= do
+    before(:each) do
+      AlsoMigrate::Gems.config.gemsets = [
+        {
+          :name => {
+            :rake => '>0.8.6',
+            :default => {
+              :externals => '=1.0.2'
+            }
+          }
+        },
+        "#{$root}/spec/fixtures/gemsets.yml"
+      ]
+    end
+    
+    describe :default do
+      before(:each) do
+        AlsoMigrate::Gems.gemset = :default
+      end
+      
+      it "should set @gemset" do
+        AlsoMigrate::Gems.gemset.should == :default
+      end
+    
+      it "should set @gemsets" do
+        AlsoMigrate::Gems.gemsets.should == {
+          :name => {
+            :rake => ">0.8.6",
+            :default => {
+              :externals => '=1.0.2',
+              :mysql => "=2.8.1",
+              :rspec => "=1.3.1"
+            },
+            :rspec2 => {
+              :mysql2 => "=0.2.6",
+              :rspec => "=2.3.0"
+            },
+            :solo => nil
+          }
+        }
+      end
+    
+      it "should set Gems.versions" do
+        AlsoMigrate::Gems.versions.should == {
+          :externals => "=1.0.2",
+          :mysql => "=2.8.1",
+          :rake => ">0.8.6",
+          :rspec => "=1.3.1"
+        }
+      end
+      
+      it "should return proper values for Gems.dependencies" do
+        AlsoMigrate::Gems.dependencies.should == [ :rake, :mysql ]
+        AlsoMigrate::Gems.development_dependencies.should == []
+      end
+      
+      it "should return proper values for Gems.gemset_names" do
+        AlsoMigrate::Gems.gemset_names.should == [ :default, :rspec2, :solo ]
+      end
+    end
+    
+    describe :rspec2 do
+      before(:each) do
+        AlsoMigrate::Gems.gemset = "rspec2"
+      end
+      
+      it "should set @gemset" do
+        AlsoMigrate::Gems.gemset.should == :rspec2
+      end
+    
+      it "should set @gemsets" do
+        AlsoMigrate::Gems.gemsets.should == {
+          :name => {
+            :rake => ">0.8.6",
+            :default => {
+              :externals => '=1.0.2',
+              :mysql => "=2.8.1",
+              :rspec => "=1.3.1"
+            },
+            :rspec2 => {
+              :mysql2=>"=0.2.6",
+              :rspec => "=2.3.0"
+            },
+            :solo => nil
+          }
+        }
+      end
+    
+      it "should set Gems.versions" do
+        AlsoMigrate::Gems.versions.should == {
+          :mysql2 => "=0.2.6",
+          :rake => ">0.8.6",
+          :rspec => "=2.3.0"
+        }
+      end
+      
+      it "should return proper values for Gems.dependencies" do
+        AlsoMigrate::Gems.dependencies.should == [ :rake, :mysql2 ]
+        AlsoMigrate::Gems.development_dependencies.should == []
+      end
+      
+      it "should return proper values for Gems.gemset_names" do
+        AlsoMigrate::Gems.gemset_names.should == [ :default, :rspec2, :solo ]
+      end
+    end
+    
+    describe :solo do
+      before(:each) do
+        AlsoMigrate::Gems.gemset = :solo
+      end
+      
+      it "should set @gemset" do
+        AlsoMigrate::Gems.gemset.should == :solo
+      end
+    
+      it "should set @gemsets" do
+        AlsoMigrate::Gems.gemsets.should == {
+          :name => {
+            :rake => ">0.8.6",
+            :default => {
+              :externals => '=1.0.2',
+              :mysql => "=2.8.1",
+              :rspec => "=1.3.1"
+            },
+            :rspec2 => {
+              :mysql2=>"=0.2.6",
+              :rspec => "=2.3.0"
+            },
+            :solo => nil
+          }
+        }
+      end
+    
+      it "should set Gems.versions" do
+        AlsoMigrate::Gems.versions.should == {:rake=>">0.8.6"}
+      end
+      
+      it "should return proper values for Gems.dependencies" do
+        AlsoMigrate::Gems.dependencies.should == [:rake]
+        AlsoMigrate::Gems.development_dependencies.should == []
+      end
+      
+      it "should return proper values for Gems.gemset_names" do
+        AlsoMigrate::Gems.gemset_names.should == [ :default, :rspec2, :solo ]
+      end
+    end
+    
+    describe :nil do
+      before(:each) do
+        AlsoMigrate::Gems.gemset = nil
+      end
+      
+      it "should set everything to nil" do
+        AlsoMigrate::Gems.gemset.should == nil
+        AlsoMigrate::Gems.gemsets.should == nil
+        AlsoMigrate::Gems.versions.should == nil
+      end
+    end
+  end
+  
+  describe :gemset_from_loaded_specs do
+    before(:each) do
+      Gem.stub!(:loaded_specs)
+    end
+    
+    it "should return the correct gemset for name gem" do
+      Gem.should_receive(:loaded_specs).and_return({ "name" => nil })
+      AlsoMigrate::Gems.send(:gemset_from_loaded_specs).should == :default
+    end
+    
+    it "should return the correct gemset for name-rspec gem" do
+      Gem.should_receive(:loaded_specs).and_return({ "name-rspec2" => nil })
+      AlsoMigrate::Gems.send(:gemset_from_loaded_specs).should == :rspec2
+    end
+  end
+  
+  describe :reload_gemspec do
+    it "should populate @gemspec" do
+      AlsoMigrate::Gems.gemspec.hash.should == {
+        "name" => "name",
+        "version" => "0.1.0",
+        "authors" => ["Author"],
+        "email" => "email@email.com",
+        "homepage" => "http://github.com/author/name",
+        "summary" => "Summary",
+        "description" => "Description",
+        "dependencies" => [
+          "rake",
+          { "default" => [ "mysql" ] },
+          { "rspec2" => [ "mysql2" ] }
+        ],
+        "development_dependencies" => nil
+       }
+    end
+  
+    it "should create methods from keys of @gemspec" do
+      AlsoMigrate::Gems.gemspec.name.should == "name"
+      AlsoMigrate::Gems.gemspec.version.should == "0.1.0"
+      AlsoMigrate::Gems.gemspec.authors.should == ["Author"]
+      AlsoMigrate::Gems.gemspec.email.should == "email@email.com"
+      AlsoMigrate::Gems.gemspec.homepage.should == "http://github.com/author/name"
+      AlsoMigrate::Gems.gemspec.summary.should == "Summary"
+      AlsoMigrate::Gems.gemspec.description.should == "Description"
+      AlsoMigrate::Gems.gemspec.dependencies.should == [
+        "rake",
+        { "default" => ["mysql"] },
+        { "rspec2" => [ "mysql2" ] }
+      ]
+      AlsoMigrate::Gems.gemspec.development_dependencies.should == nil
+    end
+  
+    it "should produce a valid gemspec" do
+      AlsoMigrate::Gems.gemset = :default
+      gemspec = File.expand_path("../../../also_migrate.gemspec", __FILE__)
+      gemspec = eval(File.read(gemspec), binding, gemspec)
+      gemspec.validate.should == true
+    end
+  end
+end

data/spec/also_migrate_spec.rb

@@ -0,0 +1,159 @@
+require "spec_helper"
+
+describe AlsoMigrate do
+  
+  [ "table doesn't exist yet", "table already exists" ].each do |description|
+    describe description do
+      describe 'with all options' do
+        
+        before(:each) do
+          reset_fixture
+        
+          if description == "table doesn't exist yet"
+            Article.also_migrate(
+              :article_archives,
+              :add => [
+                [ 'deleted_at', :datetime ]
+              ],
+              :subtract => 'restored_at',
+              :ignore => 'body',
+              :indexes => 'id'
+            )
+          end
+        
+          $db.migrate(1)
+          $db.migrate(0)
+          $db.migrate(1)
+        
+          if description == "table already exists"
+            Article.also_migrate(
+              :article_archives,
+              :add => [
+                [ 'deleted_at', :datetime ]
+              ],
+              :subtract => %w(restored_at),
+              :ignore => %w(body),
+              :indexes => %w(id)
+            )
+            $db.migrate(1)
+          end
+        end
+        
+        it "should create the add column" do
+          (columns('article_archives') - columns('articles')).should == [ 'deleted_at' ]
+        end
+      
+        it "should not create the subtract column" do
+          (columns('articles') - columns('article_archives')).should == [ 'restored_at' ]
+        end
+    
+        it 'should migrate both tables up' do
+          migrate_with_state(2)
+          (@new_article_columns - @old_article_columns).should == [ 'permalink' ]
+          (@new_archive_columns - @old_archive_columns).should == [ 'permalink' ]
+        end
+      
+        it 'should migrate both tables down' do
+          $db.migrate(2)
+          migrate_with_state(1)
+          (@old_article_columns - @new_article_columns).should == [ 'permalink' ]
+          (@old_archive_columns - @new_archive_columns).should == [ 'permalink' ]
+        end
+    
+        it "should ignore the body column" do
+          (columns('article_archives') - columns('articles')).should == [ 'deleted_at' ]
+          connection.remove_column(:articles, :body)
+          (columns('article_archives') - columns('articles')).should == [ 'body', 'deleted_at' ]
+        end
+      
+        it "should only add an index for id" do
+          indexed_columns('articles').should == [ 'id', 'read' ]
+          indexed_columns('article_archives').should == [ 'id' ]
+        end
+      end
+      
+      describe 'with no index option' do
+      
+        before(:each) do
+          reset_fixture
+          
+          if description == "table doesn't exist yet"
+            Article.also_migrate :article_archives
+          end
+      
+          $db.migrate(0)
+          $db.migrate(1)
+      
+          if description == "table already exists"
+            Article.also_migrate :article_archives
+            $db.migrate(1)
+          end
+        end
+      
+        it "should add all indexes" do
+          indexed_columns('articles').should == [ 'id', 'read' ]
+          indexed_columns('article_archives').should == [ 'id', 'read' ]
+        end
+      end
+      
+      describe "with other table" do
+      
+        before(:each) do
+          reset_fixture
+          
+          if description == "table doesn't exist yet"
+            Article.also_migrate :article_archives
+            Comment.also_migrate :comment_archives
+          end
+      
+          $db.migrate(0)
+          $db.migrate(1)
+          $db.migrate(2)
+          $db.migrate(3)
+      
+          if description == "table already exists"
+            Article.also_migrate :article_archives
+            Comment.also_migrate :comment_archives
+            $db.migrate(3)
+          end
+        end
+      
+        it "should not affect other table" do
+          columns('articles').should == columns('article_archives')
+          columns('comments').should == columns('comment_archives')
+          columns('articles').should == ["id", "title", "body", "read", "restored_at", "permalink"]
+          columns('comments').should == ["id", "header", "description"]
+        end
+      end
+      
+      if description == "table already exists"
+        describe 'with add and subtract option' do
+        
+          before(:each) do
+            reset_fixture
+          
+            Article.also_migrate :article_archives
+      
+            $db.migrate(0)
+            $db.migrate(1)
+          
+            Article.also_migrate_config = nil
+            Article.also_migrate(
+              :article_archives,
+              :add => [
+                [ 'deleted_at', :datetime ]
+              ],
+              :subtract => 'restored_at'
+            )
+          end
+        
+          it "should add and remove fields" do
+            columns('article_archives').should == %w(id title body read restored_at)
+            $db.migrate(1)
+            columns('article_archives').should == %w(id title body read deleted_at)
+          end
+        end
+      end
+    end
+  end
+end

data/spec/config/database.yml.example

@@ -0,0 +1,6 @@
+test:
+  adapter: mysql
+  database: also_migrate
+  username: root
+  password: 
+  host: localhost

data/spec/db/migrate/001_create_articles.rb

@@ -0,0 +1,15 @@
+class CreateArticles < ActiveRecord::Migration
+  def self.up
+    create_table :articles do |t|
+      t.string :title
+      t.string :body
+      t.boolean :read
+      t.datetime :restored_at
+    end
+    add_index :articles, :read
+  end
+
+  def self.down
+    drop_table :articles
+  end
+end

data/spec/db/migrate/002_add_permalink.rb

@@ -0,0 +1,9 @@
+class AddPermalink < ActiveRecord::Migration
+  def self.up
+    add_column :articles, :permalink, :string
+  end
+
+  def self.down
+    remove_column :articles, :permalink
+  end
+end

data/spec/db/migrate/003_create_comments.rb

@@ -0,0 +1,12 @@
+class CreateComments < ActiveRecord::Migration
+  def self.up
+    create_table :comments do |t|
+      t.string :header
+      t.string :description
+    end
+  end
+
+  def self.down
+    drop_table :comments
+  end
+end

data/spec/fixtures/article.rb

@@ -0,0 +1,2 @@
+class Article < ActiveRecord::Base
+end

data/spec/fixtures/comment.rb

@@ -0,0 +1,2 @@
+class Comment < ActiveRecord::Base
+end

data/spec/fixtures/gemsets.yml

@@ -0,0 +1,9 @@
+name:
+  rake: =0.8.7
+  default:
+    mysql: =2.8.1
+    rspec: =1.3.1
+  rspec2:
+    mysql2: =0.2.6
+    rspec: =2.3.0
+  solo: null

data/spec/fixtures/gemspec.yml

@@ -0,0 +1,15 @@
+name: name
+version: 0.1.0
+authors:
+  - Author
+email: email@email.com
+homepage: http://github.com/author/name
+summary: Summary
+description: Description
+dependencies:
+  - rake
+  - default:
+    - mysql
+  - rspec2:
+    - mysql2
+development_dependencies: null

data/spec/spec_helper.rb

@@ -0,0 +1,102 @@
+require 'pp'
+
+$root = File.expand_path('../../', __FILE__)
+require "#{$root}/lib/also_migrate/gems"
+
+AlsoMigrate::Gems.activate :active_wrapper, :rspec
+
+require 'active_wrapper'
+
+require "#{$root}/lib/also_migrate"
+require "#{$root}/spec/fixtures/article"
+require "#{$root}/spec/fixtures/comment"
+require 'pp'
+
+Spec::Runner.configure do |config|
+end
+
+$db, $log, $mail = ActiveWrapper.setup(
+  :base => File.dirname(__FILE__),
+  :env => 'test'
+)
+$db.establish_connection
+
+def columns(table)
+  connection.columns(table).collect(&:name)
+end
+
+def connection
+  ActiveRecord::Base.connection
+end
+
+# For use with rspec textmate bundle
+def debug(object)
+  puts "<pre>"
+  puts object.pretty_inspect.gsub('<', '&lt;').gsub('>', '&gt;')
+  puts "</pre>"
+end
+
+def indexed_columns(table_name)
+  # MySQL
+  if connection.class.to_s.include?('Mysql')
+    index_query = "SHOW INDEX FROM #{table_name}"
+    connection.select_all(index_query).collect do |r|
+      r["Column_name"]
+    end
+  # PostgreSQL
+  # http://stackoverflow.com/questions/2204058/show-which-columns-an-index-is-on-in-postgresql/2213199
+  elsif connection.class.to_s.include?('PostgreSQL')
+    index_query = <<-SQL
+      select
+        t.relname as table_name,
+        i.relname as index_name,
+        a.attname as column_name
+      from
+        pg_class t,
+        pg_class i,
+        pg_index ix,
+        pg_attribute a
+      where
+        t.oid = ix.indrelid
+        and i.oid = ix.indexrelid
+        and a.attrelid = t.oid
+        and a.attnum = ANY(ix.indkey)
+        and t.relkind = 'r'
+        and t.relname = '#{table_name}'
+      order by
+        t.relname,
+        i.relname
+    SQL
+    connection.select_all(index_query).collect do |r|
+      r["column_name"]
+    end
+  else
+    raise 'AlsoMigrate does not support this database adapter'
+  end
+end
+
+def migrate_with_state(version)
+  @old_article_columns = columns("articles")
+  @old_archive_columns = columns("article_archives")
+  $db.migrate(version)
+  @new_article_columns = columns("articles")
+  @new_archive_columns = columns("article_archives")
+end
+
+def reset_fixture
+  if Article.respond_to?(:also_migrate_config)
+    Article.also_migrate_config = nil
+  end
+  
+  if Comment.respond_to?(:also_migrate_config)
+    Comment.also_migrate_config = nil
+  end
+  
+  if connection.table_exists?('article_archives')
+    connection.execute('DROP TABLE article_archives')
+  end
+  
+  if connection.table_exists?('comment_archives')
+    connection.execute('DROP TABLE comment_archives')
+  end
+end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 3
-  - 0
-  version: 0.3.0
+  - 2
+  version: 0.3.2
 platform: ruby
 authors: 
 - Winton Welsh
@@ -54,8 +54,33 @@ extensions: []
 
 extra_rdoc_files: []
 
-files: []
-
+files: 
+- .gitignore
+- LICENSE
+- README.md
+- Rakefile
+- also_migrate.gemspec
+- config/gemsets.yml
+- config/gemspec.yml
+- init.rb
+- lib/also_migrate.rb
+- lib/also_migrate/base.rb
+- lib/also_migrate/gems.rb
+- lib/also_migrate/migration.rb
+- lib/also_migrate/migrator.rb
+- rails/init.rb
+- spec/Rakefile
+- spec/also_migrate/gems_spec.rb
+- spec/also_migrate_spec.rb
+- spec/config/database.yml.example
+- spec/db/migrate/001_create_articles.rb
+- spec/db/migrate/002_add_permalink.rb
+- spec/db/migrate/003_create_comments.rb
+- spec/fixtures/article.rb
+- spec/fixtures/comment.rb
+- spec/fixtures/gemsets.yml
+- spec/fixtures/gemspec.yml
+- spec/spec_helper.rb
 has_rdoc: true
 homepage: http://github.com/winton/also_migrate
 licenses: []
@@ -88,5 +113,16 @@ rubygems_version: 1.3.7
 signing_key: 
 specification_version: 3
 summary: Migrate multiple tables with similar schema at once.
-test_files: []
-
+test_files: 
+- spec/Rakefile
+- spec/also_migrate/gems_spec.rb
+- spec/also_migrate_spec.rb
+- spec/config/database.yml.example
+- spec/db/migrate/001_create_articles.rb
+- spec/db/migrate/002_add_permalink.rb
+- spec/db/migrate/003_create_comments.rb
+- spec/fixtures/article.rb
+- spec/fixtures/comment.rb
+- spec/fixtures/gemsets.yml
+- spec/fixtures/gemspec.yml
+- spec/spec_helper.rb