test_env.rb 2.98 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
    'flatten-dir'      => 'e56497b',
9 10 11 12 13 14 15 16
    'feature'          => '0b4bc9a',
    'feature_conflict' => 'bb5206f',
    'fix'              => '12d65c8',
    'improve/awesome'  => '5937ac0',
    'markdown'         => '0ed8c6c',
    'master'           => '5937ac0'
  }

17 18
  # Test environment
  #
19
  # See gitlab.yml.example test section for paths
20
  #
21
  def init(opts = {})
22 23 24
    # Disable mailer for spinach tests
    disable_mailer if opts[:mailer] == false

Robert Speicher's avatar
Robert Speicher committed
25
    clean_test_path
26

27
    FileUtils.mkdir_p(repos_path)
28

29 30 31
    # Setup GitLab shell for test instance
    setup_gitlab_shell

32 33
    # Create repository for FactoryGirl.create(:project)
    setup_factory_repo
34 35
  end

36 37
  def disable_mailer
    NotificationService.any_instance.stub(mailer: double.as_null_object)
38
  end
39

40
  def enable_mailer
41
    allow_any_instance_of(NotificationService).to receive(:mailer).and_call_original
42 43
  end

Robert Speicher's avatar
Robert Speicher committed
44 45 46 47 48 49 50 51 52 53 54 55 56
  # 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|
      unless File.basename(entry) =~ /\Agitlab-(shell|test)\z/
        FileUtils.rm_rf(entry)
      end
    end
  end

57
  def setup_gitlab_shell
Robert Speicher's avatar
Robert Speicher committed
58 59 60
    unless File.directory?(Rails.root.join(*%w(tmp tests gitlab-shell)))
      `rake gitlab:shell:install`
    end
61
  end
62

63
  def setup_factory_repo
64
    clone_url = "https://gitlab.com/gitlab-org/#{factory_repo_name}.git"
65

66
    unless File.directory?(factory_repo_path)
Robert Speicher's avatar
Robert Speicher committed
67
      system(*%W(git clone -q #{clone_url} #{factory_repo_path}))
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
    end

    Dir.chdir(factory_repo_path) do
      BRANCH_SHA.each do |branch, sha|
        # 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
85
    end
86 87

    # We must copy bare repositories because we will push to them.
Robert Speicher's avatar
Robert Speicher committed
88
    system(*%W(git clone -q --bare #{factory_repo_path} #{factory_repo_path_bare}))
89 90
  end

91
  def copy_repo(project)
92
    base_repo_path = File.expand_path(factory_repo_path_bare)
93 94 95 96
    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
97 98
  end

99 100
  def repos_path
    Gitlab.config.gitlab_shell.repos_path
101
  end
102 103 104 105

  private

  def factory_repo_path
106 107 108 109
    @factory_repo_path ||= Rails.root.join('tmp', 'tests', factory_repo_name)
  end

  def factory_repo_path_bare
Robert Speicher's avatar
Robert Speicher committed
110
    "#{factory_repo_path}_bare"
111 112 113 114 115
  end

  def factory_repo_name
    'gitlab-test'
  end
116
end