mover 0.1.0

data/spec/mover/migrator_spec.rb

@@ -0,0 +1,33 @@
+require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
+
+describe Mover::Migrator do
+  
+  before(:each) do
+    $db.migrate(0)
+    $db.migrate(1)
+  end
+  
+  describe :method_missing_with_mover do
+    
+    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 not touch the archive's move_id or moved_at column" do
+      connection.add_column(:articles, :move_id, :integer)
+      connection.add_column(:articles, :moved_at, :datetime)
+      migrate_with_state(3)
+      (@old_article_columns - @new_article_columns).should == [ 'move_id', 'moved_at' ]
+      (@old_archive_columns - @new_archive_columns).should == []
+    end
+  end
+end

data/spec/mover/record_spec.rb

@@ -0,0 +1,155 @@
+require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
+
+describe Mover::Base::Record do
+  
+  before(:all) do
+    $db.migrate(0)
+    $db.migrate(1)
+  end
+  
+  describe :InstanceMethods do
+    describe :move_to do
+  
+      before(:all) do
+        @articles = create_records
+        @comments = create_records(Comment)
+        @articles[0..1].each do |a|
+          a.move_to(:archived)
+        end
+      end
+  
+      it "should move some records to the archive table" do
+        Article.count.should == 3
+        ArchivedArticle.count.should == 2
+      end
+  
+      it "should preserve record attributes" do
+        2.times do |x|
+          original = @articles[x]
+          copy = ArchivedArticle.find(original.id)
+          record_match?(original, copy)
+        end
+      end
+    
+      it "should move associated records" do
+        Comment.count.should == 3
+        ArchivedComment.count.should == 2
+      end
+    
+      it "should preserve associated record attributes" do
+        2.times do |x|
+          original = @comments[x]
+          copy = ArchivedComment.find(original.id)
+          record_match?(original, copy)
+        end
+      end
+    
+      it "should populate move_id" do
+        (1..2).each do |x|
+          article = ArchivedArticle.find(x)
+          comment = ArchivedComment.find(x)
+          comment.move_id.nil?.should == false
+          comment.move_id.length.should == 32
+          comment.move_id.should == article.move_id
+        end
+      end
+    
+      it "should populate moved_at" do
+        (1..2).each do |x|
+          article = ArchivedArticle.find(x)
+          comment = ArchivedComment.find(x)
+          comment.moved_at.nil?.should == false
+          comment.moved_at.should == article.moved_at
+        end
+      end
+    end
+  
+    describe :move_from do
+    
+      before(:all) do
+        articles = create_records
+        create_records(Comment)
+        articles[0..1].each do |a|
+          a.move_to(:archived)
+        end
+        @articles = ArchivedArticle.find(1, 2)
+        @comments = ArchivedComment.find(1, 2)
+        @articles.each do |article|
+          article.move_from
+        end
+      end
+    
+      it "should move records back to the original table" do
+        Article.count.should == 5
+        ArchivedArticle.count.should == 0
+      end
+  
+      it "should preserve record attributes" do
+        2.times do |x|
+          original = @articles[x]
+          copy = Article.find(original.id)
+          record_match?(original, copy)
+        end
+      end
+    
+      it "should move associated records" do
+        Comment.count.should == 5
+        ArchivedComment.count.should == 0
+      end
+    
+      it "should preserve associated record attributes" do
+        2.times do |x|
+          original = @comments[x]
+          copy = Comment.find(original.id)
+          record_match?(original, copy)
+        end
+      end
+    end
+  end
+  
+  describe :ClassMethods do
+    describe :move_to do
+  
+      before(:all) do
+        create_records
+        create_records(Comment)
+        Article.move_to(:archived, [ 'id = ? OR id = ?', 1, 2 ])
+        Article.move_to(:drafted, [ 'id = ? OR id = ?', 3, 4 ])
+      end
+      
+      it "should move the records" do
+        Article.count.should == 1
+        ArchivedArticle.count.should == 2
+        DraftedArticle.count.should == 2
+      end
+      
+      it "should move associated records" do
+        Comment.count.should == 3
+        ArchivedComment.count.should == 2
+      end
+    end
+    
+    describe :move_from do
+  
+      before(:all) do
+        create_records
+        create_records(Comment)
+        Article.move_to(:archived, [ 'id = ? OR id = ?', 1, 2 ])
+        Article.move_to(:drafted, [ 'id = ? OR id = ?', 3, 4 ])
+        Article.move_from(:archived, [ 'id = ? OR id = ?', 1, 2 ])
+        Article.move_from(:drafted, [ 'id = ? OR id = ?', 3, 4 ])
+      end
+      
+      it "should move the records" do
+        Article.count.should == 5
+        ArchivedArticle.count.should == 0
+        DraftedArticle.count.should == 0
+      end
+      
+      it "should move associated records" do
+        Comment.count.should == 5
+        ArchivedComment.count.should == 0
+      end
+    end
+  end
+end

