test-kitchen 1.6.0 → 1.7.0

data/lib/kitchen/generator/driver_create.rb

@@ -1,177 +1,177 @@
-# -*- encoding: utf-8 -*-
-#
-# Author:: Fletcher Nichol (<fnichol@nichol.ca>)
-#
-# Copyright (C) 2013, Fletcher Nichol
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-require "thor/group"
-require "thor/util"
-
-module Kitchen
-
-  module Generator
-
-    # A generator to create a new Kitchen Driver gem project.
-    #
-    # @author Fletcher Nichol <fnichol@nichol.ca>
-    class DriverCreate < Thor::Group
-
-      include Thor::Actions
-
-      argument :name, :type => :string
-
-      class_option :license,
-        :aliases => "-l",
-        :default => "apachev2",
-        :desc => "License type for gem (apachev2, mit, lgplv3, reserved)"
-
-      # Invoke the command.
-      def create
-        self.class.source_root(Kitchen.source_root.join("templates", "driver"))
-
-        create_core_files
-        create_source_files
-        initialize_git
-      end
-
-      private
-
-      # Creates top-level project files.
-      #
-      # @api private
-      def create_core_files
-        empty_directory(target_dir)
-
-        create_template("CHANGELOG.md.erb", "CHANGELOG.md")
-        create_template("Gemfile.erb", "Gemfile")
-        create_template("Rakefile.erb", "Rakefile")
-        create_template("README.md.erb", "README.md")
-        create_template("gemspec.erb", "#{config[:gem_name]}.gemspec")
-        create_template("license_#{config[:license]}.erb", license_filename)
-        create_template("gitignore.erb", ".gitignore")
-        create_template("tailor.erb", ".tailor")
-        create_template("travis.yml.erb", ".travis.yml")
-        create_file(File.join(target_dir, ".cane"))
-      end
-
-      # Creates source code files.
-      #
-      # @api private
-      def create_source_files
-        empty_directory(File.join(target_dir, "lib/kitchen/driver"))
-
-        create_template(
-          "version.rb.erb",
-          "lib/kitchen/driver/#{name}_version.rb"
-        )
-        create_template(
-          "driver.rb.erb",
-          "lib/kitchen/driver/#{name}.rb"
-        )
-      end
-
-      # Initialize a git repository.
-      #
-      # @api private
-      def initialize_git
-        inside(target_dir) do
-          run("git init", :capture => true)
-          run("git add .", :capture => true)
-        end
-      end
-
-      # Render an ERb template to a destination file.
-      #
-      # @param erb [String] path to an ERb file
-      # @param dest [String] destination path for the rendered template
-      # @api private
-      def create_template(erb, dest)
-        template(erb, File.join(target_dir, dest), config)
-      end
-
-      # @return [String] the path to the gem skeleton project
-      # @api private
-      def target_dir
-        File.join(Dir.pwd, "kitchen-#{name}")
-      end
-
-      # @return [Hash] a configuration hash which can be used by templates as
-      #   context
-      # @api private
-      def config
-        @config ||= {
-          :name => name,
-          :gem_name => "kitchen-#{name}",
-          :gemspec => "kitchen-#{name}.gemspec",
-          :klass_name => ::Thor::Util.camel_case(name),
-          :constant_name => ::Thor::Util.snake_case(name).upcase,
-          :author => author,
-          :email => email,
-          :license => options[:license],
-          :license_string => license_string,
-          :year => Time.now.year
-        }
-      end
-
-      # @return [String] a default author name taken from git configuration if
-      #   found
-      # @api private
-      def author
-        git_user_name = `git config user.name`.chomp
-        git_user_name.empty? ? "TODO: Write your name" : git_user_name
-      end
-
-      # @return [String] a default email address taken from git configuration
-      #   if found
-      # @api private
-      def email
-        git_user_email = `git config user.email`.chomp
-        git_user_email.empty? ? "TODO: Write your email" : git_user_email
-      end
-
-      # @return [String] a rendered license string for a given license
-      # @api private
-      def license_string
-        case options[:license]
-        when "mit" then "MIT"
-        when "apachev2" then "Apache 2.0"
-        when "lgplv3" then "LGPL 3.0"
-        when "reserved" then "All rights reserved"
-        else
-          raise ArgumentError, "No such license #{options[:license]}"
-        end
-      end
-
-      # @return [String] the filename to use for the license file
-      # @api private
-      def license_filename
-        case options[:license]
-        when "mit" then "LICENSE.txt"
-        when "apachev2", "reserved" then "LICENSE"
-        when "lgplv3" then "COPYING"
-        else
-          raise ArgumentError, "No such license #{options[:license]}"
-        end
-      end
-
-      # @return [String] the license comment/preamble
-      # @api private
-      def license_comment
-        @license_comment ||= IO.read(File.join(target_dir, license_filename)).
-          gsub(/^/, "# ").gsub(/\s+$/, "")
-      end
-    end
-  end
-end
+# -*- encoding: utf-8 -*-
+#
+# Author:: Fletcher Nichol (<fnichol@nichol.ca>)
+#
+# Copyright (C) 2013, Fletcher Nichol
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require "thor/group"
+require "thor/util"
+
+module Kitchen
+
+  module Generator
+
+    # A generator to create a new Kitchen Driver gem project.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    class DriverCreate < Thor::Group
+
+      include Thor::Actions
+
+      argument :name, :type => :string
+
+      class_option :license,
+        :aliases => "-l",
+        :default => "apachev2",
+        :desc => "License type for gem (apachev2, mit, lgplv3, reserved)"
+
+      # Invoke the command.
+      def create
+        self.class.source_root(Kitchen.source_root.join("templates", "driver"))
+
+        create_core_files
+        create_source_files
+        initialize_git
+      end
+
+      private
+
+      # Creates top-level project files.
+      #
+      # @api private
+      def create_core_files
+        empty_directory(target_dir)
+
+        create_template("CHANGELOG.md.erb", "CHANGELOG.md")
+        create_template("Gemfile.erb", "Gemfile")
+        create_template("Rakefile.erb", "Rakefile")
+        create_template("README.md.erb", "README.md")
+        create_template("gemspec.erb", "#{config[:gem_name]}.gemspec")
+        create_template("license_#{config[:license]}.erb", license_filename)
+        create_template("gitignore.erb", ".gitignore")
+        create_template("tailor.erb", ".tailor")
+        create_template("travis.yml.erb", ".travis.yml")
+        create_file(File.join(target_dir, ".cane"))
+      end
+
+      # Creates source code files.
+      #
+      # @api private
+      def create_source_files
+        empty_directory(File.join(target_dir, "lib/kitchen/driver"))
+
+        create_template(
+          "version.rb.erb",
+          "lib/kitchen/driver/#{name}_version.rb"
+        )
+        create_template(
+          "driver.rb.erb",
+          "lib/kitchen/driver/#{name}.rb"
+        )
+      end
+
+      # Initialize a git repository.
+      #
+      # @api private
+      def initialize_git
+        inside(target_dir) do
+          run("git init", :capture => true)
+          run("git add .", :capture => true)
+        end
+      end
+
+      # Render an ERb template to a destination file.
+      #
+      # @param erb [String] path to an ERb file
+      # @param dest [String] destination path for the rendered template
+      # @api private
+      def create_template(erb, dest)
+        template(erb, File.join(target_dir, dest), config)
+      end
+
+      # @return [String] the path to the gem skeleton project
+      # @api private
+      def target_dir
+        File.join(Dir.pwd, "kitchen-#{name}")
+      end
+
+      # @return [Hash] a configuration hash which can be used by templates as
+      #   context
+      # @api private
+      def config
+        @config ||= {
+          :name => name,
+          :gem_name => "kitchen-#{name}",
+          :gemspec => "kitchen-#{name}.gemspec",
+          :klass_name => ::Thor::Util.camel_case(name),
+          :constant_name => ::Thor::Util.snake_case(name).upcase,
+          :author => author,
+          :email => email,
+          :license => options[:license],
+          :license_string => license_string,
+          :year => Time.now.year
+        }
+      end
+
+      # @return [String] a default author name taken from git configuration if
+      #   found
+      # @api private
+      def author
+        git_user_name = `git config user.name`.chomp
+        git_user_name.empty? ? "TODO: Write your name" : git_user_name
+      end
+
+      # @return [String] a default email address taken from git configuration
+      #   if found
+      # @api private
+      def email
+        git_user_email = `git config user.email`.chomp
+        git_user_email.empty? ? "TODO: Write your email" : git_user_email
+      end
+
+      # @return [String] a rendered license string for a given license
+      # @api private
+      def license_string
+        case options[:license]
+        when "mit" then "MIT"
+        when "apachev2" then "Apache 2.0"
+        when "lgplv3" then "LGPL 3.0"
+        when "reserved" then "All rights reserved"
+        else
+          raise ArgumentError, "No such license #{options[:license]}"
+        end
+      end
+
+      # @return [String] the filename to use for the license file
+      # @api private
+      def license_filename
+        case options[:license]
+        when "mit" then "LICENSE.txt"
+        when "apachev2", "reserved" then "LICENSE"
+        when "lgplv3" then "COPYING"
+        else
+          raise ArgumentError, "No such license #{options[:license]}"
+        end
+      end
+
+      # @return [String] the license comment/preamble
+      # @api private
+      def license_comment
+        @license_comment ||= IO.read(File.join(target_dir, license_filename)).
+          gsub(/^/, "# ").gsub(/\s+$/, "")
+      end
+    end
+  end
+end

