module TestFileHelpers
  def create_files file_array
    file_array.each do |f|
      FileUtils.mkdir_p File.dirname(f)
      FileUtils.touch f
    end
  end

  def fog_file_contents
    { :default => { :aws_access_key_id => "IMANACCESSKEY",
                    :aws_secret_access_key => "supersekritkey",
                    :aix_hypervisor_server => "aix_hypervisor.labs.net",
                    :aix_hypervisor_username => "aixer",
                    :aix_hypervisor_keyfile => "/Users/user/.ssh/id_rsa-acceptance",
                    :solaris_hypervisor_server => "solaris_hypervisor.labs.net",
                    :solaris_hypervisor_username => "harness",
                    :solaris_hypervisor_keyfile => "/Users/user/.ssh/id_rsa-old.private",
                    :solaris_hypervisor_vmpath => "rpoooool/zs",
                    :solaris_hypervisor_snappaths => ["rpoooool/USER/z0"],
                    :vsphere_server => "vsphere.labs.net",
                    :vsphere_username => "vsphere@labs.com",
                    :vsphere_password => "supersekritpassword",
    } }
  end

end

module HostHelpers
  HOST_DEFAULTS = { :platform => 'unix',
                    :snapshot => 'pe',
                    :box => 'box_name',
                    :roles => ['agent'],
                    :snapshot => 'snap',
                    :ip => 'default.ip.address',
                    :box => 'default_box_name',
                    :box_url => 'http://default.box.url',
  }

  HOST_NAME     = "vm%d"
  HOST_SNAPSHOT = "snapshot%d"
  HOST_IP       = "ip.address.for.%s"
  HOST_BOX      = "%s_of_my_box"
  HOST_BOX_URL  = "http://address.for.my.box.%s"
  HOST_TEMPLATE = "%s_has_a_template"

  def logger
    double( 'logger' ).as_null_object
  end

  def make_opts
    opts = StringifyHash.new
    opts.merge( { :logger => logger,
                  :host_config => 'sample.config',
                  :type => nil,
                  :pooling_api => 'http://vcloud.delivery.puppetlabs.net/',
                  :datastore => 'instance0',
                  :folder => 'Delivery/Quality Assurance/Staging/Dynamic',
                  :resourcepool => 'delivery/Quality Assurance/Staging/Dynamic',
                  :gce_project => 'beaker-compute',
                  :gce_keyfile => '/path/to/keyfile.p12',
                  :gce_password => 'notasecret',
                  :gce_email => '12345678910@developer.gserviceaccount.com',
    } )
  end

  def generate_result (name, opts )
    result = double( 'result' )
    stdout = opts.has_key?(:stdout) ? opts[:stdout] : name
    stderr = opts.has_key?(:stderr) ? opts[:stderr] : name
    exit_code = opts.has_key?(:exit_code) ? opts[:exit_code] :  0
    exit_code = [exit_code].flatten
    allow( result ).to receive( :stdout ).and_return( stdout )
    allow( result ).to receive( :stderr ).and_return( stderr )
    allow( result ).to receive( :exit_code ).and_return( *exit_code )
    result
  end

  def make_host_opts name, opts
    make_opts.merge( { 'HOSTS' => { name => opts } } ).merge( opts )
  end

  def make_host name, host_hash
    host_hash = StringifyHash.new.merge(HOST_DEFAULTS.merge(host_hash))

    host = make_opts.merge(host_hash)

    allow(host).to receive( :name ).and_return( name )
    allow(host).to receive( :to_s ).and_return( name )
    allow(host).to receive( :exec ).and_return( generate_result( name, host_hash ) )
    host
  end

  def make_hosts preset_opts = {}, amt = 3
    hosts = []
    (1..amt).each do |num|
      name = HOST_NAME % num
      opts = { :snapshot => HOST_SNAPSHOT % num,
               :ip => HOST_IP % name,
               :template => HOST_TEMPLATE % name,
               :box => HOST_BOX % name,
               :box_url => HOST_BOX_URL % name }.merge( preset_opts )
      hosts << make_host(name, opts)
    end
    hosts
  end

  def make_instance instance_data = {}
    OpenStruct.new instance_data
  end

end