From 242eb7cdf7480d12dad32939b26fa1d1c24d1612 Mon Sep 17 00:00:00 2001
From: Yury Lebedev <yury.lebedev@datadoghq.com>
Date: Wed, 11 Sep 2024 12:08:02 +0200
Subject: [PATCH] Improve GraphQL specs

---
 spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb | 4 +---
 spec/datadog/appsec/contrib/graphql/integration_spec.rb       | 4 ++--
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb b/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb
index 725c8bc86d5..9fb718dbde7 100644
--- a/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb
+++ b/spec/datadog/appsec/contrib/graphql/gateway/multiplex_spec.rb
@@ -371,10 +371,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