44 lines
1.6 KiB
Ruby
44 lines
1.6 KiB
Ruby
require 'test_helper'
|
|
|
|
class InheritanceColumnTest < ActiveSupport::TestCase
|
|
|
|
context 'STI models' do
|
|
setup do
|
|
@animal = Animal.create :name => 'Animal'
|
|
@animal.update_attributes :name => 'Animal from the Muppets'
|
|
@animal.update_attributes :name => 'Animal Muppet'
|
|
@animal.destroy
|
|
|
|
@dog = Dog.create :name => 'Snoopy'
|
|
@dog.update_attributes :name => 'Scooby'
|
|
@dog.update_attributes :name => 'Scooby Doo'
|
|
@dog.destroy
|
|
|
|
@cat = Cat.create :name => 'Garfield'
|
|
@cat.update_attributes :name => 'Garfield (I hate Mondays)'
|
|
@cat.update_attributes :name => 'Garfield The Cat'
|
|
@cat.destroy
|
|
end
|
|
|
|
should 'work with custom STI inheritance column' do
|
|
assert_equal 12, PaperTrail::Version.count
|
|
assert_equal 4, @animal.versions.count
|
|
assert_nil @animal.versions.first.reify
|
|
@animal.versions[1..-1].each { |v| assert_equal 'Animal', v.reify.class.name }
|
|
|
|
# For some reason `@dog.versions` doesn't include the final `destroy` version.
|
|
# Neither do `@dog.versions.scoped` nor `@dog.versions(true)` nor `@dog.versions.reload`.
|
|
dog_versions = PaperTrail::Version.where(:item_id => @dog.id)
|
|
assert_equal 4, dog_versions.count
|
|
assert_nil dog_versions.first.reify
|
|
dog_versions[1..-1].each { |v| assert_equal 'Dog', v.reify.class.name }
|
|
|
|
cat_versions = PaperTrail::Version.where(:item_id => @cat.id)
|
|
assert_equal 4, cat_versions.count
|
|
assert_nil cat_versions.first.reify
|
|
cat_versions[1..-1].each { |v| assert_equal 'Cat', v.reify.class.name }
|
|
end
|
|
end
|
|
|
|
end
|