311 lines
9.3 KiB
Ruby
311 lines
9.3 KiB
Ruby
require 'spec_helper'
|
|
|
|
describe Gitlab::Git::GitlabProjects do
|
|
after do
|
|
TestEnv.clean_test_path
|
|
end
|
|
|
|
let(:project) { create(:project, :repository) }
|
|
|
|
if $VERBOSE
|
|
let(:logger) { Logger.new(STDOUT) }
|
|
else
|
|
let(:logger) { double('logger').as_null_object }
|
|
end
|
|
|
|
let(:tmp_repos_path) { TestEnv.repos_path }
|
|
let(:repo_name) { project.disk_path + '.git' }
|
|
let(:tmp_repo_path) { File.join(tmp_repos_path, repo_name) }
|
|
let(:gl_projects) { build_gitlab_projects(tmp_repos_path, repo_name) }
|
|
|
|
describe '#initialize' do
|
|
it { expect(gl_projects.shard_path).to eq(tmp_repos_path) }
|
|
it { expect(gl_projects.repository_relative_path).to eq(repo_name) }
|
|
it { expect(gl_projects.repository_absolute_path).to eq(File.join(tmp_repos_path, repo_name)) }
|
|
it { expect(gl_projects.logger).to eq(logger) }
|
|
end
|
|
|
|
describe '#push_branches' do
|
|
let(:remote_name) { 'remote-name' }
|
|
let(:branch_name) { 'master' }
|
|
let(:cmd) { %W(git push -- #{remote_name} #{branch_name}) }
|
|
let(:force) { false }
|
|
|
|
subject { gl_projects.push_branches(remote_name, 600, force, [branch_name]) }
|
|
|
|
it 'executes the command' do
|
|
stub_spawn(cmd, 600, tmp_repo_path, success: true)
|
|
|
|
is_expected.to be_truthy
|
|
end
|
|
|
|
it 'fails' do
|
|
stub_spawn(cmd, 600, tmp_repo_path, success: false)
|
|
|
|
is_expected.to be_falsy
|
|
end
|
|
|
|
context 'with --force' do
|
|
let(:cmd) { %W(git push --force -- #{remote_name} #{branch_name}) }
|
|
let(:force) { true }
|
|
|
|
it 'executes the command' do
|
|
stub_spawn(cmd, 600, tmp_repo_path, success: true)
|
|
|
|
is_expected.to be_truthy
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#fetch_remote' do
|
|
let(:remote_name) { 'remote-name' }
|
|
let(:branch_name) { 'master' }
|
|
let(:force) { false }
|
|
let(:tags) { true }
|
|
let(:args) { { force: force, tags: tags }.merge(extra_args) }
|
|
let(:extra_args) { {} }
|
|
let(:cmd) { %W(git fetch #{remote_name} --prune --quiet --tags) }
|
|
|
|
subject { gl_projects.fetch_remote(remote_name, 600, args) }
|
|
|
|
def stub_tempfile(name, filename, opts = {})
|
|
chmod = opts.delete(:chmod)
|
|
file = StringIO.new
|
|
|
|
allow(file).to receive(:close!)
|
|
allow(file).to receive(:path).and_return(name)
|
|
|
|
expect(Tempfile).to receive(:new).with(filename).and_return(file)
|
|
expect(file).to receive(:chmod).with(chmod) if chmod
|
|
|
|
file
|
|
end
|
|
|
|
context 'with default args' do
|
|
it 'executes the command' do
|
|
stub_spawn(cmd, 600, tmp_repo_path, {}, success: true)
|
|
|
|
is_expected.to be_truthy
|
|
end
|
|
|
|
it 'fails' do
|
|
stub_spawn(cmd, 600, tmp_repo_path, {}, success: false)
|
|
|
|
is_expected.to be_falsy
|
|
end
|
|
end
|
|
|
|
context 'with --force' do
|
|
let(:force) { true }
|
|
let(:cmd) { %W(git fetch #{remote_name} --prune --quiet --force --tags) }
|
|
|
|
it 'executes the command with forced option' do
|
|
stub_spawn(cmd, 600, tmp_repo_path, {}, success: true)
|
|
|
|
is_expected.to be_truthy
|
|
end
|
|
end
|
|
|
|
context 'with --no-tags' do
|
|
let(:tags) { false }
|
|
let(:cmd) { %W(git fetch #{remote_name} --prune --quiet --no-tags) }
|
|
|
|
it 'executes the command' do
|
|
stub_spawn(cmd, 600, tmp_repo_path, {}, success: true)
|
|
|
|
is_expected.to be_truthy
|
|
end
|
|
end
|
|
|
|
describe 'with an SSH key' do
|
|
let(:extra_args) { { ssh_key: 'SSH KEY' } }
|
|
|
|
it 'sets GIT_SSH to a custom script' do
|
|
script = stub_tempfile('scriptFile', 'gitlab-shell-ssh-wrapper', chmod: 0o755)
|
|
key = stub_tempfile('/tmp files/keyFile', 'gitlab-shell-key-file', chmod: 0o400)
|
|
|
|
stub_spawn(cmd, 600, tmp_repo_path, { 'GIT_SSH' => 'scriptFile' }, success: true)
|
|
|
|
is_expected.to be_truthy
|
|
|
|
expect(script.string).to eq("#!/bin/sh\nexec ssh '-oIdentityFile=\"/tmp files/keyFile\"' '-oIdentitiesOnly=\"yes\"' \"$@\"")
|
|
expect(key.string).to eq('SSH KEY')
|
|
end
|
|
end
|
|
|
|
describe 'with known_hosts data' do
|
|
let(:extra_args) { { known_hosts: 'KNOWN HOSTS' } }
|
|
|
|
it 'sets GIT_SSH to a custom script' do
|
|
script = stub_tempfile('scriptFile', 'gitlab-shell-ssh-wrapper', chmod: 0o755)
|
|
key = stub_tempfile('/tmp files/knownHosts', 'gitlab-shell-known-hosts', chmod: 0o400)
|
|
|
|
stub_spawn(cmd, 600, tmp_repo_path, { 'GIT_SSH' => 'scriptFile' }, success: true)
|
|
|
|
is_expected.to be_truthy
|
|
|
|
expect(script.string).to eq("#!/bin/sh\nexec ssh '-oStrictHostKeyChecking=\"yes\"' '-oUserKnownHostsFile=\"/tmp files/knownHosts\"' \"$@\"")
|
|
expect(key.string).to eq('KNOWN HOSTS')
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#import_project' do
|
|
let(:project) { create(:project) }
|
|
let(:import_url) { TestEnv.factory_repo_path_bare }
|
|
let(:cmd) { %W(git clone --bare -- #{import_url} #{tmp_repo_path}) }
|
|
let(:timeout) { 600 }
|
|
|
|
subject { gl_projects.import_project(import_url, timeout) }
|
|
|
|
shared_examples 'importing repository' do
|
|
context 'success import' do
|
|
it 'imports a repo' do
|
|
expect(File.exist?(File.join(tmp_repo_path, 'HEAD'))).to be_falsy
|
|
|
|
is_expected.to be_truthy
|
|
|
|
expect(File.exist?(File.join(tmp_repo_path, 'HEAD'))).to be_truthy
|
|
end
|
|
end
|
|
|
|
context 'already exists' do
|
|
it "doesn't import" do
|
|
FileUtils.mkdir_p(tmp_repo_path)
|
|
|
|
is_expected.to be_falsy
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when Gitaly import_repository feature is enabled' do
|
|
it_behaves_like 'importing repository'
|
|
end
|
|
|
|
context 'when Gitaly import_repository feature is disabled', :disable_gitaly do
|
|
describe 'logging' do
|
|
it 'imports a repo' do
|
|
message = "Importing project from <#{import_url}> to <#{tmp_repo_path}>."
|
|
expect(logger).to receive(:info).with(message)
|
|
|
|
subject
|
|
end
|
|
end
|
|
|
|
context 'timeout' do
|
|
it 'does not import a repo' do
|
|
stub_spawn_timeout(cmd, timeout, nil)
|
|
|
|
message = "Importing project from <#{import_url}> to <#{tmp_repo_path}> failed."
|
|
expect(logger).to receive(:error).with(message)
|
|
|
|
is_expected.to be_falsy
|
|
|
|
expect(gl_projects.output).to eq("Timed out\n")
|
|
expect(File.exist?(File.join(tmp_repo_path, 'HEAD'))).to be_falsy
|
|
end
|
|
end
|
|
|
|
it_behaves_like 'importing repository'
|
|
end
|
|
end
|
|
|
|
describe '#fork_repository' do
|
|
let(:dest_repos_path) { tmp_repos_path }
|
|
let(:dest_repo_name) { File.join('@hashed', 'aa', 'bb', 'xyz.git') }
|
|
let(:dest_repo) { File.join(dest_repos_path, dest_repo_name) }
|
|
|
|
subject { gl_projects.fork_repository(dest_repos_path, dest_repo_name) }
|
|
|
|
before do
|
|
FileUtils.mkdir_p(dest_repos_path)
|
|
|
|
# Undo spec_helper stub that deletes hooks
|
|
allow_any_instance_of(described_class).to receive(:fork_repository).and_call_original
|
|
end
|
|
|
|
after do
|
|
FileUtils.rm_rf(dest_repos_path)
|
|
end
|
|
|
|
shared_examples 'forking a repository' do
|
|
it 'forks the repository' do
|
|
is_expected.to be_truthy
|
|
|
|
expect(File.exist?(dest_repo)).to be_truthy
|
|
expect(File.exist?(File.join(dest_repo, 'hooks', 'pre-receive'))).to be_truthy
|
|
expect(File.exist?(File.join(dest_repo, 'hooks', 'post-receive'))).to be_truthy
|
|
end
|
|
|
|
it 'does not fork if a project of the same name already exists' do
|
|
# create a fake project at the intended destination
|
|
FileUtils.mkdir_p(dest_repo)
|
|
|
|
is_expected.to be_falsy
|
|
end
|
|
end
|
|
|
|
context 'when Gitaly fork_repository feature is enabled' do
|
|
it_behaves_like 'forking a repository'
|
|
end
|
|
|
|
context 'when Gitaly fork_repository feature is disabled', :disable_gitaly do
|
|
it_behaves_like 'forking a repository'
|
|
|
|
# We seem to be stuck to having only one working Gitaly storage in tests, changing
|
|
# that is not very straight-forward so I'm leaving this test here for now till
|
|
# https://gitlab.com/gitlab-org/gitlab-ce/issues/41393 is fixed.
|
|
context 'different storages' do
|
|
let(:dest_repos_path) { File.join(File.dirname(tmp_repos_path), 'alternative') }
|
|
|
|
it 'forks the repo' do
|
|
is_expected.to be_truthy
|
|
|
|
expect(File.exist?(dest_repo)).to be_truthy
|
|
expect(File.exist?(File.join(dest_repo, 'hooks', 'pre-receive'))).to be_truthy
|
|
expect(File.exist?(File.join(dest_repo, 'hooks', 'post-receive'))).to be_truthy
|
|
end
|
|
end
|
|
|
|
describe 'log messages' do
|
|
describe 'successful fork' do
|
|
it do
|
|
message = "Forking repository from <#{tmp_repo_path}> to <#{dest_repo}>."
|
|
expect(logger).to receive(:info).with(message)
|
|
|
|
subject
|
|
end
|
|
end
|
|
|
|
describe 'failed fork due existing destination' do
|
|
it do
|
|
FileUtils.mkdir_p(dest_repo)
|
|
message = "fork-repository failed: destination repository <#{dest_repo}> already exists."
|
|
expect(logger).to receive(:error).with(message)
|
|
|
|
subject
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def build_gitlab_projects(*args)
|
|
described_class.new(
|
|
*args,
|
|
global_hooks_path: Gitlab.config.gitlab_shell.hooks_path,
|
|
logger: logger
|
|
)
|
|
end
|
|
|
|
def stub_spawn(*args, success: true)
|
|
exitstatus = success ? 0 : nil
|
|
expect(gl_projects).to receive(:popen_with_timeout).with(*args)
|
|
.and_return(["output", exitstatus])
|
|
end
|
|
|
|
def stub_spawn_timeout(*args)
|
|
expect(gl_projects).to receive(:popen_with_timeout).with(*args)
|
|
.and_raise(Timeout::Error)
|
|
end
|
|
end
|