require 'test_helper'
module ActiveModel
  class Serializer
    class Adapter
      class JsonApi
        class LinkedTest < Minitest::Test
          def setup
            ActionController::Base.cache_store.clear
            @author1 = Author.new(id: 1, name: 'Steve K.')
            @author2 = Author.new(id: 2, name: 'Tenderlove')
            @bio1 = Bio.new(id: 1, content: 'AMS Contributor')
            @bio2 = Bio.new(id: 2, content: 'Rails Contributor')
            @first_post = Post.new(id: 10, title: 'Hello!!', body: 'Hello, world!!')
            @second_post = Post.new(id: 20, title: 'New Post', body: 'Body')
            @third_post = Post.new(id: 30, title: 'Yet Another Post', body: 'Body')
            @blog = Blog.new({ name: 'AMS Blog' })
            @first_comment = Comment.new(id: 1, body: 'ZOMG A COMMENT')
            @second_comment = Comment.new(id: 2, body: 'ZOMG ANOTHER COMMENT')
            @first_post.blog = @blog
            @second_post.blog = @blog
            @third_post.blog = nil
            @first_post.comments = [@first_comment, @second_comment]
            @second_post.comments = []
            @third_post.comments = []
            @first_post.author = @author1
            @second_post.author = @author2
            @third_post.author = @author1
            @first_comment.post = @first_post
            @first_comment.author = nil
            @second_comment.post = @first_post
            @second_comment.author = nil
            @author1.posts = [@first_post, @third_post]
            @author1.bio = @bio1
            @author1.roles = []
            @author2.posts = [@second_post]
            @author2.bio = @bio2
            @author2.roles = []
            @bio1.author = @author1
            @bio2.author = @author2
          end

          def test_include_multiple_posts_and_linked_array
            serializer = ArraySerializer.new([@first_post, @second_post])
            adapter = ActiveModel::Serializer::Adapter::JsonApi.new(
              serializer,
              include: ['author', 'author.bio', 'comments']
            )
            alt_adapter = ActiveModel::Serializer::Adapter::JsonApi.new(
              serializer,
              include: 'author,author.bio,comments'
            )

            expected = {
              data: [
                {
                  id: "10",
                  title: "Hello!!",
                  body: "Hello, world!!",
                  type: "posts",
                  links: {
                    comments: { linkage: [ { type: "comments", id: '1' }, { type: "comments", id: '2' } ] },
                    blog: { linkage: { type: "blogs", id: "999" } },
                    author: { linkage: { type: "authors", id: "1" } }
                  }
                },
                {
                  id: "20",
                  title: "New Post",
                  body: "Body",
                  type: "posts",
                  links: {
                    comments: { linkage: [] },
                    blog: { linkage: { type: "blogs", id: "999" } },
                    author: { linkage: { type: "authors", id: "2" } }
                  }
                }
              ],
              included: [
                {
                  id: "1",
                  body: "ZOMG A COMMENT",
                  type: "comments",
                  links: {
                    post: { linkage: { type: "posts", id: "10" } },
                    author: { linkage: nil }
                  }
                }, {
                  id: "2",
                  body: "ZOMG ANOTHER COMMENT",
                  type: "comments",
                  links: {
                    post: { linkage: { type: "posts", id: "10" } },
                    author: { linkage: nil }
                  }
                }, {
                  id: "1",
                  name: "Steve K.",
                  type: "authors",
                  links: {
                    posts: { linkage: [ { type: "posts", id: "10" }, { type: "posts", id: "30" } ] },
                    roles: { linkage: [] },
                    bio: { linkage: { type: "bios", id: "1" } }
                  }
                }, {
                  id: "1",
                  rating: nil,
                  type: "bios",
                  content: "AMS Contributor",
                  links: {
                    author: { linkage: { type: "authors", id: "1" } }
                  }
                }, {
                  id: "2",
                  name: "Tenderlove",
                  type: "authors",
                  links: {
                    posts: { linkage: [ { type: "posts", id:"20" } ] },
                    roles: { linkage: [] },
                    bio: { linkage: { type: "bios", id: "2" } }
                  }
                }, {
                  id: "2",
                  rating: nil,
                  type: "bios",
                  content: "Rails Contributor",
                  links: {
                    author: { linkage: { type: "authors", id: "2" } }
                  }
                }
              ]
            }
            assert_equal expected, adapter.serializable_hash
            assert_equal expected, alt_adapter.serializable_hash
          end

          def test_include_multiple_posts_and_linked
            serializer = BioSerializer.new @bio1
            adapter = ActiveModel::Serializer::Adapter::JsonApi.new(
              serializer,
              include: ['author', 'author.posts']
            )
            alt_adapter = ActiveModel::Serializer::Adapter::JsonApi.new(
              serializer,
              include: 'author,author.posts'
            )

            expected = [
              {
                id: "1",
                type: "authors",
                name: "Steve K.",
                links: {
                  posts: { linkage: [ { type: "posts", id: "10"}, { type: "posts", id: "30" }] },
                  roles: { linkage: [] },
                  bio: { linkage: { type: "bios", id: "1" }}
                }
              }, {
                id: "10",
                type: "posts",
                title: "Hello!!",
                body: "Hello, world!!",
                links: {
                  comments: { linkage: [ { type: "comments", id: "1"}, { type: "comments", id: "2" }] },
                  blog: { linkage: { type: "blogs", id: "999" } },
                  author: { linkage: { type: "authors", id: "1" } }
                }
              }, {
                id: "30",
                type: "posts",
                title: "Yet Another Post",
                body: "Body",
                links: {
                  comments: { linkage: [] },
                  blog: { linkage: { type: "blogs", id: "999" } },
                  author: { linkage: { type: "authors", id: "1" } }
                }
              }
            ]

            assert_equal expected, adapter.serializable_hash[:included]
            assert_equal expected, alt_adapter.serializable_hash[:included]
          end

          def test_ignore_model_namespace_for_linked_resource_type
            spammy_post = Post.new(id: 123)
            spammy_post.related = [Spam::UnrelatedLink.new(id: 456)]
            serializer = SpammyPostSerializer.new(spammy_post)
            adapter = ActiveModel::Serializer::Adapter::JsonApi.new(serializer)
            links = adapter.serializable_hash[:data][:links]
            expected = {
              related: {
                linkage: [{
                  type: 'unrelated_links',
                  id: '456'
                }]
              }
            }
            assert_equal expected, links
          end

          def test_multiple_references_to_same_resource
            serializer = ArraySerializer.new([@first_comment, @second_comment])
            adapter = ActiveModel::Serializer::Adapter::JsonApi.new(
              serializer,
              include: ['post']
            )

            expected = [
              {
                id: "10",
                title: "Hello!!",
                body: "Hello, world!!",
                type: "posts",
                links: {
                  comments: {
                    linkage: [{type: "comments", id: "1"}, {type: "comments", id: "2"}]
                  },
                  blog: {
                    linkage: {type: "blogs", id: "999"}
                  },
                  author: {
                    linkage: {type: "authors", id: "1"}
                  }
                }
              }
            ]

            assert_equal expected, adapter.serializable_hash[:included]
          end

          def test_nil_link_with_specified_serializer
            @first_post.author = nil
            serializer = PostPreviewSerializer.new(@first_post)
            adapter = ActiveModel::Serializer::Adapter::JsonApi.new(
              serializer,
              include: ['author']
            )

            expected = {
              data: {
                id: "10",
                title: "Hello!!",
                body: "Hello, world!!",
                type: "posts",
                links: {
                  comments: { linkage: [ { type: "comments", id: '1' }, { type: "comments", id: '2' } ] },
                  author: { linkage: nil }
                }
              }
            }
            assert_equal expected, adapter.serializable_hash
          end
        end
      end
    end
  end
end