extraction_test.exs

defmodule OpentelemetryAbsintheTest.Extraction do
  use OpentelemetryAbsintheTest.Case

  alias OpentelemetryAbsintheTest.Support.GraphQL.Queries
  alias OpentelemetryAbsintheTest.Support.Query

  @graphql_request_selections :"graphql.request.selections"

  describe "extracts" do
    test "request query" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_request_query: true)
      query = Queries.query()

      assert ^query = query |> Query.query_for_attrs() |> Map.fetch!(:"graphql.document")
    end

    test "request variables" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_request_variables: true)
      variables = %{"isbn" => "A1", "author" => "Mara Bos"}

      assert ^variables =
               Queries.query()
               |> Query.query_for_attrs(variables: variables)
               |> Map.fetch!(:"graphql.request.variables")
               |> Jason.decode!()
    end

    test "request selections" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_request_selections: true)

      assert ["book"] =
               Queries.query()
               |> Query.query_for_attrs(variables: %{"isbn" => "A1"})
               |> Map.fetch!(@graphql_request_selections)
               |> Jason.decode!()
    end

    test "request selections on a mutation" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_request_selections: true)

      assert ["create_book"] =
               Queries.mutation()
               |> Query.query_for_attrs(variables: %{"isbn" => "A1"})
               |> Map.fetch!(@graphql_request_selections)
               |> Jason.decode!()
    end

    test "aliased request selections as their un-aliased name" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_request_selections: true)

      assert ["book"] =
               Queries.aliased_query()
               |> Query.query_for_attrs(variables: %{"isbn" => "A1"})
               |> Map.fetch!(@graphql_request_selections)
               |> Jason.decode!()
    end

    test "request selections on multiple queries" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_request_selections: true)

      assert ["book"] =
               Queries.batch_queries()
               |> Query.query_for_attrs(variables: %{"isbn" => "A1"}, operation_name: "OperationOne")
               |> Map.fetch!(@graphql_request_selections)
               |> Jason.decode!()

      assert ["books"] =
               Queries.batch_queries()
               |> Query.query_for_attrs(variables: %{"isbn" => "A1"}, operation_name: "OperationTwo")
               |> Map.fetch!(@graphql_request_selections)
               |> Jason.decode!()
    end

    test "response result" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_response_result: true)

      result =
        Queries.query()
        |> Query.query_for_attrs(variables: %{"isbn" => "A1"})
        |> Map.fetch!(:"graphql.response.result")
        |> Jason.decode!()

      assert %{"data" => %{"book" => %{"author" => %{"age" => 18, "name" => "Ale Ali"}, "title" => "Fire"}}} = result
    end

    test "response errors" do
      OpentelemetryAbsinthe.Instrumentation.setup(trace_response_errors: true)

      errors =
        Queries.query()
        |> Query.query_for_attrs()
        |> Map.fetch!(:"graphql.response.errors")
        |> Jason.decode!()

      assert [
               %{
                 "locations" => [%{"column" => 8, "line" => 2}],
                 "message" => "In argument \"isbn\": Expected type \"String!\", found null."
               },
               %{
                 "locations" => [%{"column" => 7, "line" => 1}],
                 "message" => "Variable \"isbn\": Expected non-null, found null."
               }
             ] = errors
    end
  end
end