vagrant/test/test_helper.rb

begin
  require File.expand_path('../.bundle/environment', __FILE__)
rescue LoadError
  # Fallback on doing the resolve at runtime.
  require "rubygems"
end

# ruby-debug, not necessary, but useful if we have it
begin
  require 'ruby-debug'
rescue LoadError; end


require File.join(File.dirname(__FILE__), '..', 'lib', 'vagrant')
require 'contest'
require 'mocha'

class Test::Unit::TestCase
  # Mocks an environment, setting it up with the given config.
  def mock_environment
    environment = Vagrant::Environment.new

    Vagrant::Config.reset!(environment)

    Vagrant::Config.run do |config|
      config.vagrant.dotfile_name = ".vagrant"
      config.vagrant.log_output = nil

      config.ssh.username = "foo"
      config.ssh.password = "bar"
      config.ssh.host = "baz"
      config.ssh.forwarded_port_key = "ssh"
      config.ssh.max_tries = 10
      config.ssh.timeout = 10
      config.ssh.private_key_path = '~/foo'

      config.vm.box = "foo"
      config.vm.box_ovf = "box.ovf"
      config.vm.base_mac = "42"
      config.vm.project_directory = "/vagrant"
      config.vm.rsync_project_directory = false
      config.vm.disk_image_format = 'VMDK'
      config.vm.forward_port("ssh", 22, 2222)
      config.vm.shared_folder_uid = nil
      config.vm.shared_folder_gid = nil
      config.vm.system = :linux
      config.vm.rsync_script = "/foo"
      config.vm.rsync_crontab_entry_file = "/tmp/foo"

      config.package.name = 'vagrant'
      config.package.extension = '.box'

      # Chef
      config.chef.chef_server_url = "http://localhost:4000"
      config.chef.validation_key_path = "validation.pem"
      config.chef.client_key_path = "/zoo/foo/bar.pem"
      config.chef.node_name = "baz"
      config.chef.cookbooks_path = "cookbooks"
      config.chef.provisioning_path = "/tmp/vagrant-chef"
      config.chef.log_level = :info
      config.chef.json = {
        :recipes => ["vagrant_main"]
      }

      config.vagrant.home = '~/.home'
    end

    if block_given?
      Vagrant::Config.run do |config|
        yield config
      end
    end

    config = Vagrant::Config.execute!

    environment.instance_variable_set(:@config, config)

    # Setup the logger. We create it then reset it so that subsequent
    # calls will recreate it for us.
    environment.load_logger!
    environment.logger.class.reset_singleton_logger!
    environment.logger.stubs(:flush_progress)
    environment.logger.stubs(:cl_reset).returns("")

    environment
  end

  # Sets up the mocks for a VM
  def mock_vm(env=nil)
    env ||= mock_environment
    vm = Vagrant::VM.new
    vm.stubs(:env).returns(env)
    vm.stubs(:ssh).returns(Vagrant::SSH.new(vm.env))
    vm
  end

  # Sets up the mocks and instantiates an action for testing
  def mock_action(action_klass, *args)
    vm = mock("vboxvm")
    mock_vm = mock("vm")
    action = action_klass.new(mock_vm, *args)
    stub_default_action_dependecies(action)

    mock_vm.stubs(:name).returns("foo")
    mock_vm.stubs(:vm).returns(vm)
    mock_vm.stubs(:vm=)
    mock_vm.stubs(:invoke_callback)
    mock_vm.stubs(:invoke_around_callback).yields
    mock_vm.stubs(:actions).returns([action])
    mock_vm.stubs(:env).returns(mock_environment)
    mock_vm.env.stubs(:logger).returns(quiet_logger("mock"))

    mock_ssh = Vagrant::SSH.new(mock_vm.env)
    mock_ssh.stubs(:execute)

    mock_vm.stubs(:ssh).returns(mock_ssh)

    vm.stubs(:env).returns(mock_vm.env)

    [mock_vm, vm, action]
  end

  # Returns a resource logger which is safe for tests
  def quiet_logger(resource, env=nil)
    logger = Vagrant::ResourceLogger.new(resource, env)
    logger.stubs(:flush_progress)
    logger.stubs(:cl_reset).returns("")
    logger
  end

  # Returns a linux system
  def linux_system(vm)
    Vagrant::Systems::Linux.new(vm)
  end

  def stub_default_action_dependecies(mock)
    mock.stubs(:precedes).returns([])
    mock.stubs(:follows).returns([])
  end

  # Sets up the mocks and stubs for a downloader
  def mock_downloader(downloader_klass)
    tempfile = mock("tempfile")
    tempfile.stubs(:write)

    [downloader_klass.new(mock_environment), tempfile]
  end
end