141 lines
3.9 KiB
Ruby
141 lines
3.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe ContainerRegistry::Blob do
|
|
let(:group) { create(:group, name: 'group') }
|
|
let(:project) { create(:project, path: 'test', group: group) }
|
|
|
|
let(:repository) do
|
|
create(:container_repository, name: 'image',
|
|
tags: %w[latest rc1],
|
|
project: project)
|
|
end
|
|
|
|
let(:config) do
|
|
{ 'digest' => 'sha256:0123456789012345',
|
|
'mediaType' => 'binary',
|
|
'size' => 1000 }
|
|
end
|
|
|
|
let(:blob) { described_class.new(repository, config) }
|
|
|
|
before do
|
|
stub_container_registry_config(enabled: true,
|
|
api_url: 'http://registry.gitlab',
|
|
host_port: 'registry.gitlab')
|
|
end
|
|
|
|
it { expect(blob).to respond_to(:repository) }
|
|
it { expect(blob).to delegate_method(:registry).to(:repository) }
|
|
it { expect(blob).to delegate_method(:client).to(:repository) }
|
|
|
|
describe '#path' do
|
|
it 'returns a valid path to the blob' do
|
|
expect(blob.path).to eq('group/test/image@sha256:0123456789012345')
|
|
end
|
|
end
|
|
|
|
describe '#digest' do
|
|
it 'return correct digest value' do
|
|
expect(blob.digest).to eq 'sha256:0123456789012345'
|
|
end
|
|
end
|
|
|
|
describe '#type' do
|
|
it 'returns a correct type' do
|
|
expect(blob.type).to eq 'binary'
|
|
end
|
|
end
|
|
|
|
describe '#revision' do
|
|
it 'returns a correct blob SHA' do
|
|
expect(blob.revision).to eq '0123456789012345'
|
|
end
|
|
end
|
|
|
|
describe '#short_revision' do
|
|
it 'return a short SHA' do
|
|
expect(blob.short_revision).to eq '012345678'
|
|
end
|
|
end
|
|
|
|
describe '#delete' do
|
|
before do
|
|
stub_request(:delete, 'http://registry.gitlab/v2/group/test/image/blobs/sha256:0123456789012345')
|
|
.to_return(status: 200)
|
|
end
|
|
|
|
it 'returns true when blob has been successfully deleted' do
|
|
expect(blob.delete).to be_truthy
|
|
end
|
|
end
|
|
|
|
describe '#data' do
|
|
context 'when locally stored' do
|
|
before do
|
|
stub_request(:get, 'http://registry.gitlab/v2/group/test/image/blobs/sha256:0123456789012345')
|
|
.to_return(
|
|
status: 200,
|
|
headers: { 'Content-Type' => 'application/json' },
|
|
body: '{"key":"value"}')
|
|
end
|
|
|
|
it 'returns a correct blob data' do
|
|
expect(blob.data).to eq '{"key":"value"}'
|
|
end
|
|
end
|
|
|
|
context 'when externally stored' do
|
|
let(:location) { 'http://external.com/blob/file' }
|
|
|
|
before do
|
|
stub_request(:get, 'http://registry.gitlab/v2/group/test/image/blobs/sha256:0123456789012345')
|
|
.with(headers: { 'Authorization' => 'bearer token' })
|
|
.to_return(
|
|
status: 307,
|
|
headers: { 'Location' => location })
|
|
end
|
|
|
|
context 'for a valid address' do
|
|
before do
|
|
stub_request(:get, location)
|
|
.with { |request| !request.headers.include?('Authorization') }
|
|
.to_return(
|
|
status: 200,
|
|
headers: { 'Content-Type' => 'application/json' },
|
|
body: '{"key":"value"}')
|
|
end
|
|
|
|
it 'returns correct data' do
|
|
expect(blob.data).to eq '{"key":"value"}'
|
|
end
|
|
end
|
|
|
|
context 'for a relative address' do
|
|
before do
|
|
stub_request(:get, 'http://registry.gitlab/relative')
|
|
.with { |request| !request.headers.include?('Authorization') }
|
|
.to_return(
|
|
status: 200,
|
|
headers: { 'Content-Type' => 'application/json' },
|
|
body: '{"key":"value"}')
|
|
end
|
|
|
|
let(:location) { '/relative' }
|
|
|
|
it 'returns correct data' do
|
|
expect(blob.data).to eq '{"key":"value"}'
|
|
end
|
|
end
|
|
|
|
context 'for invalid file' do
|
|
let(:location) { 'file:///etc/passwd' }
|
|
|
|
it 'raises an error' do
|
|
expect { blob.data }.to raise_error(ArgumentError, 'Invalid scheme for file:///etc/passwd')
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|