class ChefSpec::ServerRunner

def apply_chef_config!
  super
  Chef::Config[:client_key]       = client_key
  Chef::Config[:client_name]      = "chefspec"
  Chef::Config[:node_name]        = "chefspec"
  Chef::Config[:solo]             = false
  Chef::Config[:solo_legacy_mode] = false
  Chef::Config[:chef_server_url]  = server.url
  Chef::Config[:http_retry_count] = 0
end

def client_key
  tmp = Dir.mktmpdir
  path = File.join(tmp, "client.pem")
  File.open(path, "wb") { |f| f.write(ChefZero::PRIVATE_KEY) }
  at_exit { FileUtils.rm_rf(tmp) }
  path
end

def converge(*recipe_names)
  ChefSpec::ZeroServer.upload_cookbooks!
  super do
    yield if block_given?
    # Save the node back to the server for searching purposes
    client.register
    node.save
  end
end

def initialize(options = {})
  # Unlike the SoloRunner, the file_cache_path needs to remain consistent
  # for every Chef run or else the Chef client tries to loads the same
  # cookbook multiple times and will encounter deprecated logic when
  # creating LWRPs. It also slows down the entire process.
  options[:file_cache_path] ||= RSpec.configuration.file_cache_path ||
    ChefSpec::FileCachePathProxy.instance.file_cache_path
  # Call super, but do not pass in the block because we want to customize
  # our yielding.
  super(options, &nil)
  # Unlike the SoloRunner, the node AND server object are yielded for
  # customization
  yield node, self if block_given?
end