data/lib/kitchen/generator/init.rb

@@ -1,296 +1,296 @@
-# -*- encoding: utf-8 -*-
-#
-# Author:: Fletcher Nichol (<fnichol@nichol.ca>)
-#
-# Copyright (C) 2013, Fletcher Nichol
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-require "rubygems/gem_runner"
-require "thor/group"
-
-module Kitchen
-
-  module Generator
-
-    # A project initialization generator, to help prepare a cookbook project
-    # for testing with Kitchen.
-    #
-    # @author Fletcher Nichol <fnichol@nichol.ca>
-    class Init < Thor::Group
-
-      include Thor::Actions
-
-      class_option :driver,
-        :type => :array,
-        :aliases => "-D",
-        :default => "kitchen-vagrant",
-        :desc => <<-D.gsub(/^\s+/, "").gsub(/\n/, " ")
-          One or more Kitchen Driver gems to be installed or added to a
-          Gemfile
-        D
-
-      class_option :provisioner,
-        :type => :string,
-        :aliases => "-P",
-        :default => "chef_solo",
-        :desc => <<-D.gsub(/^\s+/, "").gsub(/\n/, " ")
-          The default Kitchen Provisioner to use
-        D
-
-      class_option :create_gemfile,
-        :type => :boolean,
-        :default => false,
-        :desc => <<-D.gsub(/^\s+/, "").gsub(/\n/, " ")
-          Whether or not to create a Gemfile if one does not exist.
-          Default: false
-        D
-
-      # Invoke the command.
-      def init
-        self.class.source_root(Kitchen.source_root.join("templates", "init"))
-
-        create_kitchen_yaml
-        create_chefignore
-        prepare_rakefile
-        prepare_thorfile
-        create_test_dir
-        prepare_gitignore
-        prepare_gemfile
-        add_drivers
-        display_bundle_message
-      end
-
-      private
-
-      # Creates the `.kitchen.yml` file.
-      #
-      # @api private
-      def create_kitchen_yaml
-        cookbook_name = if File.exist?(File.expand_path("metadata.rb"))
-          MetadataChopper.extract("metadata.rb").first
-        else
-          nil
-        end
-        run_list = cookbook_name ? "recipe[#{cookbook_name}::default]" : nil
-        driver_plugin = Array(options[:driver]).first || "dummy"
-
-        template("kitchen.yml.erb", ".kitchen.yml",
-          :driver_plugin => driver_plugin.sub(/^kitchen-/, ""),
-          :provisioner => options[:provisioner],
-          :run_list => Array(run_list)
-        )
-      end
-
-      # Creates the `chefignore` file.
-      #
-      # @api private
-      def create_chefignore
-        template("chefignore.erb", "chefignore")
-      end
-
-      # @return [true,false] whether or not a Gemfile needs to be initialized
-      # @api private
-      def init_gemfile?
-        File.exist?(File.join(destination_root, "Gemfile")) ||
-          options[:create_gemfile]
-      end
-
-      # @return [true,false] whether or not a Rakefile needs to be initialized
-      # @api private
-      def init_rakefile?
-        File.exist?(File.join(destination_root, "Rakefile")) &&
-          not_in_file?("Rakefile", %r{require 'kitchen/rake_tasks'})
-      end
-
-      # @return [true,false] whether or not a Thorfile needs to be initialized
-      # @api private
-      def init_thorfile?
-        File.exist?(File.join(destination_root, "Thorfile")) &&
-          not_in_file?("Thorfile", %r{require 'kitchen/thor_tasks'})
-      end
-
-      # @return [true,false] whether or not a test directory needs to be
-      #   initialized
-      # @api private
-      def init_test_dir?
-        Dir.glob("test/integration/*").select { |d| File.directory?(d) }.empty?
-      end
-
-      # @return [true,false] whether or not a `.gitignore` file needs to be
-      #   initialized
-      # @api private
-      def init_git?
-        File.directory?(File.join(destination_root, ".git"))
-      end
-
-      # Prepares a Rakefile.
-      #
-      # @api private
-      def prepare_rakefile
-        return unless init_rakefile?
-
-        rakedoc = <<-RAKE.gsub(/^ {10}/, "")
-
-          begin
-            require 'kitchen/rake_tasks'
-            Kitchen::RakeTasks.new
-          rescue LoadError
-            puts '>>>>> Kitchen gem not loaded, omitting tasks' unless ENV['CI']
-          end
-        RAKE
-        append_to_file(File.join(destination_root, "Rakefile"), rakedoc)
-      end
-
-      # Prepares a Thorfile.
-      #
-      # @api private
-      def prepare_thorfile
-        return unless init_thorfile?
-
-        thordoc = <<-THOR.gsub(/^ {10}/, "")
-
-          begin
-            require 'kitchen/thor_tasks'
-            Kitchen::ThorTasks.new
-          rescue LoadError
-            puts '>>>>> Kitchen gem not loaded, omitting tasks' unless ENV['CI']
-          end
-        THOR
-        append_to_file(File.join(destination_root, "Thorfile"), thordoc)
-      end
-
-      # Create the default test directory
-      #
-      # @api private
-      def create_test_dir
-        empty_directory "test/integration/default" if init_test_dir?
-      end
-
-      # Prepares the .gitignore file
-      #
-      # @api private
-      def prepare_gitignore
-        return unless init_git?
-
-        append_to_gitignore(".kitchen/")
-        append_to_gitignore(".kitchen.local.yml")
-      end
-
-      # Appends a line to the .gitignore file.
-      #
-      # @api private
-      def append_to_gitignore(line)
-        create_file(".gitignore") unless File.exist?(File.join(destination_root, ".gitignore"))
-
-        if IO.readlines(File.join(destination_root, ".gitignore")).grep(%r{^#{line}}).empty?
-          append_to_file(".gitignore", "#{line}\n")
-        end
-      end
-
-      # Prepares a Gemfile.
-      #
-      # @api private
-      def prepare_gemfile
-        return unless init_gemfile?
-
-        create_gemfile_if_missing
-        add_gem_to_gemfile
-      end
-
-      # Creates a Gemfile if missing
-      #
-      # @api private
-      def create_gemfile_if_missing
-        unless File.exist?(File.join(destination_root, "Gemfile"))
-          create_file("Gemfile", %{source "https://rubygems.org"\n\n})
-        end
-      end
-
-      # Appends entries to a Gemfile.
-      #
-      # @api private
-      def add_gem_to_gemfile
-        if not_in_file?("Gemfile", %r{gem ('|")test-kitchen('|")})
-          append_to_file("Gemfile", %{gem "test-kitchen"\n})
-          @display_bundle_msg = true
-        end
-      end
-
-      # Appends driver gems to a Gemfile or installs them.
-      #
-      # @api private
-      def add_drivers
-        return if options[:driver].nil? || options[:driver].empty?
-
-        Array(options[:driver]).each do |driver_gem|
-          if File.exist?(File.join(destination_root, "Gemfile")) || options[:create_gemfile]
-            add_driver_to_gemfile(driver_gem)
-          else
-            install_gem(driver_gem)
-          end
-        end
-      end
-
-      # Appends a driver gem to a Gemfile.
-      #
-      # @api private
-      def add_driver_to_gemfile(driver_gem)
-        if not_in_file?("Gemfile", %r{gem ('|")#{driver_gem}('|")})
-          append_to_file("Gemfile", %{gem "#{driver_gem}"\n})
-          @display_bundle_msg = true
-        end
-      end
-
-      # Installs a driver gem.
-      #
-      # @api private
-      def install_gem(driver_gem)
-        unbundlerize { Gem::GemRunner.new.run(["install", driver_gem]) }
-      rescue Gem::SystemExitException => e
-        raise unless e.exit_code == 0
-      end
-
-      # Displays a bundle warning message to the user.
-      #
-      # @api private
-      def display_bundle_message
-        if @display_bundle_msg
-          say "You must run `bundle install' to fetch any new gems.", :red
-        end
-      end
-
-      # Determines whether or not a pattern is found in a file.
-      #
-      # @param filename [String] filename to read
-      # @param regexp [Regexp] a regular expression
-      # @return [true,false] whether or not a pattern is found in a file
-      # @api private
-      def not_in_file?(filename, regexp)
-        IO.readlines(File.join(destination_root, filename)).grep(regexp).empty?
-      end
-
-      # Save off any Bundler/Ruby-related environment variables so that the
-      # yielded block can run "bundler-free" (and restore at the end).
-      #
-      # @api private
-      def unbundlerize
-        keys = ENV.keys.select { |key| key =~ /^BUNDLER?_/ } + %w[RUBYOPT]
-
-        keys.each { |key| ENV["__#{key}"] = ENV[key]; ENV.delete(key) }
-        yield
-        keys.each { |key| ENV[key] = ENV.delete("__#{key}") }
-      end
-    end
-  end
-end
+# -*- encoding: utf-8 -*-
+#
+# Author:: Fletcher Nichol (<fnichol@nichol.ca>)
+#
+# Copyright (C) 2013, Fletcher Nichol
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require "rubygems/gem_runner"
+require "thor/group"
+
+module Kitchen
+
+  module Generator
+
+    # A project initialization generator, to help prepare a cookbook project
+    # for testing with Kitchen.
+    #
+    # @author Fletcher Nichol <fnichol@nichol.ca>
+    class Init < Thor::Group
+
+      include Thor::Actions
+
+      class_option :driver,
+        :type => :array,
+        :aliases => "-D",
+        :default => "kitchen-vagrant",
+        :desc => <<-D.gsub(/^\s+/, "").gsub(/\n/, " ")
+          One or more Kitchen Driver gems to be installed or added to a
+          Gemfile
+        D
+
+      class_option :provisioner,
+        :type => :string,
+        :aliases => "-P",
+        :default => "chef_solo",
+        :desc => <<-D.gsub(/^\s+/, "").gsub(/\n/, " ")
+          The default Kitchen Provisioner to use
+        D
+
+      class_option :create_gemfile,
+        :type => :boolean,
+        :default => false,
+        :desc => <<-D.gsub(/^\s+/, "").gsub(/\n/, " ")
+          Whether or not to create a Gemfile if one does not exist.
+          Default: false
+        D
+
+      # Invoke the command.
+      def init
+        self.class.source_root(Kitchen.source_root.join("templates", "init"))
+
+        create_kitchen_yaml
+        create_chefignore
+        prepare_rakefile
+        prepare_thorfile
+        create_test_dir
+        prepare_gitignore
+        prepare_gemfile
+        add_drivers
+        display_bundle_message
+      end
+
+      private
+
+      # Creates the `.kitchen.yml` file.
+      #
+      # @api private
+      def create_kitchen_yaml
+        cookbook_name = if File.exist?(File.expand_path("metadata.rb"))
+          MetadataChopper.extract("metadata.rb").first
+        else
+          nil
+        end
+        run_list = cookbook_name ? "recipe[#{cookbook_name}::default]" : nil
+        driver_plugin = Array(options[:driver]).first || "dummy"
+
+        template("kitchen.yml.erb", ".kitchen.yml",
+          :driver_plugin => driver_plugin.sub(/^kitchen-/, ""),
+          :provisioner => options[:provisioner],
+          :run_list => Array(run_list)
+        )
+      end
+
+      # Creates the `chefignore` file.
+      #
+      # @api private
+      def create_chefignore
+        template("chefignore.erb", "chefignore")
+      end
+
+      # @return [true,false] whether or not a Gemfile needs to be initialized
+      # @api private
+      def init_gemfile?
+        File.exist?(File.join(destination_root, "Gemfile")) ||
+          options[:create_gemfile]
+      end
+
+      # @return [true,false] whether or not a Rakefile needs to be initialized
+      # @api private
+      def init_rakefile?
+        File.exist?(File.join(destination_root, "Rakefile")) &&
+          not_in_file?("Rakefile", %r{require 'kitchen/rake_tasks'})
+      end
+
+      # @return [true,false] whether or not a Thorfile needs to be initialized
+      # @api private
+      def init_thorfile?
+        File.exist?(File.join(destination_root, "Thorfile")) &&
+          not_in_file?("Thorfile", %r{require 'kitchen/thor_tasks'})
+      end
+
+      # @return [true,false] whether or not a test directory needs to be
+      #   initialized
+      # @api private
+      def init_test_dir?
+        Dir.glob("test/integration/*").select { |d| File.directory?(d) }.empty?
+      end
+
+      # @return [true,false] whether or not a `.gitignore` file needs to be
+      #   initialized
+      # @api private
+      def init_git?
+        File.directory?(File.join(destination_root, ".git"))
+      end
+
+      # Prepares a Rakefile.
+      #
+      # @api private
+      def prepare_rakefile
+        return unless init_rakefile?
+
+        rakedoc = <<-RAKE.gsub(/^ {10}/, "")
+
+          begin
+            require 'kitchen/rake_tasks'
+            Kitchen::RakeTasks.new
+          rescue LoadError
+            puts '>>>>> Kitchen gem not loaded, omitting tasks' unless ENV['CI']
+          end
+        RAKE
+        append_to_file(File.join(destination_root, "Rakefile"), rakedoc)
+      end
+
+      # Prepares a Thorfile.
+      #
+      # @api private
+      def prepare_thorfile
+        return unless init_thorfile?
+
+        thordoc = <<-THOR.gsub(/^ {10}/, "")
+
+          begin
+            require 'kitchen/thor_tasks'
+            Kitchen::ThorTasks.new
+          rescue LoadError
+            puts '>>>>> Kitchen gem not loaded, omitting tasks' unless ENV['CI']
+          end
+        THOR
+        append_to_file(File.join(destination_root, "Thorfile"), thordoc)
+      end
+
+      # Create the default test directory
+      #
+      # @api private
+      def create_test_dir
+        empty_directory "test/integration/default" if init_test_dir?
+      end
+
+      # Prepares the .gitignore file
+      #
+      # @api private
+      def prepare_gitignore
+        return unless init_git?
+
+        append_to_gitignore(".kitchen/")
+        append_to_gitignore(".kitchen.local.yml")
+      end
+
+      # Appends a line to the .gitignore file.
+      #
+      # @api private
+      def append_to_gitignore(line)
+        create_file(".gitignore") unless File.exist?(File.join(destination_root, ".gitignore"))
+
+        if IO.readlines(File.join(destination_root, ".gitignore")).grep(%r{^#{line}}).empty?
+          append_to_file(".gitignore", "#{line}\n")
+        end
+      end
+
+      # Prepares a Gemfile.
+      #
+      # @api private
+      def prepare_gemfile
+        return unless init_gemfile?
+
+        create_gemfile_if_missing
+        add_gem_to_gemfile
+      end
+
+      # Creates a Gemfile if missing
+      #
+      # @api private
+      def create_gemfile_if_missing
+        unless File.exist?(File.join(destination_root, "Gemfile"))
+          create_file("Gemfile", %{source "https://rubygems.org"\n\n})
+        end
+      end
+
+      # Appends entries to a Gemfile.
+      #
+      # @api private
+      def add_gem_to_gemfile
+        if not_in_file?("Gemfile", %r{gem ('|")test-kitchen('|")})
+          append_to_file("Gemfile", %{gem "test-kitchen"\n})
+          @display_bundle_msg = true
+        end
+      end
+
+      # Appends driver gems to a Gemfile or installs them.
+      #
+      # @api private
+      def add_drivers
+        return if options[:driver].nil? || options[:driver].empty?
+
+        Array(options[:driver]).each do |driver_gem|
+          if File.exist?(File.join(destination_root, "Gemfile")) || options[:create_gemfile]
+            add_driver_to_gemfile(driver_gem)
+          else
+            install_gem(driver_gem)
+          end
+        end
+      end
+
+      # Appends a driver gem to a Gemfile.
+      #
+      # @api private
+      def add_driver_to_gemfile(driver_gem)
+        if not_in_file?("Gemfile", %r{gem ('|")#{driver_gem}('|")})
+          append_to_file("Gemfile", %{gem "#{driver_gem}"\n})
+          @display_bundle_msg = true
+        end
+      end
+
+      # Installs a driver gem.
+      #
+      # @api private
+      def install_gem(driver_gem)
+        unbundlerize { Gem::GemRunner.new.run(["install", driver_gem]) }
+      rescue Gem::SystemExitException => e
+        raise unless e.exit_code == 0
+      end
+
+      # Displays a bundle warning message to the user.
+      #
+      # @api private
+      def display_bundle_message
+        if @display_bundle_msg
+          say "You must run `bundle install' to fetch any new gems.", :red
+        end
+      end
+
+      # Determines whether or not a pattern is found in a file.
+      #
+      # @param filename [String] filename to read
+      # @param regexp [Regexp] a regular expression
+      # @return [true,false] whether or not a pattern is found in a file
+      # @api private
+      def not_in_file?(filename, regexp)
+        IO.readlines(File.join(destination_root, filename)).grep(regexp).empty?
+      end
+
+      # Save off any Bundler/Ruby-related environment variables so that the
+      # yielded block can run "bundler-free" (and restore at the end).
+      #
+      # @api private
+      def unbundlerize
+        keys = ENV.keys.select { |key| key =~ /^BUNDLER?_/ } + %w[RUBYOPT]
+
+        keys.each { |key| ENV["__#{key}"] = ENV[key]; ENV.delete(key) }
+        yield
+        keys.each { |key| ENV[key] = ENV.delete("__#{key}") }
+      end
+    end
+  end
+end