require "test_helper"

class CommandHelpersTest < Test::Unit::TestCase
  setup do
    @module = Vagrant::Command::Helpers
    @command = Class.new(Vagrant::Command::Base) do
      argument :name, :optional => true, :type => :string
    end
  end

  def command(args, env)
    @command.new(args, {}, { :env => env })
  end

  context "initializing environment" do
    should "raise an exception if no environment is given" do
      assert_raises(Vagrant::Errors::CLIMissingEnvironment) { command([], nil) }
    end

    should "not raise an exception if environment is given and setup UI" do
      env = mock_environment
      assert_nothing_raised { command([], env) }
      assert env.ui.is_a?(Vagrant::UI::Shell)
    end
  end

  context "requiring environment" do
    setup do
      @env = mock_environment
    end

    should "raise an exception if no environment" do
      @env.stubs(:root_path).returns(nil)
      assert_raises(Vagrant::Errors::NoEnvironmentError) { command([], @env).require_environment }
    end

    should "not raise an exception if there is an environment" do
      @env.stubs(:root_path).returns(7)
      assert_nothing_raised { command([], @env).require_environment }
    end
  end

  context "vms from args" do
    setup do
      @env = mock_environment
      @env.stubs(:root_path).returns(7)
    end

    should "only calculate the result once" do
      instance = command([], @env)
      result = instance.target_vms
      assert instance.target_vms.equal?(result)
    end

    context "without multivm" do
      setup do
        @env.stubs(:vms).returns({ :one => 1 })
      end

      should "raise an exception if a name is specified" do
        instance = command(["foo"], @env)
        assert_raises(Vagrant::Errors::MultiVMEnvironmentRequired) {
          instance.target_vms
        }
      end

      should "return the VM if no name is specified" do
        instance = command([], @env)
        assert_nothing_raised {
          assert_equal @env.vms.values, instance.target_vms
        }
      end
    end

    context "with multivm" do
      setup do
        @env.stubs(:vms).returns(:one => 1, :two => 2)
      end

      should "return all the VMs if no name is specified" do
        instance = command([], @env)
        assert_equal @env.vms.values, instance.target_vms
      end

      should "return only the specified VM if a name is given" do
        instance = command(["one"], @env)
        assert_equal @env.vms[:one], instance.target_vms.first
      end

      should "raise an exception if an invalid name is given" do
        instance = command(["foo"], @env)
        assert_raises(Vagrant::Errors::VMNotFoundError) {
          instance.target_vms
        }
      end
    end
  end
end