require 'test_helper'

module GraphQL
  module Client
    module Query
      class DocumentTest < Minitest::Test
        def setup
          @schema = GraphQLSchema.new(schema_fixture('schema.json'))
          @graphql_schema = GraphQL::Schema::Loader.load(schema_fixture('schema.json'))
        end

        def test_initialize_yields_self
          document_object = nil

          document = Document.new(@schema) do |d|
            document_object = d
          end

          assert_equal document_object, document
        end

        def test_add_mutation_creates_a_mutation_operation
          document = Document.new(@schema)
          mutation = document.add_mutation('createUser')

          assert_instance_of MutationOperation, mutation
          assert_equal({ 'createUser' => mutation }, document.operations)
        end

        def test_add_query_creates_a_query_operation
          document = Document.new(@schema)
          query = document.add_query('getUser')

          assert_instance_of QueryOperation, query
          assert_equal({ 'getUser' => query }, document.operations)
          assert_equal 1, document.operations.size
        end

        def test_add_operation_sets_default_name
          document = Document.new(@schema)
          query = document.add_query

          assert_instance_of QueryOperation, query
          assert_equal({ 'default' => query }, document.operations)
          assert_equal 1, document.operations.size
        end

        def test_add_operation_yields_block
          document = Document.new(@schema)
          query_object = nil

          query = document.add_query do |q|
            query_object = q
          end

          assert_equal query_object, query
        end

        def test_add_operation_supports_multiple_unique_operations
          document = Document.new(@schema)
          document.add_query('getUser')
          document.add_query('getPosts')

          assert_equal 2, document.operations.size
          assert_equal %w(getUser getPosts), document.operations.keys
        end

        def test_add_operation_enforces_unique_names
          document = Document.new(@schema)
          document.add_query('getUser')

          assert_raises Document::DUPLICATE_OPERATION_NAME do
            document.add_query('getUser')
          end
        end

        def test_add_operation_requires_a_document_with_multiple_operations_to_all_be_named
          document = Document.new(@schema)
          document.add_query

          assert_raises Document::INVALID_DOCUMENT do
            document.add_query('getUser')
          end
        end

        def test_define_fragment_creates_a_fragment
          document = Document.new(@schema)

          fragment = document.define_fragment('imageFields', on: 'Image')

          assert_equal 'imageFields', fragment.name
          assert_equal @schema.type('Image'), fragment.type
          assert_equal document, fragment.document
          assert_equal({ 'imageFields' => fragment }, document.fragments)
        end

        def test_define_fragment_yields_fragment
          document = Document.new(@schema)
          fragment_object = nil

          fragment = document.define_fragment('imageFields', on: 'Image') do |f|
            fragment_object = f
          end

          assert_equal fragment_object, fragment
        end

        def test_define_fragment_raises_exception_for_invalid_targets
          document = Document.new(@schema)

          assert_raises Document::INVALID_FRAGMENT_TARGET do
            document.define_fragment('imageFields', on: 'String')
          end
        end

        def test_fragment_definitions_is_the_fragments_definition_string
          document = Document.new(@schema)

          document.define_fragment('imageFields', on: 'Image') do |f|
            f.add_field('src')
          end

          document.define_fragment('shopName', on: 'Shop') do |f|
            f.add_field('name')
          end

          fragment_string = <<~QUERY
            fragment imageFields on Image {
              src
            }

            fragment shopName on Shop {
              name
            }
          QUERY

          assert_equal fragment_string, document.fragment_definitions
        end

        def test_to_query_joins_all_operations
          document = Document.new(@schema) do |d|
            d.add_query('shopQuery') do |q|
              q.add_field('shop') do |shop|
                shop.add_field('name')
              end
            end

            d.add_mutation('customers') do |c|
              c.add_field('customerCreate', input: { email: 'email', password: 'password' }) do |create|
                create.add_field('customer') do |customer|
                  customer.add_field('email')
                end
              end
            end
          end

          query_string = <<~QUERY
            query shopQuery {
              shop {
                name
              }
            }

            mutation customers {
              customerCreate(input: { email: \"email\", password: \"password\" }) {
                customer {
                  email
                }
              }
            }
          QUERY

          assert_equal query_string, document.to_query
          assert_valid_query query_string, @graphql_schema, operation_name: 'shopQuery'
          assert_valid_query query_string, @graphql_schema, operation_name: 'customers'
        end

        def test_to_query_includes_fragment_definitions
          document = Document.new(@schema) do |d|
            d.define_fragment('imageFields', on: 'Image') do |f|
              f.add_field('src')
            end

            d.add_query('getShop') do |q|
              q.add_field('shop') do |shop|
                shop.add_field('name')
              end
            end

            d.add_query('getProductImages') do |q|
              q.add_field('shop') do |shop|
                shop.add_field('productByHandle', handle: 'test') do |product|
                  product.add_connection('images', first: 10) do |connection|
                    connection.add_fragment('imageFields')

                    connection.add_inline_fragment do |f|
                      f.add_field('altText')
                    end
                  end
                end
              end
            end
          end

          query_string = <<~QUERY
            fragment imageFields on Image {
              src
            }

            query getShop {
              shop {
                name
              }
            }

            query getProductImages {
              shop {
                productByHandle(handle: \"test\") {
                  id
                  images(first: 10) {
                    edges {
                      cursor
                      node {
                        ...imageFields
                        ... on Image {
                          altText
                        }
                      }
                    }
                    pageInfo {
                      hasPreviousPage
                      hasNextPage
                    }
                  }
                }
              }
            }
          QUERY

          assert_equal query_string, document.to_query
          assert_valid_query query_string, @graphql_schema, operation_name: 'getShop'
        end
      end
    end
  end
end