1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
|
# frozen_string_literal: true
require "spec_helper"
describe GraphQL::Tracing::DataDogTracing do
module DataDogTest
class Thing < GraphQL::Schema::Object
field :str, String
def str
"blah"
end
end
class Query < GraphQL::Schema::Object
include GraphQL::Types::Relay::HasNodeField
field :int, Integer, null: false
def int
1
end
field :thing, Thing
def thing
:thing
end
end
class TestSchema < GraphQL::Schema
query(Query)
use(GraphQL::Tracing::DataDogTracing)
end
class CustomTracerTestSchema < GraphQL::Schema
class CustomDataDogTracing < GraphQL::Tracing::DataDogTracing
def prepare_span(trace_key, data, span)
span.set_tag("custom:#{trace_key}", data.keys.join(","))
end
end
query(Query)
use(CustomDataDogTracing)
end
end
before do
Datadog.clear_all
end
it "falls back to a :tracing_fallback_transaction_name when provided" do
DataDogTest::TestSchema.execute("{ int }", context: { tracing_fallback_transaction_name: "Abcd" })
assert_equal ["Abcd"], Datadog::SPAN_RESOURCE_NAMES
end
it "does not use the :tracing_fallback_transaction_name if an operation name is present" do
DataDogTest::TestSchema.execute(
"query Ab { int }",
context: { tracing_fallback_transaction_name: "Cd" }
)
assert_equal ["Ab"], Datadog::SPAN_RESOURCE_NAMES
end
it "does not set resource if no value can be derived" do
DataDogTest::TestSchema.execute("{ int }")
assert_equal [], Datadog::SPAN_RESOURCE_NAMES
end
it "sets component and operation tags" do
DataDogTest::TestSchema.execute("{ int }")
assert_includes Datadog::SPAN_TAGS, ['component', 'graphql']
assert_includes Datadog::SPAN_TAGS, ['operation', 'execute_multiplex']
end
it "sets custom tags tags" do
DataDogTest::CustomTracerTestSchema.execute("{ thing { str } }")
expected_custom_tags = [
(USING_C_PARSER ? ["custom:lex", "query_string"] : nil),
["custom:parse", "query_string"],
["custom:execute_multiplex", "multiplex"],
["custom:analyze_multiplex", "multiplex"],
["custom:validate", "validate,query"],
["custom:analyze_query", "query"],
["custom:execute_query", "query"],
["custom:authorized", "context,type,object,path"],
["custom:execute_field", "field,query,ast_node,arguments,object,owner,path"],
["custom:authorized", "context,type,object,path"],
["custom:execute_query_lazy", "multiplex,query"],
].compact
actual_custom_tags = Datadog::SPAN_TAGS.reject { |t| t[0] == "operation" || t[0] == "component" || t[0].is_a?(Symbol) }
assert_equal expected_custom_tags, actual_custom_tags
end
end
|