1 files changed, 68 insertions, 0 deletions
diff --git a/spec/graphql/resolvers/data_transfer/project_data_transfer_resolver_spec.rb b/spec/graphql/resolvers/data_transfer/project_data_transfer_resolver_spec.rb
new file mode 100644
index 00000000000..7307c1a54a9
--- /dev/null
+++ b/spec/graphql/resolvers/data_transfer/project_data_transfer_resolver_spec.rb
@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Resolvers::DataTransfer::ProjectDataTransferResolver, feature_category: :source_code_management do
+  include GraphqlHelpers
+
+  let_it_be(:project) { create(:project) }
+  let_it_be(:current_user) { create(:user) }
+
+  let(:from) { Date.new(2022, 1, 1) }
+  let(:to) { Date.new(2023, 1, 1) }
+  let(:finder_results) do
+    [
+      {
+        date: to,
+        repository_egress: 250000
+      }
+    ]
+  end
+
+  context 'with anonymous access' do
+    let_it_be(:current_user) { nil }
+
+    it 'does not raise an error and returns no data' do
+      expect { resolve_egress }.not_to raise_error
+      expect(resolve_egress).to be_nil
+    end
+  end
+
+  context 'with authorized user but without enough permissions' do
+    it 'does not raise an error and returns no data' do
+      project.add_developer(current_user)
+
+      expect { resolve_egress }.not_to raise_error
+      expect(resolve_egress).to be_nil
+    end
+  end
+
+  context 'when user has permissions to see data transfer' do
+    before do
+      project.add_owner(current_user)
+    end
+
+    include_examples 'Data transfer resolver'
+
+    context 'when data_transfer_monitoring_mock_data is disabled' do
+      let(:finder) { instance_double(::DataTransfer::ProjectDataTransferFinder) }
+
+      before do
+        stub_feature_flags(data_transfer_monitoring_mock_data: false)
+      end
+
+      it 'calls ProjectDataTransferFinder with expected arguments' do
+        expect(::DataTransfer::ProjectDataTransferFinder).to receive(:new).with(
+          project: project, from: from, to: to, user: current_user
+        ).once.and_return(finder)
+        allow(finder).to receive(:execute).once.and_return(finder_results)
+
+        expect(resolve_egress).to eq({ egress_nodes: finder_results })
+      end
+    end
+  end
+
+  def resolve_egress
+    resolve(described_class, obj: project, args: { from: from, to: to }, ctx: { current_user: current_user })
+  end
+end