111 lines
3.7 KiB
Ruby
111 lines
3.7 KiB
Ruby
require 'minitest/autorun'
|
|
require 'prism'
|
|
require 'tmpdir'
|
|
|
|
class OrmPolymorphicTouchTest < Minitest::Test
|
|
def setup
|
|
@dir = Dir.mktmpdir('prism-polymorphic-touch-store')
|
|
@store = Prism::Orm::Store.new(File.join(@dir, 'test.pstore'))
|
|
@store.create_table(:posts, %w[id title updated_at])
|
|
@store.create_table(:photos, %w[id caption updated_at reviewed_at])
|
|
@store.create_table(:comments, %w[id body commentable_id commentable_type updated_at])
|
|
|
|
post_class = Class.new(Prism::Orm::Model) do
|
|
table_name :posts
|
|
has_many :comments, class_name: 'Comment', as: :commentable, dependent: :destroy
|
|
end
|
|
photo_class = Class.new(Prism::Orm::Model) do
|
|
table_name :photos
|
|
has_many :comments, class_name: 'Comment', as: :commentable, dependent: :destroy
|
|
end
|
|
comment_class = Class.new(Prism::Orm::Model) do
|
|
table_name :comments
|
|
belongs_to :commentable, polymorphic: true, touch: true
|
|
end
|
|
[post_class, photo_class, comment_class].each { |klass| klass.store = @store }
|
|
Object.const_set(:Post, post_class)
|
|
Object.const_set(:Photo, photo_class)
|
|
Object.const_set(:Comment, comment_class)
|
|
end
|
|
|
|
def teardown
|
|
FileUtils.remove_entry(@dir)
|
|
%i[Post Photo Comment].each { |name| Object.send(:remove_const, name) if Object.const_defined?(name) }
|
|
end
|
|
|
|
def test_creating_a_child_touches_the_polymorphic_owner
|
|
post = Post.create!(title: 'hello')
|
|
original = post.updated_at
|
|
sleep 0.01
|
|
|
|
Comment.create!(body: 'first', commentable_id: post.id, commentable_type: 'Post')
|
|
|
|
assert post.reload.updated_at > original
|
|
end
|
|
|
|
def test_touches_the_correct_owner_type_when_two_types_share_ids
|
|
post = Post.create!(title: 'hello')
|
|
photo = Photo.create!(caption: 'sunset')
|
|
original_post = post.reload.updated_at
|
|
original_photo = photo.reload.updated_at
|
|
sleep 0.01
|
|
|
|
Comment.create!(body: 'on the photo', commentable_id: photo.id, commentable_type: 'Photo')
|
|
|
|
assert_equal original_post, post.reload.updated_at
|
|
assert photo.reload.updated_at > original_photo
|
|
end
|
|
|
|
def test_updating_a_child_touches_the_owner_even_without_changing_the_association
|
|
post = Post.create!(title: 'hello')
|
|
comment = Comment.create!(body: 'first', commentable_id: post.id, commentable_type: 'Post')
|
|
original = post.reload.updated_at
|
|
sleep 0.01
|
|
|
|
comment.update!(body: 'first, edited')
|
|
|
|
assert post.reload.updated_at > original
|
|
end
|
|
|
|
def test_destroying_a_child_touches_the_owner
|
|
post = Post.create!(title: 'hello')
|
|
comment = Comment.create!(body: 'first', commentable_id: post.id, commentable_type: 'Post')
|
|
original = post.reload.updated_at
|
|
sleep 0.01
|
|
|
|
comment.destroy
|
|
|
|
assert post.reload.updated_at > original
|
|
end
|
|
|
|
def test_reassigning_across_owner_types_touches_both_the_old_and_new_owner
|
|
post = Post.create!(title: 'hello')
|
|
photo = Photo.create!(caption: 'sunset')
|
|
comment = Comment.create!(body: 'wandering', commentable_id: post.id, commentable_type: 'Post')
|
|
original_post = post.reload.updated_at
|
|
original_photo = photo.reload.updated_at
|
|
sleep 0.01
|
|
|
|
comment.commentable = photo
|
|
comment.save!
|
|
|
|
assert post.reload.updated_at > original_post
|
|
assert photo.reload.updated_at > original_photo
|
|
end
|
|
|
|
def test_destroying_the_owner_via_dependent_destroy_cascade_does_not_re_raise
|
|
post = Post.create!(title: 'hello')
|
|
Comment.create!(body: 'first', commentable_id: post.id, commentable_type: 'Post')
|
|
Comment.create!(body: 'second', commentable_id: post.id, commentable_type: 'Post')
|
|
|
|
post.destroy
|
|
|
|
assert post.destroyed?
|
|
assert_nil Post.find_by(id: post.id)
|
|
end
|
|
|
|
def test_creating_a_child_without_an_owner_does_not_raise
|
|
Comment.create!(body: 'orphan', commentable_id: nil, commentable_type: nil)
|
|
end
|
|
end
|