data/spec/mover/table_spec.rb

@@ -0,0 +1,79 @@
+require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
+
+describe Mover::Base::Table do
+
+  before(:all) do
+    $db.migrate(0)
+    $db.migrate(1)
+  end
+
+  describe :create_movable_table do
+
+    before(:all) do
+      @article_columns = connection.columns("articles").collect(&:name)
+      @archive_columns = connection.columns("archived_articles").collect(&:name)
+    end
+
+    it "should create an archive table" do
+      connection.table_exists?("archived_articles").should == true
+    end
+
+    it "should create an archive table with the same structure as the original table" do
+      @article_columns.each do |col|
+        @archive_columns.include?(col).should == true
+      end
+    end
+    
+    describe 'with options' do
+      
+      before(:all) do
+        Article.drop_movable_table(:archived)
+        Article.create_movable_table(
+          :archived,
+          :columns => %w(id read),
+          :indexes => %w(read)
+        )
+        @archive_columns = connection.columns("archived_articles").collect(&:name)
+      end
+      
+      after(:all) do
+        Article.drop_movable_table(:archived)
+        Article.create_movable_table(:archived)
+      end
+      
+      it "should create the correct columns" do
+        @archive_columns.length.should == 2
+        %w(id read).each do |col|
+          @archive_columns.include?(col).should == true
+        end
+      end
+      
+      it "should create archive indexes" do
+        indexes = Article.send(:indexed_columns, 'archived_articles')
+        indexes.to_set.should == [ "read" ].to_set
+      end
+    end
+    
+    describe 'without options' do
+      
+      it "should create archive indexes" do
+        indexes = Article.send(:indexed_columns, 'archived_articles')
+        indexes.to_set.should == [ "id", "title" ].to_set
+      end
+    end
+  end
+  
+  describe :drop_movable_table do
+    
+    it "should drop the table" do
+      Article.drop_movable_table(:archived)
+      output = connection.execute(<<-SQL)
+        SELECT COUNT(*)
+        FROM information_schema.tables 
+        WHERE table_schema = '#{Article.configurations['test']['database']}' 
+        AND table_name = 'archived_articles';
+      SQL
+      output.fetch_row.should == ['0']
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,48 @@
+require File.expand_path("#{File.dirname(__FILE__)}/../require")
+Require.spec_helper!
+
+Spec::Runner.configure do |config|
+end
+
+$db, $log = ActiveWrapper.setup(
+  :base => File.dirname(__FILE__),
+  :env => 'test'
+)
+$db.establish_connection
+
+def record_match?(original, copy)
+  (original.class.column_names & copy.class.column_names).each do |col|
+    copy.send(col).should == original.send(col)
+  end
+end
+
+def columns(table)
+  connection.columns(table).collect(&:name)
+end
+
+def connection
+  ActiveRecord::Base.connection
+end
+
+def create_records(klass=Article, values={})
+  klass.delete_all
+  (1..5).collect do |x|
+    klass.column_names.each do |column|
+      if column == 'article_id'
+        values[:article_id] = x
+      else
+        values[column.intern] = "#{klass} #{x} #{column}"
+      end
+    end
+    values[:id] = x
+    klass.create(values)
+  end
+end
+
+def migrate_with_state(version)
+  @old_article_columns = columns("articles")
+  @old_archive_columns = columns("archived_articles")
+  $db.migrate(version)
+  @new_article_columns = columns("articles")
+  @new_archive_columns = columns("archived_articles")
+end

metadata

@@ -0,0 +1,86 @@
+--- !ruby/object:Gem::Specification 
+name: mover
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Winton Welsh
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-04-01 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: require
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 0.2.6
+    version: 
+description: 
+email: mail@wintoni.us
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+files: 
+- init.rb
+- lib/mover/migrator.rb
+- lib/mover/record.rb
+- lib/mover/table.rb
+- lib/mover.rb
+- log/development.log
+- MIT-LICENSE
+- rails/init.rb
+- Rakefile
+- README.markdown
+- require.rb
+- spec/config/database.yml.example
+- spec/db/migrate/001_create_articles.rb
+- spec/db/migrate/002_add_permalink.rb
+- spec/db/migrate/003_remove_magic_columns.rb
+- spec/fixtures/article.rb
+- spec/fixtures/comment.rb
+- spec/log/test.log
+- spec/mover/migrator_spec.rb
+- spec/mover/record_spec.rb
+- spec/mover/table_spec.rb
+- spec/Rakefile
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/winton/mover
+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: Move ActiveRecord records across tables like it ain't no thang
+test_files: []
+