require "test_helper"

class CommandsInitTest < Test::Unit::TestCase
  setup do
    @klass = Vagrant::Commands::Init

    @env = mock_environment
    @instance = @klass.new(@env)
  end

  context "execute" do
    should "create a vagrant file without any args" do
      args = []
      @instance.expects(:create_vagrantfile).with(:default_box => nil, :default_box_url => nil)
      @instance.execute(args)
    end
    context "when any arg is provided" do
      should "create the vagrant file using the first arg as default_box and the second as default_box_url" do
        args = []
        args[0] = "foo"
        args[1] = "foo.box"

        @instance.expects(:create_vagrantfile).with(:default_box => "foo", :default_box_url => "foo.box")
        @instance.execute(args)
      end
    end
  end

  context "creating the vagrantfile" do
    setup do
      @file = mock("file")
      @file.stubs(:write)
      File.stubs(:open).yields(@file)
      @rootfile_path = File.join(Dir.pwd, Vagrant::Environment::ROOTFILE_NAME)

      Vagrant::Util::TemplateRenderer.stubs(:render)
    end

    should "error and exit if a rootfile already exists" do
      File.expects(:exist?).with(@rootfile_path).returns(true)
      @instance.expects(:error_and_exit).with(:rootfile_already_exists).once
      @instance.create_vagrantfile
    end

    should "write to the rootfile path using the template renderer" do
      result = "foo"
      Vagrant::Util::TemplateRenderer.expects(:render).returns(result).once
      @file.expects(:write).with(result).once
      File.expects(:open).with(@rootfile_path, 'w+').yields(@file)

      @instance.create_vagrantfile
    end

    should "use the given base box if given" do
      box = "zooo"
      Vagrant::Util::TemplateRenderer.expects(:render).with(Vagrant::Environment::ROOTFILE_NAME, :default_box => box)
      @instance.create_vagrantfile :default_box => box
    end

    should "use the box_url if given" do
      box_url = "fubar.box"
      Vagrant::Util::TemplateRenderer.expects(:render).with(Vagrant::Environment::ROOTFILE_NAME, :default_box => "base", :default_box_url => "fubar.box")
      @instance.create_vagrantfile :default_box_url => box_url
    end

    should "use the default `base` if no box is given" do
      Vagrant::Util::TemplateRenderer.expects(:render).with(Vagrant::Environment::ROOTFILE_NAME, :default_box => "base")
      @instance.create_vagrantfile
    end
  end
end