require 'al-test-utils'

class TestConnectionPerClass < Test::Unit::TestCase
  include AlTestUtils

  priority :must
  def test_multi_setup_connections_with_association
    make_ou("Sub,ou=Users")
    make_ou("Sub2,ou=Users")

    sub_user_class = Class.new(@user_class)
    sub_user_class.prefix = "ou=Sub"
    sub2_user_class = Class.new(@user_class)
    sub2_user_class.prefix = "ou=Sub2"
    sub_user_class.has_many(:related_entries, :wrap => "seeAlso")
    sub_user_class.set_associated_class(:related_entries, sub2_user_class)

    make_temporary_user(:uid => "uid=user1,ou=Sub") do |user,|
      make_temporary_user(:uid => "uid=user2,ou=Sub2") do |user2,|
        sub_user = sub_user_class.find(user.uid)
        sub2_user = sub2_user_class.find(user2.uid)
        sub_user.see_also = sub2_user.dn
        assert(sub_user.save)

        sub_user = sub_user_class.find(user.uid)
        assert_equal(["ou=Sub2"],
                     sub_user.related_entries.collect {|e| e.class.prefix})
      end
    end
  end

  priority :normal
  def test_multi_setup_connections
    make_ou("Sub")
    make_ou("Sub2")
    sub_class = ou_class("ou=Sub")
    sub2_class = ou_class("ou=Sub2")

    configuration = current_configuration.symbolize_keys
    configuration[:scope] = :base
    current_base = configuration[:base]
    sub_configuration = configuration.dup
    sub_base = "ou=Sub,#{current_base}"
    sub_configuration[:base] = sub_base
    sub2_configuration = configuration.dup
    sub2_base = "ou=Sub2,#{current_base}"
    sub2_configuration[:base] = sub2_base

    sub_class.setup_connection(sub_configuration)
    sub_class.prefix = nil
    sub2_class.setup_connection(sub2_configuration)
    sub2_class.prefix = nil

    assert_equal([sub_base], sub_class.find(:all).collect(&:dn))
    assert_equal([sub2_base], sub2_class.find(:all).collect(&:dn))
    assert_equal([sub_base], sub_class.find(:all).collect(&:dn))
    assert_equal([sub2_base], sub2_class.find(:all).collect(&:dn))
  end

  def test_multi_same_setup_connections
    make_ou("Sub")
    sub_class1 = ou_class("ou=Sub")
    sub_class2 = ou_class("ou=Sub")

    configuration = current_configuration.symbolize_keys
    configuration[:scope] = :base
    current_base = configuration[:base]
    sub_configuration = configuration.dup
    sub_base = "ou=Sub,#{current_base}"
    sub_configuration[:base] = sub_base

    sub_class1.setup_connection(sub_configuration)
    sub_class2.setup_connection(sub_configuration)

    sub_configuration1 = sub_class1.configuration
    sub_configuration2 = sub_class2.configuration

    assert_not_nil(sub_configuration1)
    assert_not_nil(sub_configuration2)

    sub_class1.setup_connection(sub_configuration.dup)
    sub_class1.prefix = nil

    assert_equal([
                   sub_configuration1,
                   sub_configuration2,
                 ],
                 [
                   sub_class1.configuration,
                   sub_class2.configuration,
                 ])
  end

  def test_bind
    non_anon_class = ou_class("ou=NonAnonymous")
    anon_class = ou_class("ou=Anonymous")

    assert(non_anon_class.connection.bound?)
    assert(anon_class.connection.bound?)

    anon_class.connection.unbind
    assert(!non_anon_class.connection.bound?)
    assert(!anon_class.connection.bound?)

    anon_class.connection.rebind
    assert(non_anon_class.connection.bound?)
    assert(anon_class.connection.bound?)

    assert_raises(ActiveLdap::AuthenticationError) do
      connect(non_anon_class,
              :bind_dn => nil,
              :allow_anonymous => false,
              :retry_limit => 0)
    end

    assert(!non_anon_class.connection.bound?)
    assert(anon_class.connection.bound?)

    anon_class.connection.unbind
    assert(!non_anon_class.connection.bound?)
    assert(!anon_class.connection.bound?)

    anon_class.connection.rebind
    assert(!non_anon_class.connection.bound?)
    assert(anon_class.connection.bound?)

    anon_class.connection.unbind
    assert(!non_anon_class.connection.bound?)
    assert(!anon_class.connection.bound?)

    assert_nothing_raised do
      connect(anon_class,
              :bind_dn => nil,
              :allow_anonymous => true)
    end

    assert(!non_anon_class.connection.bound?)
    assert(anon_class.connection.bound?)
  end

  private
  def connect(klass, config)
    klass.setup_connection({:adapter => adapter}.merge(config))
    klass.connection.connect
  end
end