require 'beaker-puppet'

test_name 'validate host stubbing behavior'

def get_hosts_file(host)
  if host['platform'] =~ /win/
    'C:\\\\Windows\\\\System32\\\\Drivers\\\\etc\\\\hosts'
  else
    '/etc/hosts'
  end
end

step 'verify stub_host_on' do
  step 'should add entry to hosts file' do
    hosts.each do |host|
      stub_hosts_on(host, { 'foo' => '1.1.1.1' }, { 'foo' => %w[bar baz] })
      hosts_file = get_hosts_file(host)
      result = on host, "cat #{hosts_file}"
      assert_match(/foo/, result.stdout)
    end
  end

  step 'stubbed value should be available for other steps in the test' do
    hosts.each do |host|
      hosts_file = get_hosts_file(host)
      result = on host, "cat #{hosts_file}"
      assert_match(/foo/, result.stdout)
    end
  end
end

step 'verify with_stub_host_on' do
  step 'should add entry to hosts file' do
    hosts.each do |host|
      hosts_file = get_hosts_file(host)
      result = with_host_stubbed_on(host, { 'sleepy' => '1.1.1.2' }, { 'sleepy' => %w[grumpy dopey] }) do
        on host, "cat #{hosts_file}"
      end
      assert_match(/sleepy/, result.stdout)
    end
  end

  step 'stubbed value should be reverted after the execution of the block' do
    hosts.each do |host|
      hosts_file = get_hosts_file(host)
      result = on host, "cat #{hosts_file}"
      assert_no_match(/sleepy/, result.stdout)
    end
  end
end