From bc3b6ea461ee82a4c34877168fd498b81f12763c Mon Sep 17 00:00:00 2001 From: Kevin Deisz Date: Tue, 29 May 2018 15:57:56 -0400 Subject: Reflection for attachments Add the ability to reflect on the attachments that have been defined using ActiveRecord::Reflection. --- activestorage/test/models/reflection_test.rb | 29 ++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 activestorage/test/models/reflection_test.rb (limited to 'activestorage/test') diff --git a/activestorage/test/models/reflection_test.rb b/activestorage/test/models/reflection_test.rb new file mode 100644 index 0000000000..1ddfafc0f0 --- /dev/null +++ b/activestorage/test/models/reflection_test.rb @@ -0,0 +1,29 @@ +# frozen_string_literal: true + +require "test_helper" + +class ActiveStorage::ReflectionTest < ActiveSupport::TestCase + test "allows reflecting for all attachment" do + expected_classes = + User.reflect_on_all_attachments.all? do |reflection| + reflection.is_a?(ActiveRecord::Reflection::HasOneAttachedReflection) || + reflection.is_a?(ActiveRecord::Reflection::HasManyAttachedReflection) + end + + assert expected_classes + end + + test "allows reflecting on a singular has_one_attached attachment" do + reflection = User.reflect_on_attachment(:avatar) + + assert_equal :avatar, reflection.name + assert_equal :has_one_attached, reflection.macro + end + + test "allows reflecting on a singular has_many_attached attachment" do + reflection = User.reflect_on_attachment(:highlights) + + assert_equal :highlights, reflection.name + assert_equal :has_many_attached, reflection.macro + end +end -- cgit v1.2.3 From ce337d1757fdd01e5f496f741e33275a7440b9ac Mon Sep 17 00:00:00 2001 From: Kevin Deisz Date: Thu, 31 May 2018 09:33:14 -0400 Subject: Move ActiveStorage reflection logic entirely into ActiveStorage --- activestorage/test/models/reflection_test.rb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'activestorage/test') diff --git a/activestorage/test/models/reflection_test.rb b/activestorage/test/models/reflection_test.rb index 1ddfafc0f0..da866ca996 100644 --- a/activestorage/test/models/reflection_test.rb +++ b/activestorage/test/models/reflection_test.rb @@ -6,8 +6,8 @@ class ActiveStorage::ReflectionTest < ActiveSupport::TestCase test "allows reflecting for all attachment" do expected_classes = User.reflect_on_all_attachments.all? do |reflection| - reflection.is_a?(ActiveRecord::Reflection::HasOneAttachedReflection) || - reflection.is_a?(ActiveRecord::Reflection::HasManyAttachedReflection) + reflection.is_a?(ActiveStorage::Reflection::HasOneAttachedReflection) || + reflection.is_a?(ActiveStorage::Reflection::HasManyAttachedReflection) end assert expected_classes -- cgit v1.2.3