Chris@0: require 'test/unit'
Chris@0: 
Chris@0: require 'rubygems'
Chris@0: gem 'activerecord', '>= 1.15.4.7794'
Chris@0: require 'active_record'
Chris@0: 
Chris@0: require "#{File.dirname(__FILE__)}/../init"
Chris@0: 
Chris@0: ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :dbfile => ":memory:")
Chris@0: 
Chris@0: def setup_db
Chris@0:   ActiveRecord::Schema.define(:version => 1) do
Chris@0:     create_table :mixins do |t|
Chris@0:       t.column :pos, :integer
Chris@0:       t.column :parent_id, :integer
Chris@0:       t.column :created_at, :datetime      
Chris@0:       t.column :updated_at, :datetime
Chris@0:     end
Chris@0:   end
Chris@0: end
Chris@0: 
Chris@0: def teardown_db
Chris@0:   ActiveRecord::Base.connection.tables.each do |table|
Chris@0:     ActiveRecord::Base.connection.drop_table(table)
Chris@0:   end
Chris@0: end
Chris@0: 
Chris@0: class Mixin < ActiveRecord::Base
Chris@0: end
Chris@0: 
Chris@0: class ListMixin < Mixin
Chris@0:   acts_as_list :column => "pos", :scope => :parent
Chris@0: 
Chris@0:   def self.table_name() "mixins" end
Chris@0: end
Chris@0: 
Chris@0: class ListMixinSub1 < ListMixin
Chris@0: end
Chris@0: 
Chris@0: class ListMixinSub2 < ListMixin
Chris@0: end
Chris@0: 
Chris@0: class ListWithStringScopeMixin < ActiveRecord::Base
Chris@0:   acts_as_list :column => "pos", :scope => 'parent_id = #{parent_id}'
Chris@0: 
Chris@0:   def self.table_name() "mixins" end
Chris@0: end
Chris@0: 
Chris@0: 
Chris@0: class ListTest < Test::Unit::TestCase
Chris@0: 
Chris@0:   def setup
Chris@0:     setup_db
Chris@0:     (1..4).each { |counter| ListMixin.create! :pos => counter, :parent_id => 5 }
Chris@0:   end
Chris@0: 
Chris@0:   def teardown
Chris@0:     teardown_db
Chris@0:   end
Chris@0: 
Chris@0:   def test_reordering
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).move_lower
Chris@0:     assert_equal [1, 3, 2, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).move_higher
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(1).move_to_bottom
Chris@0:     assert_equal [2, 3, 4, 1], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(1).move_to_top
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).move_to_bottom
Chris@0:     assert_equal [1, 3, 4, 2], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(4).move_to_top
Chris@0:     assert_equal [4, 1, 3, 2], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0:   end
Chris@0: 
Chris@0:   def test_move_to_bottom_with_next_to_last_item
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0:     ListMixin.find(3).move_to_bottom
Chris@0:     assert_equal [1, 2, 4, 3], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0:   end
Chris@0: 
Chris@0:   def test_next_prev
Chris@0:     assert_equal ListMixin.find(2), ListMixin.find(1).lower_item
Chris@0:     assert_nil ListMixin.find(1).higher_item
Chris@0:     assert_equal ListMixin.find(3), ListMixin.find(4).higher_item
Chris@0:     assert_nil ListMixin.find(4).lower_item
Chris@0:   end
Chris@0: 
Chris@0:   def test_injection
Chris@0:     item = ListMixin.new(:parent_id => 1)
Chris@0:     assert_equal "parent_id = 1", item.scope_condition
Chris@0:     assert_equal "pos", item.position_column
Chris@0:   end
Chris@0: 
Chris@0:   def test_insert
Chris@0:     new = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 1, new.pos
Chris@0:     assert new.first?
Chris@0:     assert new.last?
Chris@0: 
Chris@0:     new = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 2, new.pos
Chris@0:     assert !new.first?
Chris@0:     assert new.last?
Chris@0: 
Chris@0:     new = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 3, new.pos
Chris@0:     assert !new.first?
Chris@0:     assert new.last?
Chris@0: 
Chris@0:     new = ListMixin.create(:parent_id => 0)
Chris@0:     assert_equal 1, new.pos
Chris@0:     assert new.first?
Chris@0:     assert new.last?
Chris@0:   end
Chris@0: 
Chris@0:   def test_insert_at
Chris@0:     new = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 1, new.pos
Chris@0: 
Chris@0:     new = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 2, new.pos
Chris@0: 
Chris@0:     new = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 3, new.pos
Chris@0: 
Chris@0:     new4 = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 4, new4.pos
Chris@0: 
Chris@0:     new4.insert_at(3)
Chris@0:     assert_equal 3, new4.pos
Chris@0: 
Chris@0:     new.reload
Chris@0:     assert_equal 4, new.pos
Chris@0: 
Chris@0:     new.insert_at(2)
Chris@0:     assert_equal 2, new.pos
Chris@0: 
Chris@0:     new4.reload
Chris@0:     assert_equal 4, new4.pos
Chris@0: 
Chris@0:     new5 = ListMixin.create(:parent_id => 20)
Chris@0:     assert_equal 5, new5.pos
Chris@0: 
Chris@0:     new5.insert_at(1)
Chris@0:     assert_equal 1, new5.pos
Chris@0: 
Chris@0:     new4.reload
Chris@0:     assert_equal 5, new4.pos
Chris@0:   end
Chris@0: 
Chris@0:   def test_delete_middle
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).destroy
Chris@0: 
Chris@0:     assert_equal [1, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     assert_equal 1, ListMixin.find(1).pos
Chris@0:     assert_equal 2, ListMixin.find(3).pos
Chris@0:     assert_equal 3, ListMixin.find(4).pos
Chris@0: 
Chris@0:     ListMixin.find(1).destroy
Chris@0: 
Chris@0:     assert_equal [3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     assert_equal 1, ListMixin.find(3).pos
Chris@0:     assert_equal 2, ListMixin.find(4).pos
Chris@0:   end
Chris@0: 
Chris@0:   def test_with_string_based_scope
Chris@0:     new = ListWithStringScopeMixin.create(:parent_id => 500)
Chris@0:     assert_equal 1, new.pos
Chris@0:     assert new.first?
Chris@0:     assert new.last?
Chris@0:   end
Chris@0: 
Chris@0:   def test_nil_scope
Chris@0:     new1, new2, new3 = ListMixin.create, ListMixin.create, ListMixin.create
Chris@0:     new2.move_higher
Chris@0:     assert_equal [new2, new1, new3], ListMixin.find(:all, :conditions => 'parent_id IS NULL', :order => 'pos')
Chris@0:   end
Chris@0:   
Chris@0:   
Chris@0:   def test_remove_from_list_should_then_fail_in_list? 
Chris@0:     assert_equal true, ListMixin.find(1).in_list?
Chris@0:     ListMixin.find(1).remove_from_list
Chris@0:     assert_equal false, ListMixin.find(1).in_list?
Chris@0:   end 
Chris@0:   
Chris@0:   def test_remove_from_list_should_set_position_to_nil 
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0:   
Chris@0:     ListMixin.find(2).remove_from_list 
Chris@0:   
Chris@0:     assert_equal [2, 1, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0:   
Chris@0:     assert_equal 1,   ListMixin.find(1).pos
Chris@0:     assert_equal nil, ListMixin.find(2).pos
Chris@0:     assert_equal 2,   ListMixin.find(3).pos
Chris@0:     assert_equal 3,   ListMixin.find(4).pos
Chris@0:   end 
Chris@0:   
Chris@0:   def test_remove_before_destroy_does_not_shift_lower_items_twice 
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0:   
Chris@0:     ListMixin.find(2).remove_from_list 
Chris@0:     ListMixin.find(2).destroy 
Chris@0:   
Chris@0:     assert_equal [1, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5', :order => 'pos').map(&:id)
Chris@0:   
Chris@0:     assert_equal 1, ListMixin.find(1).pos
Chris@0:     assert_equal 2, ListMixin.find(3).pos
Chris@0:     assert_equal 3, ListMixin.find(4).pos
Chris@0:   end 
Chris@0:   
Chris@0: end
Chris@0: 
Chris@0: class ListSubTest < Test::Unit::TestCase
Chris@0: 
Chris@0:   def setup
Chris@0:     setup_db
Chris@0:     (1..4).each { |i| ((i % 2 == 1) ? ListMixinSub1 : ListMixinSub2).create! :pos => i, :parent_id => 5000 }
Chris@0:   end
Chris@0: 
Chris@0:   def teardown
Chris@0:     teardown_db
Chris@0:   end
Chris@0: 
Chris@0:   def test_reordering
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).move_lower
Chris@0:     assert_equal [1, 3, 2, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).move_higher
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(1).move_to_bottom
Chris@0:     assert_equal [2, 3, 4, 1], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(1).move_to_top
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).move_to_bottom
Chris@0:     assert_equal [1, 3, 4, 2], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(4).move_to_top
Chris@0:     assert_equal [4, 1, 3, 2], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0:   end
Chris@0: 
Chris@0:   def test_move_to_bottom_with_next_to_last_item
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0:     ListMixin.find(3).move_to_bottom
Chris@0:     assert_equal [1, 2, 4, 3], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0:   end
Chris@0: 
Chris@0:   def test_next_prev
Chris@0:     assert_equal ListMixin.find(2), ListMixin.find(1).lower_item
Chris@0:     assert_nil ListMixin.find(1).higher_item
Chris@0:     assert_equal ListMixin.find(3), ListMixin.find(4).higher_item
Chris@0:     assert_nil ListMixin.find(4).lower_item
Chris@0:   end
Chris@0: 
Chris@0:   def test_injection
Chris@0:     item = ListMixin.new("parent_id"=>1)
Chris@0:     assert_equal "parent_id = 1", item.scope_condition
Chris@0:     assert_equal "pos", item.position_column
Chris@0:   end
Chris@0: 
Chris@0:   def test_insert_at
Chris@0:     new = ListMixin.create("parent_id" => 20)
Chris@0:     assert_equal 1, new.pos
Chris@0: 
Chris@0:     new = ListMixinSub1.create("parent_id" => 20)
Chris@0:     assert_equal 2, new.pos
Chris@0: 
Chris@0:     new = ListMixinSub2.create("parent_id" => 20)
Chris@0:     assert_equal 3, new.pos
Chris@0: 
Chris@0:     new4 = ListMixin.create("parent_id" => 20)
Chris@0:     assert_equal 4, new4.pos
Chris@0: 
Chris@0:     new4.insert_at(3)
Chris@0:     assert_equal 3, new4.pos
Chris@0: 
Chris@0:     new.reload
Chris@0:     assert_equal 4, new.pos
Chris@0: 
Chris@0:     new.insert_at(2)
Chris@0:     assert_equal 2, new.pos
Chris@0: 
Chris@0:     new4.reload
Chris@0:     assert_equal 4, new4.pos
Chris@0: 
Chris@0:     new5 = ListMixinSub1.create("parent_id" => 20)
Chris@0:     assert_equal 5, new5.pos
Chris@0: 
Chris@0:     new5.insert_at(1)
Chris@0:     assert_equal 1, new5.pos
Chris@0: 
Chris@0:     new4.reload
Chris@0:     assert_equal 5, new4.pos
Chris@0:   end
Chris@0: 
Chris@0:   def test_delete_middle
Chris@0:     assert_equal [1, 2, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     ListMixin.find(2).destroy
Chris@0: 
Chris@0:     assert_equal [1, 3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     assert_equal 1, ListMixin.find(1).pos
Chris@0:     assert_equal 2, ListMixin.find(3).pos
Chris@0:     assert_equal 3, ListMixin.find(4).pos
Chris@0: 
Chris@0:     ListMixin.find(1).destroy
Chris@0: 
Chris@0:     assert_equal [3, 4], ListMixin.find(:all, :conditions => 'parent_id = 5000', :order => 'pos').map(&:id)
Chris@0: 
Chris@0:     assert_equal 1, ListMixin.find(3).pos
Chris@0:     assert_equal 2, ListMixin.find(4).pos
Chris@0:   end
Chris@0: 
Chris@0: end