Add filter to allow users implement filter method to include/exclude attributes and relations

test/unit/active_model/serializer/filter_test.rb

@@ -0,0 +1,49 @@
+require 'test_helper'
+
+module ActiveModel
+  class Serializer
+    class FilterAttributesTest < ActiveModel::TestCase
+      def setup
+        @profile = Profile.new({ name: 'Name 1', description: 'Description 1', comments: 'Comments 1' })
+        @profile_serializer = ProfileSerializer.new(@profile)
+        @profile_serializer.instance_eval do
+          def filter(keys)
+            keys - [:description]
+          end
+        end
+      end
+
+      def test_filtered_attributes_serialization
+        assert_equal({
+          'profile' => { name: 'Name 1' }
+        }, @profile_serializer.as_json)
+      end
+    end
+
+    class FilterAssociationsTest < ActiveModel::TestCase
+      def setup
+        @association = PostSerializer._associations[:comments]
+        @old_association = @association.dup
+        @association.embed = :ids
+        @association.embed_in_root = true
+        @post = Post.new({ title: 'Title 1', body: 'Body 1', date: '1/1/2000' })
+        @post_serializer = PostSerializer.new(@post)
+        @post_serializer.instance_eval do
+          def filter(keys)
+            keys - [:body, :comments]
+          end
+        end
+      end
+
+      def teardown
+        PostSerializer._associations[:comments] = @old_association
+      end
+
+      def test_filtered_associations_serialization
+        assert_equal({
+          'post' => { title: 'Title 1' }
+        }, @post_serializer.as_json)
+      end
+    end
+  end
+end

test/unit/active_model/serializer/has_many_test.rb

@@ -4,7 +4,7 @@ module ActiveModel
   class Serializer
     class HasManyTest < ActiveModel::TestCase
       def setup
-        @association = PostSerializer._associations[0]
+        @association = PostSerializer._associations[:comments]
         @old_association = @association.dup
         @association.embed = :ids
         @post = Post.new({ title: 'Title 1', body: 'Body 1', date: '1/1/2000' })
@@ -12,7 +12,7 @@ module ActiveModel
       end
 
       def teardown
-        PostSerializer._associations[0] = @old_association
+        PostSerializer._associations[:comments] = @old_association
       end
 
       def test_associations_definition
@@ -85,7 +85,7 @@ module ActiveModel
 
       def test_associations_embedding_ids_including_objects_serialization_using_as_json
         PostSerializer.embed :ids, include: true
-        PostSerializer._associations[0].send :initialize, @association.name, @association.options
+        PostSerializer._associations[:comments].send :initialize, @association.name, @association.options
 
         @post_serializer.root = nil
         assert_equal({

test/unit/active_model/serializer/has_one_test.rb

@@ -4,7 +4,7 @@ module ActiveModel
   class Serializer
     class HasOneTest < ActiveModel::TestCase
       def setup
-        @association = UserSerializer._associations[0]
+        @association = UserSerializer._associations[:profile]
         @old_association = @association.dup
         @association.embed = :ids
         @user = User.new({ name: 'Name 1', email: 'mail@server.com', gender: 'M' })
@@ -12,7 +12,7 @@ module ActiveModel
       end
 
       def teardown
-        UserSerializer._associations[0] = @old_association
+        UserSerializer._associations[:profile] = @old_association
       end
 
       def test_associations_definition