diff options
Diffstat (limited to 'spec/services/dependency_proxy/download_blob_service_spec.rb')
-rw-r--r-- | spec/services/dependency_proxy/download_blob_service_spec.rb | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/spec/services/dependency_proxy/download_blob_service_spec.rb b/spec/services/dependency_proxy/download_blob_service_spec.rb index 4b5c6b5bd6a..2f293b8a46b 100644 --- a/spec/services/dependency_proxy/download_blob_service_spec.rb +++ b/spec/services/dependency_proxy/download_blob_service_spec.rb @@ -8,7 +8,7 @@ RSpec.describe DependencyProxy::DownloadBlobService do let(:token) { Digest::SHA256.hexdigest('123') } let(:blob_sha) { Digest::SHA256.hexdigest('ruby:2.7.0') } - subject { described_class.new(image, blob_sha, token).execute } + subject(:download_blob) { described_class.new(image, blob_sha, token).execute } context 'remote request is successful' do before do @@ -18,6 +18,21 @@ RSpec.describe DependencyProxy::DownloadBlobService do it { expect(subject[:status]).to eq(:success) } it { expect(subject[:file]).to be_a(Tempfile) } it { expect(subject[:file].size).to eq(6) } + + it 'streams the download' do + expected_options = { headers: anything, stream_body: true } + + expect(Gitlab::HTTP).to receive(:perform_request).with(Net::HTTP::Get, anything, expected_options) + + download_blob + end + + it 'skips read_total_timeout', :aggregate_failures do + stub_const('GitLab::HTTP::DEFAULT_READ_TOTAL_TIMEOUT', 0) + + expect(Gitlab::Metrics::System).not_to receive(:monotonic_time) + expect(download_blob).to include(status: :success) + end end context 'remote request is not found' do |