Commit 8b0d2283 authored by Douwe Maan's avatar Douwe Maan

Merge branch 'gitaly-disk-access-3' into 'master'

Find and mark more Git disk access locations, part 2

See merge request gitlab-org/gitlab-ce!19437
parents a2785eb8 f376347f
...@@ -23,10 +23,11 @@ class RepositoryForkWorker ...@@ -23,10 +23,11 @@ class RepositoryForkWorker
source_repository_storage_path, source_disk_path = *args source_repository_storage_path, source_disk_path = *args
source_repository_storage_name = Gitlab.config.repositories.storages.find do |_, info| source_repository_storage_name = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Gitlab.config.repositories.storages.find do |_, info|
info.legacy_disk_path == source_repository_storage_path info.legacy_disk_path == source_repository_storage_path
end&.first || raise("no shard found for path '#{source_repository_storage_path}'") end&.first || raise("no shard found for path '#{source_repository_storage_path}'")
end
fork_repository(target_project, source_repository_storage_name, source_disk_path) fork_repository(target_project, source_repository_storage_name, source_disk_path)
end end
end end
......
...@@ -53,25 +53,12 @@ module Gitlab ...@@ -53,25 +53,12 @@ module Gitlab
# Import project via git clone --bare # Import project via git clone --bare
# URL must be publicly cloneable # URL must be publicly cloneable
def import_project(source, timeout) def import_project(source, timeout)
Gitlab::GitalyClient.migrate(:import_repository, status: Gitlab::GitalyClient::MigrationStatus::OPT_OUT) do |is_enabled|
if is_enabled
gitaly_import_repository(source)
else
git_import_repository(source, timeout) git_import_repository(source, timeout)
end end
end
end
def fork_repository(new_shard_name, new_repository_relative_path) def fork_repository(new_shard_name, new_repository_relative_path)
Gitlab::GitalyClient.migrate(:fork_repository,
status: Gitlab::GitalyClient::MigrationStatus::OPT_OUT) do |is_enabled|
if is_enabled
gitaly_fork_repository(new_shard_name, new_repository_relative_path)
else
git_fork_repository(new_shard_name, new_repository_relative_path) git_fork_repository(new_shard_name, new_repository_relative_path)
end end
end
end
def fetch_remote(name, timeout, force:, tags:, ssh_key: nil, known_hosts: nil, prune: true) def fetch_remote(name, timeout, force:, tags:, ssh_key: nil, known_hosts: nil, prune: true)
tags_option = tags ? '--tags' : '--no-tags' tags_option = tags ? '--tags' : '--no-tags'
...@@ -241,16 +228,6 @@ module Gitlab ...@@ -241,16 +228,6 @@ module Gitlab
true true
end end
def gitaly_import_repository(source)
raw_repository = Gitlab::Git::Repository.new(shard_name, repository_relative_path, nil)
Gitlab::GitalyClient::RepositoryService.new(raw_repository).import_repository(source)
true
rescue GRPC::BadStatus => e
@output << e.message
false
end
def git_fork_repository(new_shard_name, new_repository_relative_path) def git_fork_repository(new_shard_name, new_repository_relative_path)
from_path = repository_absolute_path from_path = repository_absolute_path
new_shard_path = Gitlab.config.repositories.storages.fetch(new_shard_name).legacy_disk_path new_shard_path = Gitlab.config.repositories.storages.fetch(new_shard_name).legacy_disk_path
...@@ -270,16 +247,6 @@ module Gitlab ...@@ -270,16 +247,6 @@ module Gitlab
run(cmd, nil) && Gitlab::Git::Repository.create_hooks(to_path, global_hooks_path) run(cmd, nil) && Gitlab::Git::Repository.create_hooks(to_path, global_hooks_path)
end end
def gitaly_fork_repository(new_shard_name, new_repository_relative_path)
target_repository = Gitlab::Git::Repository.new(new_shard_name, new_repository_relative_path, nil)
raw_repository = Gitlab::Git::Repository.new(shard_name, repository_relative_path, nil)
Gitlab::GitalyClient::RepositoryService.new(target_repository).fork_repository(raw_repository)
rescue GRPC::BadStatus => e
logger.error "fork-repository failed: #{e.message}"
false
end
end end
end end
end end
...@@ -24,8 +24,10 @@ module Gitlab ...@@ -24,8 +24,10 @@ module Gitlab
address = val['gitaly_address'] address = val['gitaly_address']
end end
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
storages << { name: key, path: val.legacy_disk_path } storages << { name: key, path: val.legacy_disk_path }
end end
end
if Rails.env.test? if Rails.env.test?
storages << { name: 'test_second_storage', path: Rails.root.join('tmp', 'tests', 'second_storage').to_s } storages << { name: 'test_second_storage', path: Rails.root.join('tmp', 'tests', 'second_storage').to_s }
......
...@@ -106,10 +106,17 @@ module Gitlab ...@@ -106,10 +106,17 @@ module Gitlab
raise Error.new("don't use disk paths with import_repository: #{url.inspect}") raise Error.new("don't use disk paths with import_repository: #{url.inspect}")
end end
relative_path = "#{name}.git"
cmd = gitaly_migrate(:import_repository, status: Gitlab::GitalyClient::MigrationStatus::OPT_OUT) do |is_enabled|
if is_enabled
GitalyGitlabProjects.new(storage, relative_path)
else
# The timeout ensures the subprocess won't hang forever # The timeout ensures the subprocess won't hang forever
cmd = gitlab_projects(storage, "#{name}.git") gitlab_projects(storage, relative_path)
success = cmd.import_project(url, git_timeout) end
end
success = cmd.import_project(url, git_timeout)
raise Error, cmd.output unless success raise Error, cmd.output unless success
success success
...@@ -165,8 +172,16 @@ module Gitlab ...@@ -165,8 +172,16 @@ module Gitlab
# #
# Gitaly migration: https://gitlab.com/gitlab-org/gitaly/issues/817 # Gitaly migration: https://gitlab.com/gitlab-org/gitaly/issues/817
def fork_repository(forked_from_storage, forked_from_disk_path, forked_to_storage, forked_to_disk_path) def fork_repository(forked_from_storage, forked_from_disk_path, forked_to_storage, forked_to_disk_path)
gitlab_projects(forked_from_storage, "#{forked_from_disk_path}.git") forked_from_relative_path = "#{forked_from_disk_path}.git"
.fork_repository(forked_to_storage, "#{forked_to_disk_path}.git") fork_args = [forked_to_storage, "#{forked_to_disk_path}.git"]
gitaly_migrate(:fork_repository, status: Gitlab::GitalyClient::MigrationStatus::OPT_OUT) do |is_enabled|
if is_enabled
GitalyGitlabProjects.new(forked_from_storage, forked_from_relative_path).fork_repository(*fork_args)
else
gitlab_projects(forked_from_storage, forked_from_relative_path).fork_repository(*fork_args)
end
end
end end
# Removes a repository from file system, using rm_diretory which is an alias # Removes a repository from file system, using rm_diretory which is an alias
...@@ -452,5 +467,39 @@ module Gitlab ...@@ -452,5 +467,39 @@ module Gitlab
# need to do the same here... # need to do the same here...
raise Error, e raise Error, e
end end
class GitalyGitlabProjects
attr_reader :shard_name, :repository_relative_path, :output
def initialize(shard_name, repository_relative_path)
@shard_name = shard_name
@repository_relative_path = repository_relative_path
@output = ''
end
def import_project(source, _timeout)
raw_repository = Gitlab::Git::Repository.new(shard_name, repository_relative_path, nil)
Gitlab::GitalyClient::RepositoryService.new(raw_repository).import_repository(source)
true
rescue GRPC::BadStatus => e
@output = e.message
false
end
def fork_repository(new_shard_name, new_repository_relative_path)
target_repository = Gitlab::Git::Repository.new(new_shard_name, new_repository_relative_path, nil)
raw_repository = Gitlab::Git::Repository.new(shard_name, repository_relative_path, nil)
Gitlab::GitalyClient::RepositoryService.new(target_repository).fork_repository(raw_repository)
rescue GRPC::BadStatus => e
logger.error "fork-repository failed: #{e.message}"
false
end
def logger
Rails.logger
end
end
end end
end end
...@@ -140,8 +140,10 @@ module Gitlab ...@@ -140,8 +140,10 @@ module Gitlab
end end
def repository_storage_paths_args def repository_storage_paths_args
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Gitlab.config.repositories.storages.values.map { |rs| rs.legacy_disk_path } Gitlab.config.repositories.storages.values.map { |rs| rs.legacy_disk_path }
end end
end
def user_home def user_home
Rails.env.test? ? Rails.root.join('tmp/tests') : Gitlab.config.gitlab.user_home Rails.env.test? ? Rails.root.join('tmp/tests') : Gitlab.config.gitlab.user_home
......
...@@ -4,6 +4,7 @@ module SystemCheck ...@@ -4,6 +4,7 @@ module SystemCheck
set_name 'Orphaned namespaces:' set_name 'Orphaned namespaces:'
def multi_check def multi_check
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Gitlab.config.repositories.storages.each do |storage_name, repository_storage| Gitlab.config.repositories.storages.each do |storage_name, repository_storage|
$stdout.puts $stdout.puts
$stdout.puts "* Storage: #{storage_name} (#{repository_storage.legacy_disk_path})".color(:yellow) $stdout.puts "* Storage: #{storage_name} (#{repository_storage.legacy_disk_path})".color(:yellow)
...@@ -12,6 +13,7 @@ module SystemCheck ...@@ -12,6 +13,7 @@ module SystemCheck
orphans = (toplevel_namespace_dirs - existing_namespaces) orphans = (toplevel_namespace_dirs - existing_namespaces)
print_orphans(orphans, storage_name) print_orphans(orphans, storage_name)
end end
end
clear_namespaces! # releases memory when check finishes clear_namespaces! # releases memory when check finishes
end end
......
...@@ -44,11 +44,13 @@ namespace :gitlab do ...@@ -44,11 +44,13 @@ namespace :gitlab do
start_checking "GitLab Shell" start_checking "GitLab Shell"
check_gitlab_shell check_gitlab_shell
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
check_repo_base_exists check_repo_base_exists
check_repo_base_is_not_symlink check_repo_base_is_not_symlink
check_repo_base_user_and_group check_repo_base_user_and_group
check_repo_base_permissions check_repo_base_permissions
check_repos_hooks_directory_is_link check_repos_hooks_directory_is_link
end
check_gitlab_shell_self_test check_gitlab_shell_self_test
finished_checking "GitLab Shell" finished_checking "GitLab Shell"
......
...@@ -67,9 +67,11 @@ namespace :gitlab do ...@@ -67,9 +67,11 @@ namespace :gitlab do
puts "GitLab Shell".color(:yellow) puts "GitLab Shell".color(:yellow)
puts "Version:\t#{gitlab_shell_version || "unknown".color(:red)}" puts "Version:\t#{gitlab_shell_version || "unknown".color(:red)}"
puts "Repository storage paths:" puts "Repository storage paths:"
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Gitlab.config.repositories.storages.each do |name, repository_storage| Gitlab.config.repositories.storages.each do |name, repository_storage|
puts "- #{name}: \t#{repository_storage.legacy_disk_path}" puts "- #{name}: \t#{repository_storage.legacy_disk_path}"
end end
end
puts "Hooks:\t\t#{Gitlab.config.gitlab_shell.hooks_path}" puts "Hooks:\t\t#{Gitlab.config.gitlab_shell.hooks_path}"
puts "Git:\t\t#{Gitlab.config.git.bin_path}" puts "Git:\t\t#{Gitlab.config.git.bin_path}"
end end
......
...@@ -151,11 +151,6 @@ FactoryBot.define do ...@@ -151,11 +151,6 @@ FactoryBot.define do
trait :empty_repo do trait :empty_repo do
after(:create) do |project| after(:create) do |project|
raise "Failed to create repository!" unless project.create_repository raise "Failed to create repository!" unless project.create_repository
# We delete hooks so that gitlab-shell will not try to authenticate with
# an API that isn't running
project.gitlab_shell.rm_directory(project.repository_storage,
File.join("#{project.disk_path}.git", 'hooks'))
end end
end end
...@@ -180,13 +175,6 @@ FactoryBot.define do ...@@ -180,13 +175,6 @@ FactoryBot.define do
trait :wiki_repo do trait :wiki_repo do
after(:create) do |project| after(:create) do |project|
raise 'Failed to create wiki repository!' unless project.create_wiki raise 'Failed to create wiki repository!' unless project.create_wiki
# We delete hooks so that gitlab-shell will not try to authenticate with
# an API that isn't running
project.gitlab_shell.rm_directory(
project.repository_storage,
File.join("#{project.wiki.repository.disk_path}.git", "hooks")
)
end end
end end
......
...@@ -79,7 +79,9 @@ describe Gitlab::Diff::File do ...@@ -79,7 +79,9 @@ describe Gitlab::Diff::File do
let(:diffs) { commit.diffs } let(:diffs) { commit.diffs }
before do before do
info_dir_path = File.join(project.repository.path_to_repo, 'info') info_dir_path = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
File.join(project.repository.path_to_repo, 'info')
end
FileUtils.mkdir(info_dir_path) unless File.exist?(info_dir_path) FileUtils.mkdir(info_dir_path) unless File.exist?(info_dir_path)
File.write(File.join(info_dir_path, 'attributes'), "*.md -diff\n") File.write(File.join(info_dir_path, 'attributes'), "*.md -diff\n")
......
...@@ -149,7 +149,9 @@ describe Gitlab::Git::Blob, seed_helper: true do ...@@ -149,7 +149,9 @@ describe Gitlab::Git::Blob, seed_helper: true do
it 'limits the size of a large file' do it 'limits the size of a large file' do
blob_size = Gitlab::Git::Blob::MAX_DATA_DISPLAY_SIZE + 1 blob_size = Gitlab::Git::Blob::MAX_DATA_DISPLAY_SIZE + 1
buffer = Array.new(blob_size, 0) buffer = Array.new(blob_size, 0)
rugged_blob = Rugged::Blob.from_buffer(repository.rugged, buffer.join('')) rugged_blob = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Rugged::Blob.from_buffer(repository.rugged, buffer.join(''))
end
blob = Gitlab::Git::Blob.raw(repository, rugged_blob) blob = Gitlab::Git::Blob.raw(repository, rugged_blob)
expect(blob.size).to eq(blob_size) expect(blob.size).to eq(blob_size)
...@@ -164,7 +166,9 @@ describe Gitlab::Git::Blob, seed_helper: true do ...@@ -164,7 +166,9 @@ describe Gitlab::Git::Blob, seed_helper: true do
context 'when sha references a tree' do context 'when sha references a tree' do
it 'returns nil' do it 'returns nil' do
tree = repository.rugged.rev_parse('master^{tree}') tree = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
repository.rugged.rev_parse('master^{tree}')
end
blob = Gitlab::Git::Blob.raw(repository, tree.oid) blob = Gitlab::Git::Blob.raw(repository, tree.oid)
...@@ -278,7 +282,11 @@ describe Gitlab::Git::Blob, seed_helper: true do ...@@ -278,7 +282,11 @@ describe Gitlab::Git::Blob, seed_helper: true do
end end
describe '.batch_lfs_pointers' do describe '.batch_lfs_pointers' do
let(:tree_object) { repository.rugged.rev_parse('master^{tree}') } let(:tree_object) do
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
repository.rugged.rev_parse('master^{tree}')
end
end
let(:non_lfs_blob) do let(:non_lfs_blob) do
Gitlab::Git::Blob.find( Gitlab::Git::Blob.find(
......
...@@ -69,7 +69,9 @@ describe Gitlab::Git::Branch, seed_helper: true do ...@@ -69,7 +69,9 @@ describe Gitlab::Git::Branch, seed_helper: true do
Gitlab::Git.committer_hash(email: user.email, name: user.name) Gitlab::Git.committer_hash(email: user.email, name: user.name)
end end
let(:params) do let(:params) do
parents = [repository.rugged.head.target] parents = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
[repository.rugged.head.target]
end
tree = parents.first.tree tree = parents.first.tree
{ {
......
...@@ -4,12 +4,15 @@ describe Gitlab::Git::Commit, seed_helper: true do ...@@ -4,12 +4,15 @@ describe Gitlab::Git::Commit, seed_helper: true do
let(:repository) { Gitlab::Git::Repository.new('default', TEST_REPO_PATH, '') } let(:repository) { Gitlab::Git::Repository.new('default', TEST_REPO_PATH, '') }
let(:commit) { described_class.find(repository, SeedRepo::Commit::ID) } let(:commit) { described_class.find(repository, SeedRepo::Commit::ID) }
let(:rugged_commit) do let(:rugged_commit) do
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
repository.rugged.lookup(SeedRepo::Commit::ID) repository.rugged.lookup(SeedRepo::Commit::ID)
end end
end
describe "Commit info" do describe "Commit info" do
before do before do
repo = Gitlab::Git::Repository.new('default', TEST_REPO_PATH, '').rugged repo = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Gitlab::Git::Repository.new('default', TEST_REPO_PATH, '').rugged
end
@committer = { @committer = {
email: 'mike@smith.com', email: 'mike@smith.com',
...@@ -58,7 +61,9 @@ describe Gitlab::Git::Commit, seed_helper: true do ...@@ -58,7 +61,9 @@ describe Gitlab::Git::Commit, seed_helper: true do
after do after do
# Erase the new commit so other tests get the original repo # Erase the new commit so other tests get the original repo
repo = Gitlab::Git::Repository.new('default', TEST_REPO_PATH, '').rugged repo = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Gitlab::Git::Repository.new('default', TEST_REPO_PATH, '').rugged
end
repo.references.update("refs/heads/master", SeedRepo::LastCommit::ID) repo.references.update("refs/heads/master", SeedRepo::LastCommit::ID)
end end
end end
...@@ -115,7 +120,9 @@ describe Gitlab::Git::Commit, seed_helper: true do ...@@ -115,7 +120,9 @@ describe Gitlab::Git::Commit, seed_helper: true do
describe '.find' do describe '.find' do
it "should return first head commit if without params" do it "should return first head commit if without params" do
expect(described_class.last(repository).id).to eq( expect(described_class.last(repository).id).to eq(
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
repository.rugged.head.target.oid repository.rugged.head.target.oid
end
) )
end end
......
...@@ -27,9 +27,11 @@ EOT ...@@ -27,9 +27,11 @@ EOT
too_large: false too_large: false
} }
@rugged_diff = repository.rugged.diff("5937ac0a7beb003549fc5fd26fc247adbce4a52e^", "5937ac0a7beb003549fc5fd26fc247adbce4a52e", paths: @rugged_diff = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
repository.rugged.diff("5937ac0a7beb003549fc5fd26fc247adbce4a52e^", "5937ac0a7beb003549fc5fd26fc247adbce4a52e", paths:
[".gitmodules"]).patches.first [".gitmodules"]).patches.first
end end
end
describe '.new' do describe '.new' do
context 'using a Hash' do context 'using a Hash' do
......
...@@ -5,6 +5,13 @@ describe Gitlab::Git::GitlabProjects do ...@@ -5,6 +5,13 @@ describe Gitlab::Git::GitlabProjects do
TestEnv.clean_test_path TestEnv.clean_test_path
end end
around do |example|
# TODO move this spec to gitaly-ruby. GitlabProjects is not used in gitlab-ce
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
example.run
end
end
let(:project) { create(:project, :repository) } let(:project) { create(:project, :repository) }
if $VERBOSE if $VERBOSE
...@@ -190,11 +197,6 @@ describe Gitlab::Git::GitlabProjects do ...@@ -190,11 +197,6 @@ describe Gitlab::Git::GitlabProjects do
end 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 describe 'logging' do
it 'imports a repo' do it 'imports a repo' do
message = "Importing project from <#{import_url}> to <#{tmp_repo_path}>." message = "Importing project from <#{import_url}> to <#{tmp_repo_path}>."
...@@ -220,7 +222,6 @@ describe Gitlab::Git::GitlabProjects do ...@@ -220,7 +222,6 @@ describe Gitlab::Git::GitlabProjects do
it_behaves_like 'importing repository' it_behaves_like 'importing repository'
end end
end
describe '#fork_repository' do describe '#fork_repository' do
let(:dest_repos) { TestEnv::REPOS_STORAGE } let(:dest_repos) { TestEnv::REPOS_STORAGE }
...@@ -232,9 +233,6 @@ describe Gitlab::Git::GitlabProjects do ...@@ -232,9 +233,6 @@ describe Gitlab::Git::GitlabProjects do
before do before do
FileUtils.mkdir_p(dest_repos_path) 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 end
after do after do
...@@ -258,11 +256,6 @@ describe Gitlab::Git::GitlabProjects do ...@@ -258,11 +256,6 @@ describe Gitlab::Git::GitlabProjects do
end end
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' it_behaves_like 'forking a repository'
# We seem to be stuck to having only one working Gitaly storage in tests, changing # We seem to be stuck to having only one working Gitaly storage in tests, changing
...@@ -306,7 +299,6 @@ describe Gitlab::Git::GitlabProjects do ...@@ -306,7 +299,6 @@ describe Gitlab::Git::GitlabProjects do
end end
end end
end end
end
def build_gitlab_projects(*args) def build_gitlab_projects(*args)
described_class.new( described_class.new(
......
...@@ -8,6 +8,13 @@ describe Gitlab::Git::Hook do ...@@ -8,6 +8,13 @@ describe Gitlab::Git::Hook do
allow_any_instance_of(described_class).to receive(:trigger).and_call_original allow_any_instance_of(described_class).to receive(:trigger).and_call_original
end end
around do |example|
# TODO move hook tests to gitaly-ruby. Hook will disappear from gitlab-ce
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
example.run
end
end
describe "#trigger" do describe "#trigger" do
set(:project) { create(:project, :repository) } set(:project) { create(:project, :repository) }
let(:repository) { project.repository.raw_repository } let(:repository) { project.repository.raw_repository }
......
...@@ -8,6 +8,13 @@ describe Gitlab::Git::Index, seed_helper: true do ...@@ -8,6 +8,13 @@ describe Gitlab::Git::Index, seed_helper: true do
index.read_tree(repository.lookup('master').tree) index.read_tree(repository.lookup('master').tree)
end end
around do |example|
# TODO move these specs to gitaly-ruby. The Index class will disappear from gitlab-ce
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
example.run
end
end
describe '#create' do describe '#create' do
let(:options) do let(:options) do
{ {
......
...@@ -19,8 +19,10 @@ describe Gitlab::ImportExport::MergeRequestParser do ...@@ -19,8 +19,10 @@ describe Gitlab::ImportExport::MergeRequestParser do
end end
after do after do
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
FileUtils.rm_rf(project.repository.path_to_repo) FileUtils.rm_rf(project.repository.path_to_repo)
end end
end
it 'has a source branch' do it 'has a source branch' do
expect(project.repository.branch_exists?(parsed_merge_request.source_branch)).to be true expect(project.repository.branch_exists?(parsed_merge_request.source_branch)).to be true
......
...@@ -498,6 +498,23 @@ describe Gitlab::Shell do ...@@ -498,6 +498,23 @@ describe Gitlab::Shell do
) )
end end
context 'with gitaly' do
it 'returns true when the command succeeds' do
expect_any_instance_of(Gitlab::GitalyClient::RepositoryService).to receive(:fork_repository)
.with(repository.raw_repository) { :gitaly_response_object }
is_expected.to be_truthy
end
it 'return false when the command fails' do
expect_any_instance_of(Gitlab::GitalyClient::RepositoryService).to receive(:fork_repository)
.with(repository.raw_repository) { raise GRPC::BadStatus, 'bla' }
is_expected.to be_falsy
end
end
context 'without gitaly', :disable_gitaly do
it 'returns true when the command succeeds' do it 'returns true when the command succeeds' do
expect(gitlab_projects).to receive(:fork_repository).with('nfs-file05', 'fork/path.git') { true } expect(gitlab_projects).to receive(:fork_repository).with('nfs-file05', 'fork/path.git') { true }
...@@ -510,6 +527,7 @@ describe Gitlab::Shell do ...@@ -510,6 +527,7 @@ describe Gitlab::Shell do
is_expected.to be_falsy is_expected.to be_falsy
end end
end end
end
shared_examples 'fetch_remote' do |gitaly_on| shared_examples 'fetch_remote' do |gitaly_on|
def fetch_remote(ssh_auth = nil, prune = true) def fetch_remote(ssh_auth = nil, prune = true)
...@@ -662,6 +680,27 @@ describe Gitlab::Shell do ...@@ -662,6 +680,27 @@ describe Gitlab::Shell do
describe '#import_repository' do describe '#import_repository' do
let(:import_url) { 'https://gitlab.com/gitlab-org/gitlab-ce.git' } let(:import_url) { 'https://gitlab.com/gitlab-org/gitlab-ce.git' }
context 'with gitaly' do
it 'returns true when the command succeeds' do
expect_any_instance_of(Gitlab::GitalyClient::RepositoryService).to receive(:import_repository).with(import_url)
result = gitlab_shell.import_repository(project.repository_storage, project.disk_path, import_url)
expect(result).to be_truthy
end
it 'raises an exception when the command fails' do
expect_any_instance_of(Gitlab::GitalyClient::RepositoryService).to receive(:import_repository)
.with(import_url) { raise GRPC::BadStatus, 'bla' }
expect_any_instance_of(Gitlab::Shell::GitalyGitlabProjects).to receive(:output) { 'error'}
expect do
gitlab_shell.import_repository(project.repository_storage, project.disk_path, import_url)
end.to raise_error(Gitlab::Shell::Error, "error")
end
end
context 'without gitaly', :disable_gitaly do
it 'returns true when the command succeeds' do it 'returns true when the command succeeds' do
expect(gitlab_projects).to receive(:import_project).with(import_url, timeout) { true } expect(gitlab_projects).to receive(:import_project).with(import_url, timeout) { true }
...@@ -680,6 +719,7 @@ describe Gitlab::Shell do ...@@ -680,6 +719,7 @@ describe Gitlab::Shell do
end end
end end
end end
end
describe 'namespace actions' do describe 'namespace actions' do
subject { described_class.new } subject { described_class.new }
......
...@@ -118,7 +118,9 @@ describe GitTagPushService do ...@@ -118,7 +118,9 @@ describe GitTagPushService do
before do before do
# Create the lightweight tag # Create the lightweight tag
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
project.repository.raw_repository.rugged.tags.create(tag_name, newrev) project.repository.raw_repository.rugged.tags.create(tag_name, newrev)
end
# Clear tag list cache # Clear tag list cache
project.repository.expire_tags_cache project.repository.expire_tags_cache
......
...@@ -63,7 +63,9 @@ describe MergeRequests::SquashService do ...@@ -63,7 +63,9 @@ describe MergeRequests::SquashService do
end end
it 'has the same diff as the merge request, but a different SHA' do it 'has the same diff as the merge request, but a different SHA' do
rugged = project.repository.rugged rugged = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
project.repository.rugged
end
mr_diff = rugged.diff(merge_request.diff_base_sha, merge_request.diff_head_sha) mr_diff = rugged.diff(merge_request.diff_base_sha, merge_request.diff_head_sha)
squash_diff = rugged.diff(merge_request.diff_start_sha, squash_sha) squash_diff = rugged.diff(merge_request.diff_start_sha, squash_sha)
......
...@@ -32,7 +32,7 @@ describe Projects::AfterImportService do ...@@ -32,7 +32,7 @@ describe Projects::AfterImportService do
end end
it 'removes refs/pull/**/*' do it 'removes refs/pull/**/*' do
expect(repository.rugged.references.map(&:name)) expect(rugged.references.map(&:name))
.not_to include(%r{\Arefs/pull/}) .not_to include(%r{\Arefs/pull/})
end end
end end
...@@ -46,10 +46,14 @@ describe Projects::AfterImportService do ...@@ -46,10 +46,14 @@ describe Projects::AfterImportService do
end end
it "does not remove refs/#{name}/tmp" do it "does not remove refs/#{name}/tmp" do
expect(repository.rugged.references.map(&:name)) expect(rugged.references.map(&:name))
.to include("refs/#{name}/tmp") .to include("refs/#{name}/tmp")
end end
end end
end end
def rugged
Gitlab::GitalyClient::StorageSettings.allow_disk_access { repository.rugged }
end
end end
end end
...@@ -5,7 +5,11 @@ describe Projects::DestroyService do ...@@ -5,7 +5,11 @@ describe Projects::DestroyService do
let!(:user) { create(:user) } let!(:user) { create(:user) }
let!(:project) { create(:project, :repository, namespace: user.namespace) } let!(:project) { create(:project, :repository, namespace: user.namespace) }
let!(:path) { project.repository.path_to_repo } let!(:path) do
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
project.repository.path_to_repo
end
end
let!(:remove_path) { path.sub(/\.git\Z/, "+#{project.id}+deleted.git") } let!(:remove_path) { path.sub(/\.git\Z/, "+#{project.id}+deleted.git") }
let!(:async) { false } # execute or async_execute let!(:async) { false } # execute or async_execute
......
...@@ -37,7 +37,11 @@ describe Projects::HashedStorage::MigrateRepositoryService do ...@@ -37,7 +37,11 @@ describe Projects::HashedStorage::MigrateRepositoryService do
it 'writes project full path to .git/config' do it 'writes project full path to .git/config' do
service.execute service.execute
expect(project.repository.rugged.config['gitlab.fullpath']).to eq project.full_path rugged_config = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
project.repository.rugged.config['gitlab.fullpath']
end
expect(rugged_config).to eq project.full_path
end end
end end
......
...@@ -107,19 +107,6 @@ RSpec.configure do |config| ...@@ -107,19 +107,6 @@ RSpec.configure do |config|
end end
config.before(:example) do config.before(:example) do
# Skip pre-receive hook check so we can use the web editor and merge.
allow_any_instance_of(Gitlab::Git::Hook).to receive(:trigger).and_return([true, nil])
allow_any_instance_of(Gitlab::Git::GitlabProjects).to receive(:fork_repository).and_wrap_original do |m, *args|
m.call(*args)
shard_name, repository_relative_path = args
# We can't leave the hooks in place after a fork, as those would fail in tests
# The "internal" API is not available
Gitlab::Shell.new.rm_directory(shard_name,
File.join(repository_relative_path, 'hooks'))
end
# Enable all features by default for testing # Enable all features by default for testing
allow(Feature).to receive(:enabled?) { true } allow(Feature).to receive(:enabled?) { true }
end end
......
...@@ -135,6 +135,16 @@ module TestEnv ...@@ -135,6 +135,16 @@ module TestEnv
install_dir: Gitlab.config.gitlab_shell.path, install_dir: Gitlab.config.gitlab_shell.path,
version: Gitlab::Shell.version_required, version: Gitlab::Shell.version_required,
task: 'gitlab:shell:install') task: 'gitlab:shell:install')
create_fake_git_hooks
end
def create_fake_git_hooks
# gitlab-shell hooks don't work in our test environment because they try to make internal API calls
hooks_dir = File.join(Gitlab.config.gitlab_shell.path, 'hooks')
%w[pre-receive post-receive update].each do |hook|
File.open(File.join(hooks_dir, hook), 'w', 0755) { |f| f.puts '#!/bin/sh' }
end
end end
def setup_gitaly def setup_gitaly
......
...@@ -134,8 +134,10 @@ describe 'gitlab:gitaly namespace rake task' do ...@@ -134,8 +134,10 @@ describe 'gitlab:gitaly namespace rake task' do
parsed_output = TomlRB.parse(expected_output) parsed_output = TomlRB.parse(expected_output)
config.each do |name, params| config.each do |name, params|
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
expect(parsed_output['storage']).to include({ 'name' => name, 'path' => params.legacy_disk_path }) expect(parsed_output['storage']).to include({ 'name' => name, 'path' => params.legacy_disk_path })
end end
end end
end end
end
end end
...@@ -7,11 +7,17 @@ describe 'gitlab:shell rake tasks' do ...@@ -7,11 +7,17 @@ describe 'gitlab:shell rake tasks' do
stub_warn_user_is_not_gitlab stub_warn_user_is_not_gitlab
end end
after do
TestEnv.create_fake_git_hooks
end
describe 'install task' do describe 'install task' do
it 'invokes create_hooks task' do it 'invokes create_hooks task' do
expect(Rake::Task['gitlab:shell:create_hooks']).to receive(:invoke) expect(Rake::Task['gitlab:shell:create_hooks']).to receive(:invoke)
storages = Gitlab.config.repositories.storages.values.map(&:legacy_disk_path) storages = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Gitlab.config.repositories.storages.values.map(&:legacy_disk_path)
end
expect(Kernel).to receive(:system).with('bin/install', *storages).and_call_original expect(Kernel).to receive(:system).with('bin/install', *storages).and_call_original
expect(Kernel).to receive(:system).with('bin/compile').and_call_original expect(Kernel).to receive(:system).with('bin/compile').and_call_original
......
...@@ -218,7 +218,9 @@ describe GitGarbageCollectWorker do ...@@ -218,7 +218,9 @@ describe GitGarbageCollectWorker do
# Create a new commit on a random new branch # Create a new commit on a random new branch
def create_objects(project) def create_objects(project)
rugged = project.repository.rugged rugged = Gitlab::GitalyClient::StorageSettings.allow_disk_access do
project.repository.rugged
end
old_commit = rugged.branches.first.target old_commit = rugged.branches.first.target
new_commit_sha = Rugged::Commit.create( new_commit_sha = Rugged::Commit.create(
rugged, rugged,
...@@ -237,8 +239,10 @@ describe GitGarbageCollectWorker do ...@@ -237,8 +239,10 @@ describe GitGarbageCollectWorker do
end end
def packs(project) def packs(project)
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
Dir["#{project.repository.path_to_repo}/objects/pack/*.pack"] Dir["#{project.repository.path_to_repo}/objects/pack/*.pack"]
end end
end
def packed_refs(project) def packed_refs(project)
path = "#{project.repository.path_to_repo}/packed-refs" path = "#{project.repository.path_to_repo}/packed-refs"
......
...@@ -2,7 +2,11 @@ require 'spec_helper' ...@@ -2,7 +2,11 @@ require 'spec_helper'
describe ProjectDestroyWorker do describe ProjectDestroyWorker do
let(:project) { create(:project, :repository, pending_delete: true) } let(:project) { create(:project, :repository, pending_delete: true) }
let(:path) { project.repository.path_to_repo } let(:path) do
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
project.repository.path_to_repo
end
end
subject { described_class.new } subject { described_class.new }
......
...@@ -94,6 +94,9 @@ describe RepositoryForkWorker do ...@@ -94,6 +94,9 @@ describe RepositoryForkWorker do
it_behaves_like 'RepositoryForkWorker performing' it_behaves_like 'RepositoryForkWorker performing'
it 'logs a message about forking with old-style arguments' do it 'logs a message about forking with old-style arguments' do
allow(subject).to receive(:gitlab_shell).and_return(shell)
expect(shell).to receive(:fork_repository) { true }
allow(Rails.logger).to receive(:info).with(anything) # To compensate for other logs allow(Rails.logger).to receive(:info).with(anything) # To compensate for other logs
expect(Rails.logger).to receive(:info).with("Project #{fork_project.id} is being forked using old-style arguments.") expect(Rails.logger).to receive(:info).with("Project #{fork_project.id} is being forked using old-style arguments.")
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment