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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
# frozen_string_literal: true
require "spec_helper"
describe GraphQL::Tracing::DataDogTrace do
module DataDogTraceTest
class Box
def initialize(value)
@value = value
end
attr_reader :value
end
class Thing < GraphQL::Schema::Object
field :str, String
def str; Box.new("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
field :str, String
def str
dataloader.with(EchoSource).load("hello")
end
end
class EchoSource < GraphQL::Dataloader::Source
def fetch(strs)
strs
end
end
class TestSchema < GraphQL::Schema
query(Query)
use GraphQL::Dataloader
trace_with(GraphQL::Tracing::DataDogTrace)
lazy_resolve(Box, :value)
end
class CustomTracerTestSchema < GraphQL::Schema
module CustomDataDogTracing
include GraphQL::Tracing::DataDogTrace
def prepare_span(trace_key, object, span)
span.set_tag("custom:#{trace_key}", object.class.name)
end
end
query(Query)
trace_with(CustomDataDogTracing)
lazy_resolve(Box, :value)
end
end
before do
Datadog.clear_all
end
it "falls back to a :tracing_fallback_transaction_name when provided" do
DataDogTraceTest::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
DataDogTraceTest::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
DataDogTraceTest::TestSchema.execute("{ int }")
assert_equal [], Datadog::SPAN_RESOURCE_NAMES
end
it "sets component and operation tags" do
DataDogTraceTest::TestSchema.execute("{ int }")
assert_includes Datadog::SPAN_TAGS, ['component', 'graphql']
assert_includes Datadog::SPAN_TAGS, ['operation', 'execute']
end
it "works with dataloader" do
DataDogTraceTest::TestSchema.execute("{ str }")
expected_keys = [
"execute.graphql",
(USING_C_PARSER ? "lex.graphql" : nil),
"parse.graphql",
"analyze.graphql",
"validate.graphql",
"Query.authorized.graphql",
"DataDogTraceTest_EchoSource.fetch.graphql"
].compact
assert_equal expected_keys, Datadog::TRACE_KEYS
end
it "sets custom tags" do
DataDogTraceTest::CustomTracerTestSchema.execute("{ thing { str } }")
expected_custom_tags = [
(USING_C_PARSER ? ["custom:lex", "String"] : nil),
["custom:parse", "String"],
["selected_operation_name", nil],
["selected_operation_type", "query"],
["query_string", "{ thing { str } }"],
["custom:execute", "GraphQL::Execution::Multiplex"],
["custom:validate", "GraphQL::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
|
