diff options
Diffstat (limited to 'spec/lib/gitlab/graphql/tracing_spec.rb')
-rw-r--r-- | spec/lib/gitlab/graphql/tracing_spec.rb | 18 |
1 files changed, 8 insertions, 10 deletions
diff --git a/spec/lib/gitlab/graphql/tracing_spec.rb b/spec/lib/gitlab/graphql/tracing_spec.rb index 6bae737d0f6..7300a9a572e 100644 --- a/spec/lib/gitlab/graphql/tracing_spec.rb +++ b/spec/lib/gitlab/graphql/tracing_spec.rb @@ -3,17 +3,15 @@ require 'spec_helper' describe Gitlab::Graphql::Tracing do - let!(:graphql_duration_seconds) { double('Gitlab::Metrics::NullMetric') } - - before do - allow(Gitlab::Metrics) - .to receive(:histogram) - .with(:graphql_duration_seconds, 'GraphQL execution time') - .and_return(graphql_duration_seconds) - end + let(:graphql_duration_seconds_histogram) { double('Gitlab::Metrics::NullMetric') } it 'updates graphql histogram with expected labels' do query = 'query { users { id } }' + tracer = described_class.new + + allow(tracer) + .to receive(:graphql_duration_seconds) + .and_return(graphql_duration_seconds_histogram) expect_metric('graphql.lex', 'lex') expect_metric('graphql.parse', 'parse') @@ -22,13 +20,13 @@ describe Gitlab::Graphql::Tracing do expect_metric('graphql.execute', 'execute_query_lazy') expect_metric('graphql.execute', 'execute_multiplex') - GitlabSchema.execute(query) + GitlabSchema.execute(query, context: { tracers: [tracer] }) end private def expect_metric(platform_key, key) - expect(graphql_duration_seconds) + expect(graphql_duration_seconds_histogram) .to receive(:observe) .with({ platform_key: platform_key, key: key }, be > 0.0) end |