From fd622511578ca277c680ecef7fca8851bb2149d0 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Thu, 5 Dec 2013 14:24:10 -0800
Subject: [PATCH] update rake tasks for acceptance testing

---
 tasks/acceptance.rake | 119 +++++-------------------------------------
 tasks/test.rake       |   5 --
 2 files changed, 14 insertions(+), 110 deletions(-)

diff --git a/tasks/acceptance.rake b/tasks/acceptance.rake
index 2175bb0d3..813ff86d8 100644
--- a/tasks/acceptance.rake
+++ b/tasks/acceptance.rake
@@ -1,113 +1,22 @@
-require 'digest/sha1'
-require 'net/http'
-require 'pathname'
-require 'uri'
-require 'yaml'
-
-require 'childprocess'
-
-require 'vagrant/util/file_checksum'
-
 namespace :acceptance do
-  desc "Downloads the boxes required for running the acceptance tests."
-  task :boxes, :directory do |t, args|
-    # Create the directory where the boxes will be downloaded
-    box_dir = Pathname.new(args[:directory] || File.expand_path("../../boxes", __FILE__))
-    box_dir.mkpath
-    puts "Boxes will be placed in: #{box_dir}"
-
-    # Load the required boxes
-    boxes = YAML.load_file(File.expand_path("../../test/config/acceptance_boxes.yml", __FILE__))
-
-    boxes.each do |box|
-      puts "Box: #{box["name"]}"
-      box_file = box_dir.join("#{box["name"]}.box")
-      checksum = FileChecksum.new(box_file, Digest::SHA1)
-
-      # If the box exists, we need to check the checksum and determine if we need
-      # to redownload the file.
-      if box_file.exist?
-        print "Box exists, checking SHA1 sum... "
-        if checksum.checksum == box["checksum"]
-          print "OK\n"
-          next
-        else
-          print "FAIL\n"
-        end
-      end
-
-      # Download the file. Note that this has no error checking and just uses
-      # pure net/http. There could be a better way.
-      puts "Downloading: #{box["url"]}"
-      destination = box_file.open("wb")
-      uri = URI.parse(box["url"])
-      Net::HTTP.new(uri.host, uri.port).start do |http|
-        http.request_get(uri.request_uri) do |response|
-          total = response.content_length
-          progress = 0
-          count = 0
-
-          response.read_body do |segment|
-            # Really elementary progress meter
-            progress += segment.length
-            count += 1
-            puts "Progress: #{(progress.to_f / total.to_f) * 100}%" if count % 300 == 0
-
-            # Write out to the destination file
-            destination.write(segment)
-          end
-        end
-      end
-
-      destination.close
-
-      # Check the checksum of the new file to verify that it
-      # downloaded properly. This shouldn't happen, but it can!
-      if checksum.checksum != box["checksum"]
-        puts "Checksum didn't match! Download was corrupt!"
-        abort
-      end
-    end
+  desc "shows components that can be tested"
+  task :components do
+    exec("vagrant-spec components --config=vagrant-spec.config.rb")
   end
 
-  desc "Generates the configuration for acceptance tests from current source."
-  task :config, :box_dir do |t, args|
-    require File.expand_path("../../lib/vagrant/version", __FILE__)
-    require File.expand_path('../../test/support/tempdir', __FILE__)
+  desc "runs acceptance tests"
+  task :run do
+    args = [
+      "--config=vagrant-spec.config.rb",
+    ]
 
-    # Get the directory for the boxes
-    box_dir = Pathname.new(args[:box_dir] || File.expand_path("../../boxes", __FILE__))
-
-    # Generate the binstubs for the Vagrant binary
-    tempdir = Tempdir.new
-    process = ChildProcess.build("bundle", "install", "--binstubs", tempdir.path)
-    process.io.inherit!
-    process.start
-    process.poll_for_exit(64000)
-    if process.exit_code != 0
-      # Bundle install failed...
-      puts "Bundle install failed!"
-      abort
+    if ENV["COMPONENTS"]
+      args << "--components=\"#{ENV["COMPONENTS"]}\""
     end
 
-    # Generate the actual configuration
-    config = {
-      "vagrant_path" => File.join(tempdir.path, "vagrant"),
-      "vagrant_version" => Vagrant::VERSION,
-      "env" => {
-        "BUNDLE_GEMFILE" => File.expand_path("../../Gemfile", __FILE__)
-      },
-      "box_directory" => box_dir.to_s
-    }
-
-    File.open("acceptance_config.yml", "w+") do |f|
-      f.write(YAML.dump(config))
-    end
-
-    puts <<-OUTPUT
-Acceptance test configuration is now in this directory in
-"acceptance_config.yml." Set your ACCEPTANCE_CONFIG environmental
-variable to this file and run any of the acceptance tests now.
-OUTPUT
+    command = "vagrant-spec test #{args.join(" ")}"
+    puts command
+    puts
+    exec(command)
   end
 end
diff --git a/tasks/test.rake b/tasks/test.rake
index da984db0d..18e4395b1 100644
--- a/tasks/test.rake
+++ b/tasks/test.rake
@@ -11,9 +11,4 @@ namespace :test do
     t.libs << "test/unit_legacy"
     t.pattern = "test/unit_legacy/**/*_test.rb"
   end
-
-  RSpec::Core::RakeTask.new(:acceptance) do |t|
-    $: << File.expand_path("../test/acceptance", __FILE__)
-    t.pattern = "test/acceptance/**/*_test.rb"
-  end
 end