Unsuffixed association keys tests

Author: gdlx

test/unit/active_model/serializer/has_many_test.rb

@@ -242,6 +242,32 @@ def test_associations_embedding_objects_using_embed_namespace
           }
         }, @post_serializer.as_json)
       end
+
+      CONFIG.default_key_type = :name
+      class NameKeyPostSerializer < ActiveModel::Serializer
+        attributes :title, :body
+        
+        has_many :comments
+      end
+      CONFIG.default_key_type = nil
+
+      def test_associations_name_key_embedding_ids_serialization_using_serializable_hash
+        @association = NameKeyPostSerializer._associations[:comments]
+        @association.embed = :ids
+
+        assert_equal({
+          title: 'Title 1', body: 'Body 1', 'comments' => @post.comments.map { |c| c.object_id }
+        }, NameKeyPostSerializer.new(@post).serializable_hash)
+      end
+
+      def test_associations_name_key_embedding_ids_serialization_using_as_json
+        @association = NameKeyPostSerializer._associations[:comments]
+        @association.embed = :ids
+
+        assert_equal({
+          'name_key_post' => { title: 'Title 1', body: 'Body 1', 'comments' => @post.comments.map { |c| c.object_id } }
+        }, NameKeyPostSerializer.new(@post).as_json)
+      end
     end
   end
 end

test/unit/active_model/serializer/has_one_test.rb

@@ -216,6 +216,32 @@ def test_associations_embedding_ids_using_embed_namespace_and_embed_in_root_key
           }
         }, @user_serializer.as_json)
       end
+
+      CONFIG.default_key_type = :name
+      class NameKeyUserSerializer < ActiveModel::Serializer
+        attributes :name, :email
+
+        has_one :profile
+      end
+      CONFIG.default_key_type = nil
+
+      def test_associations_name_key_embedding_ids_serialization_using_serializable_hash
+        @association = NameKeyUserSerializer._associations[:profile]
+        @association.embed = :ids
+
+        assert_equal({
+          name: 'Name 1', email: '[email protected]', 'profile' => @user.profile.object_id
+        }, NameKeyUserSerializer.new(@user).serializable_hash)
+      end
+
+      def test_associations_name_key_embedding_ids_serialization_using_as_json
+        @association = NameKeyUserSerializer._associations[:profile]
+        @association.embed = :ids
+
+        assert_equal({
+          'name_key_user' => { name: 'Name 1', email: '[email protected]', 'profile' => @user.profile.object_id }
+        }, NameKeyUserSerializer.new(@user).as_json)
+      end
     end
   end
 end