1 files changed, 54 insertions, 0 deletions
diff --git a/spec/support/shared_examples/graphql/connection_redaction_shared_examples.rb b/spec/support/shared_examples/graphql/connection_redaction_shared_examples.rb
new file mode 100644
index 00000000000..12a7b3fe414
--- /dev/null
+++ b/spec/support/shared_examples/graphql/connection_redaction_shared_examples.rb
@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+# requires:
+#  - `connection` (no-empty, containing `unwanted` and at least one more item)
+#  - `unwanted` (single item in collection)
+RSpec.shared_examples 'a redactable connection' do
+  context 'no redactor set' do
+    it 'contains the unwanted item' do
+      expect(connection.nodes).to include(unwanted)
+    end
+
+    it 'does not redact more than once' do
+      connection.nodes
+      r_state = connection.send(:redaction_state)
+
+      expect(r_state.redacted { raise 'Should not be called!' }).to be_present
+    end
+  end
+
+  let_it_be(:constant_redactor) do
+    Class.new do
+      def initialize(remove)
+        @remove = remove
+      end
+
+      def redact(items)
+        items - @remove
+      end
+    end
+  end
+
+  context 'redactor is set' do
+    let(:redactor) do
+      constant_redactor.new([unwanted])
+    end
+
+    before do
+      connection.redactor = redactor
+    end
+
+    it 'does not contain the unwanted item' do
+      expect(connection.nodes).not_to include(unwanted)
+      expect(connection.nodes).not_to be_empty
+    end
+
+    it 'does not redact more than once' do
+      expect(redactor).to receive(:redact).once.and_call_original
+
+      connection.nodes
+      connection.nodes
+      connection.nodes
+    end
+  end
+end