require 'rubygems'
gem 'test-unit'

require 'test/unit'
require 'tempfile'
require 'test_helper'

class EnvironmentTest < Test::Unit::TestCase
  include Rubber::Configuration

  def test_known_roles
    env = Rubber::Configuration::Environment.new("#{File.dirname(__FILE__)}/fixtures/basic")
    assert_equal ['role1', 'role2'], env.known_roles, "list of know roles not correct"
  end

  def test_env
    env = Rubber::Configuration::Environment.new("#{File.dirname(__FILE__)}/fixtures/basic")
    e = env.bind()
    assert_equal 'val1', e['var1'], 'env not retrieving right val'
    assert_equal 'val2', e['var2'], 'env not retrieving right val'
    assert_equal 'val1', e.var1, 'env not retrieving right val for method missing'
    assert_equal 'val2', e.var2, 'env not retrieving right val for method missing'
    
    assert_equal 'val3', e.var3, 'env not retrieving right val for config in supplemental file'

    e = env.bind('role1', 'nohost')
    assert_equal 'val1', e['var1'], 'env not retrieving right val'
    assert_equal 'role1val2', e['var2'], 'env not retrieving right val'
    assert_equal 'val1', e.var1, 'env not retrieving right val for method missing'
    assert_equal 'role1val2', e.var2, 'env not retrieving right val for method missing'

    e = env.bind('role1', 'host1')
    assert_equal 'val1', e['var1'], 'env not retrieving right val'
    assert_equal 'host1val2', e['var2'], 'env not retrieving right val'
    assert_equal 'val1', e.var1, 'env not retrieving right val for method missing'
    assert_equal 'host1val2', e.var2, 'env not retrieving right val for method missing'

    e = env.bind('norole', 'host1')
    assert_equal 'val1', e['var1'], 'env not retrieving right val'
    assert_equal 'host1val2', e['var2'], 'env not retrieving right val'
    assert_equal 'val1', e.var1, 'env not retrieving right val for method missing'
    assert_equal 'host1val2', e.var2, 'env not retrieving right val for method missing'
  end

  def test_combine
    env = Rubber::Configuration::Environment
    assert_equal "new", env.combine("old", "new"), "Last should win for scalar combine"
    assert_equal 5, env.combine(1, 5), "Last should win for scalar combine"
    assert_equal [1, 2, 3, 4], env.combine([1, 2, 3], [3, 4]), "Arrays should be unioned when combined"
    assert_equal({1 => "1", 2 => "2", 3 => "3", 4 => "4"}, env.combine({1 => "1", 2 => "2"}, {3 => "3", 4 => "4"}), "Maps should be unioned when combined")
    assert_equal({1 => "2"}, env.combine({1 => "1"}, {1 => "2"}), "Last should win for scalars in maps when combined")
    assert_equal({1 => {1 => "1", 2 => "2"}}, env.combine({1 => {1 => "1"}}, {1 => {2 => "2"}}), "Maps should be unioned recursively when combined")
  end

  def test_expansion
    env = Rubber::Configuration::Environment.new("#{File.dirname(__FILE__)}/fixtures/expansion")
    e = env.bind()
    assert_equal 'val1', e['var1']
    assert_equal 'val2', e['var2']
    assert_equal 'val1', e['var3']
    assert_equal '4 is val2', e['var4']
    assert_equal 'val1', e['var5']
    assert_equal %w[lv1 lv2 val1], e['list1']
    expected = {'mk1' => 'mv1', 'mk2' => 'mv2', 'mk3' => 'val2'}
    e.map1.each do |k, v|
      assert_equal expected[k], v
    end
      
    e = env.bind('role1', 'nohost')
    assert_equal 'role1val1', e['var1']
    assert_equal 'role1val1', e['var3']
    assert_equal %w[lv1 lv2 role1val1 role1lv1 role1val2], e['list1']

    e = env.bind('role1', 'host1')
    assert_equal 'host1val1', e['var1']
    assert_equal 'host1val1', e['var3']
    assert_equal %w[lv1 lv2 host1val1 role1lv1 host1val2 host1lv1], e['list1'] # lists are additive
    
    e = env.bind('norole', 'host1')
    assert_equal 'host1val1', e['var1']
    assert_equal 'host1val1', e['var3']
    assert_equal %w[lv1 lv2 host1val1 host1lv1 host1val2], e['list1']
  end

  def test_bool_expansion
    env = Rubber::Configuration::Environment.new("#{File.dirname(__FILE__)}/fixtures/expansion")
    e = env.bind()
    assert_equal true, e['truevar']
    assert_equal false, e['falsevar']
    assert_equal true, e['truevar_exp']
    assert 'true' != e['truevar_exp']
    assert_equal false, e['falsevar_exp']
    assert 'false' != e['falsevar_exp']
    assert_equal 'true thing', e['faketruevar_exp']      
  end

  def test_secret_env
    env = Rubber::Configuration::Environment.new("#{File.dirname(__FILE__)}/fixtures/basic")
    e = env.bind()
    assert_nil e['rubber_secret'], 'env should not have secret set'
    
    fixture_dir = File.expand_path("#{File.dirname(__FILE__)}/fixtures/secret")
    env = Rubber::Configuration::Environment.new(fixture_dir)
    e = env.bind()
    assert_equal "#{fixture_dir}/secret.yml", e['rubber_secret'], 'env should have secret set'
    assert_equal "secret_val", e['secret_key'], 'env should have gotten setting from secret file'    
  end

  def test_nested_ref
    env = Rubber::Configuration::Environment.new("#{File.dirname(__FILE__)}/fixtures/nested")
    e = env.bind()
    assert_equal 'val1', e.var1, 'env not retrieving right val'
    assert_equal 'val3', e.var2.var3, 'env not retrieving right val'
    assert_equal({'var5' => 'val5'}, e.var2.var4, 'env not retrieving right val')
    assert_equal ['val6a', 'val6b'], e.var2.var6, 'env not retrieving right val'
    assert_equal 'val1', e.var2.var7, 'env not retrieving right val'
    assert_equal 'val3', e.var2.var8, 'env not retrieving right val'
    assert_equal 'val5', e.var2.var9, 'env not retrieving right val'
  end

  def test_instances_in_expansion
    instance = InstanceItem.new('host1', 'domain.com', [RoleItem.new('role1')], '')
    instance.external_ip = "1.2.3.4"
    instances = Instance.new(Tempfile.new('testforinstanceexpansion').path)
    instances.add(instance)

    File.expects(:exist?).returns(true)
    YAML.expects(:load_file).returns({'var1' =>'"#{rubber_instances.for_role("role1").first.external_ip}"'})
    Rubber::Configuration.expects(:rubber_instances).returns(instances)
    env = Rubber::Configuration::Environment.new(nil)
    e = env.bind()

    assert_equal "\"1.2.3.4\"", e['var1']    
  end

end