BigW Consortium Gitlab

test_env.rb 4.74 KB
Newer Older
1 2
require 'rspec/mocks'

3 4 5
module TestEnv
  extend self

6 7
  # When developing the seed repository, comment out the branch you will modify.
  BRANCH_SHA = {
8
    'empty-branch'     => '7efb185',
9
    'flatten-dir'      => 'e56497b',
10 11 12 13 14
    'feature'          => '0b4bc9a',
    'feature_conflict' => 'bb5206f',
    'fix'              => '12d65c8',
    'improve/awesome'  => '5937ac0',
    'markdown'         => '0ed8c6c',
15 16
    'master'           => '5937ac0',
    "'test'"           => 'e56497b',
17 18
  }

19 20 21 22 23 24 25
  # gitlab-test-fork is a fork of gitlab-fork, but we don't necessarily
  # need to keep all the branches in sync.
  # We currently only need a subset of the branches
  FORKED_BRANCH_SHA = {
    'add-submodule-version-bump' => '3f547c08',
    'master' => '5937ac0'
  }
26

27 28
  # Test environment
  #
29
  # See gitlab.yml.example test section for paths
30
  #
31
  def init(opts = {})
32 33 34
    # Disable mailer for spinach tests
    disable_mailer if opts[:mailer] == false

Robert Speicher committed
35
    clean_test_path
36

37
    FileUtils.mkdir_p(repos_path)
38
    FileUtils.mkdir_p(backup_path)
39

40 41 42
    # Setup GitLab shell for test instance
    setup_gitlab_shell

43 44
    # Create repository for FactoryGirl.create(:project)
    setup_factory_repo
45

46 47
    # Create repository for FactoryGirl.create(:forked_project_with_submodules)
    setup_forked_repo
48 49
  end

50
  def disable_mailer
51 52
    allow_any_instance_of(NotificationService).to receive(:mailer).
      and_return(double.as_null_object)
53
  end
54

55
  def enable_mailer
56 57
    allow_any_instance_of(NotificationService).to receive(:mailer).
      and_call_original
58 59
  end

60
  def disable_pre_receive
61
    allow_any_instance_of(Gitlab::Git::Hook).to receive(:trigger).and_return(true)
62 63
  end

Robert Speicher committed
64 65 66 67 68 69 70
  # Clean /tmp/tests
  #
  # Keeps gitlab-shell and gitlab-test
  def clean_test_path
    tmp_test_path = Rails.root.join('tmp', 'tests', '**')

    Dir[tmp_test_path].each do |entry|
71
      unless File.basename(entry) =~ /\Agitlab-(shell|test|test-fork)\z/
Robert Speicher committed
72 73 74 75 76
        FileUtils.rm_rf(entry)
      end
    end
  end

77
  def setup_gitlab_shell
Robert Speicher committed
78 79 80
    unless File.directory?(Rails.root.join(*%w(tmp tests gitlab-shell)))
      `rake gitlab:shell:install`
    end
81
  end
82

83
  def setup_factory_repo
84 85 86 87 88 89 90 91 92 93 94 95 96
    setup_repo(factory_repo_path, factory_repo_path_bare, factory_repo_name,
               BRANCH_SHA)
  end

  # This repo has a submodule commit that is not present in the main test
  # repository.
  def setup_forked_repo
    setup_repo(forked_repo_path, forked_repo_path_bare, forked_repo_name,
               FORKED_BRANCH_SHA)
  end

  def setup_repo(repo_path, repo_path_bare, repo_name, branch_sha)
    clone_url = "https://gitlab.com/gitlab-org/#{repo_name}.git"
97

98 99
    unless File.directory?(repo_path)
      system(*%W(git clone -q #{clone_url} #{repo_path}))
100 101
    end

102 103
    Dir.chdir(repo_path) do
      branch_sha.each do |branch, sha|
104 105 106 107 108 109 110 111 112 113 114 115 116
        # Try to reset without fetching to avoid using the network.
        reset = %W(git update-ref refs/heads/#{branch} #{sha})
        unless system(*reset)
          if system(*%w(git fetch origin))
            unless system(*reset)
              raise 'The fetched test seed '\
              'does not contain the required revision.'
            end
          else
            raise 'Could not fetch test seed repository.'
          end
        end
      end
117
    end
118 119

    # We must copy bare repositories because we will push to them.
120
    system(git_env, *%W(git clone -q --bare #{repo_path} #{repo_path_bare}))
121 122
  end

123
  def copy_repo(project)
124
    base_repo_path = File.expand_path(factory_repo_path_bare)
125 126 127 128
    target_repo_path = File.expand_path(repos_path + "/#{project.namespace.path}/#{project.path}.git")
    FileUtils.mkdir_p(target_repo_path)
    FileUtils.cp_r("#{base_repo_path}/.", target_repo_path)
    FileUtils.chmod_R 0755, target_repo_path
129 130
  end

131 132
  def repos_path
    Gitlab.config.gitlab_shell.repos_path
133
  end
134

135 136 137 138
  def backup_path
    Gitlab.config.backup.path
  end

139 140 141 142 143 144 145 146
  def copy_forked_repo_with_submodules(project)
    base_repo_path = File.expand_path(forked_repo_path_bare)
    target_repo_path = File.expand_path(repos_path + "/#{project.namespace.path}/#{project.path}.git")
    FileUtils.mkdir_p(target_repo_path)
    FileUtils.cp_r("#{base_repo_path}/.", target_repo_path)
    FileUtils.chmod_R 0755, target_repo_path
  end

147 148 149
  private

  def factory_repo_path
150 151 152 153
    @factory_repo_path ||= Rails.root.join('tmp', 'tests', factory_repo_name)
  end

  def factory_repo_path_bare
Robert Speicher committed
154
    "#{factory_repo_path}_bare"
155 156 157 158 159
  end

  def factory_repo_name
    'gitlab-test'
  end
160

161 162 163 164 165 166 167 168 169 170 171 172 173
  def forked_repo_path
    @forked_repo_path ||= Rails.root.join('tmp', 'tests', forked_repo_name)
  end

  def forked_repo_path_bare
    "#{forked_repo_path}_bare"
  end

  def forked_repo_name
    'gitlab-test-fork'
  end


174 175 176
  # Prevent developer git configurations from being persisted to test
  # repositories
  def git_env
177
    { 'GIT_TEMPLATE_DIR' => '' }
178
  end
179
end