diff --git a/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb b/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb index 725c8bc86d5..2a3ea7bdd5f 100644 --- a/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb +++ b/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb @@ -12,16 +12,16 @@ stub_const('TestSchema', Class.new(::GraphQL::Schema)) end - let(:multiplex) do - ::GraphQL::Execution::Multiplex.new( - schema: schema, - queries: queries, - context: { dataloader: GraphQL::Dataloader.new(nonblocking: nil) }, - max_complexity: nil - ) - end - describe '#arguments' do + let(:multiplex) do + ::GraphQL::Execution::Multiplex.new( + schema: schema, + queries: queries, + context: { dataloader: GraphQL::Dataloader.new(nonblocking: nil) }, + max_complexity: nil + ) + end + context 'query with argument values provided inline in the query' do let(:queries) do [ @@ -360,6 +360,15 @@ end describe '#queries' do + let(:multiplex) do + ::GraphQL::Execution::Multiplex.new( + schema: schema, + queries: [query], + context: { dataloader: GraphQL::Dataloader.new(nonblocking: nil) }, + max_complexity: nil + ) + end + let(:query) do ::GraphQL::Query.new( schema, @@ -371,10 +380,8 @@ ) end - let(:queries) { [query] } - it 'returns queries' do - expect(dd_multiplex.queries).to eq(queries) + expect(dd_multiplex.queries).to eq([query]) end end end diff --git a/spec/datadog/appsec/contrib/graphql/integration_spec.rb b/spec/datadog/appsec/contrib/graphql/integration_spec.rb index 99f1d44b568..c8093e1aea9 100644 --- a/spec/datadog/appsec/contrib/graphql/integration_spec.rb +++ b/spec/datadog/appsec/contrib/graphql/integration_spec.rb @@ -6,12 +6,12 @@ RSpec.describe Datadog::AppSec::Contrib::GraphQL::Integration do describe '.ast_node_classes_defined?' do it 'returns true when all AST node classes are defined' do - expect(described_class.ast_node_classes_defined?).to eq(true) + expect(described_class.ast_node_classes_defined?).to be(true) end it 'returns false when at least one of AST node classes is not defined' do hide_const('GraphQL::Language::Nodes::Field') - expect(described_class.ast_node_classes_defined?).to eq(false) + expect(described_class.ast_node_classes_defined?).to be(false) end end end