vagrant/test/acceptance/vagrant_test.rb

require File.expand_path("../base", __FILE__)
require "acceptance/support/matchers/have_color"

describe "vagrant and color output" do
  include_context "acceptance"

  # This is a check to see if the `expect` program is installed on this
  # computer. Some tests require this and if this doesn't exist then the
  # test itself will be skipped.
  def self.has_expect?
    `which expect`
    $?.success?
  end

  it "outputs color if there is a TTY", :if => has_expect? do
    environment.workdir.join("color.exp").open("w+") do |f|
      f.puts(<<-SCRIPT)
spawn #{environment.replace_command("vagrant")} status
expect default {}
SCRIPT
    end

    result = execute("expect", "color.exp")
    result.stdout.should have_color
  end

  it "doesn't output color if there is a TTY but --no-color is present", :if => has_expect? do
    environment.workdir.join("color.exp").open("w+") do |f|
      f.puts(<<-SCRIPT)
spawn #{environment.replace_command("vagrant")} status --no-color
expect default {}
SCRIPT
    end

    result = execute("expect", "color.exp")
    result.stdout.should_not have_color
  end

  it "doesn't output color in the absense of a TTY" do
    # This should always output an error, which on a TTY would
    # output color. We check that this doesn't output color.
    # If `vagrant status` itself is broken, another acceptance test
    # should catch that. We just assume it works here.
    result = execute("vagrant", "status")
    result.stdout.should_not have_color
  end
end