active_model_serializers/test/unit/serializer/relationships_test.rb

# frozen_string_literal: true
require "test_helper"

module AMS
  class Serializer
    class RelationshipsTest < Test
      class ParentModelSerializer < Serializer
        relation :child_models, type: :comments, to: :many, ids: "object.child_models.map(&:id)"
        relation :child_model, type: :posts, to: :one, id: "object.child_model.id"
      end

      def setup
        super
        @object = ParentModel.new(
          child_models: [ ChildModel.new(id: 2, name: "to_many") ],
          child_model: ChildModel.new(id: 1, name: "to_one")
        )
        @serializer_class = ParentModelSerializer
        @serializer_instance = @serializer_class.new(@object)
      end

      def test_relation_macro_missing_type
        exception = assert_raises(ArgumentError) do
          ParentModelSerializer.relation :missing_type, to: :anything
        end
        assert_match(/missing keyword: type/, exception.message)
      end

      def test_relation_macro_bad_to
        exception = assert_raises(ArgumentError) do
          ParentModelSerializer.relation :unknown_relation_to, type: :anything, to: :unknown_option
        end
        assert_match(/UnknownRelationship to='unknown_option'/, exception.message)
      end

      def test_relation_macro_missing_to
        exception = assert_raises(ArgumentError) do
          ParentModelSerializer.relation :missing_relation_to, type: :anything
        end
        assert_match(/missing keyword: to/, exception.message)
      end

      def test_model_instance_relations
        expected_relations = {
          child_models: {
            data: [{ type: "comments", id: "2" }]
          },
          child_model: {
            data: { type: "posts", id: "1" }
          }
        }
        assert_equal expected_relations, @serializer_instance.relations
      end

      def test_model_instance_relationship_data
        expected = {
          type: :bananas, id: "5"
        }
        assert_equal expected, @serializer_instance.relationship_data(5, :bananas)
      end

      def test_model_instance_relationship_to_one
        expected = {
          data: { id: @object.child_model.id.to_s, type: "posts" }
        }
        assert_equal expected, @serializer_instance.child_model
      end

      def test_model_instance_relationship_to_one_id
        expected = @object.child_model.id
        assert_equal expected, @serializer_instance.related_child_model_id
      end

      def test_model_instance_relationship_to_many
        expected = {
          data: [{ id: @object.child_models.first.id.to_s, type: "comments" }]
        }
        assert_equal expected, @serializer_instance.child_models
      end

      def test_model_instance_relationship_to_many_ids
        expected = @object.child_models.map(&:id)
        assert_equal expected, @serializer_instance.related_child_models_ids
      end
    end
  end
end