require 'pg'
require 'stringio'
require "minitest/autorun"
require 'minitest/unit'

# WARNING: Test output is dependent on the order these classes include
# certain modules; i.e. on the 'require' order.
require 'common/configuration'
require 'common/domain'
require 'common/user'
require "rm/plugins/agendav"
require "rm/plugins/davical"
require "rm/plugins/postfixadmin"
require "rm/plugins/roundcube"
require "rm/rm_runner"
require "rm/rm_dummy_runner"

class TestRm < MiniTest::Unit::TestCase
  TESTCONF_PATH = 'test/mailshears.test.conf.yml'

  def connect_superuser()
    db_host = 'localhost'
    db_port = 5432
    db_opts = nil
    db_tty = nil
    db_name = 'postgres'
    db_user = 'postgres'
    db_pass = nil

    connection = PGconn.connect(db_host,
                                db_port,
                                db_opts,
                                db_tty,
                                db_name,
                                db_user,
                                db_pass)

    return connection
  end

  def setup
    # Create databases using from the test configuration file.
    cfg = Configuration.new(TESTCONF_PATH)
    connection = connect_superuser()

    cfg.plugins.each do |plugin|
      plugin_dbname = cfg.send("#{plugin}_dbname")
      query = "CREATE DATABASE #{plugin_dbname};"
      connection.query(query)

      plugin_dbhost = cfg.send("#{plugin}_dbhost")
      plugin_dbport = cfg.send("#{plugin}_dbport")
      plugin_dbopts = cfg.send("#{plugin}_dbopts")
      plugin_dbtty  = cfg.send("#{plugin}_dbtty")
      plugin_dbuser = cfg.send("#{plugin}_dbuser")
      plugin_dbpass = cfg.send("#{plugin}_dbpass")

      plugin_conn = PGconn.connect(plugin_dbhost,
                                   plugin_dbport,
                                   plugin_dbopts,
                                   plugin_dbtty,
                                   plugin_dbname,
                                   plugin_dbuser,
                                   plugin_dbpass)

      sql = File.open("test/sql/#{plugin}.sql").read()
      plugin_conn.query(sql)
      sql = File.open("test/sql/#{plugin}-fixtures.sql").read()
      plugin_conn.query(sql)
      plugin_conn.close()
    end

    connection.close()
  end



  def test_rm_user
    cfg = Configuration.new(TESTCONF_PATH)
    users = [ User.new("adam@example.net") ]

    output_buffer = StringIO.new()

    $stdout = output_buffer
    RmPlugin.run(cfg, *users)
    $stdout = STDOUT

    actual = output_buffer.string()

    expected =
      "AgendavRm - Removed user adam@example.net.\n" +
      "DavicalRm - User adam@example.net not found.\n" +
      "PostfixadminRm - Removed user adam@example.net.\n" +
      "RoundcubeRm - Removed user adam@example.net (User ID: 2).\n"

    assert_equal(expected, actual)

    # Now make sure the database has what we expect.

    arm = AgendavRm.new(cfg)
    actual = arm.list_users()
    expected = []
    assert_equal(expected, actual)

    drm = DavicalRm.new(cfg)
    actual = drm.list_users()
    expected = [User.new('alice@example.com')]
    assert_equal(expected, actual)

    pfarm = PostfixadminRm.new(cfg)
    actual = pfarm.list_users()
    expected = [User.new('alice@example.com'),
                User.new('bob@example.com'),
                User.new('beth@example.net'),
                User.new('carol@example.net')]
    assert_equal(expected, actual)

    actual = pfarm.list_domains()
    expected = [Domain.new('example.com'), Domain.new('example.net')]
    assert_equal(expected, actual)

    actual = pfarm.list_aliases()
    expected = [{'address' => 'alice@example.com',
                    'goto' => 'alice@example.com'},
                {'address' => 'bob@example.com',
                    'goto' => 'bob@example.com'},
                {'address' => 'beth@example.net',
                    'goto' => 'beth@example.net'},
                {'address' => 'carol@example.net',
                    'goto' => 'carol@example.net'}]
    assert_equal(expected, actual)

    rrm = RoundcubeRm.new(cfg)
    actual = rrm.list_users()
    expected = [User.new('alice@example.com')]
    assert_equal(expected, actual)
  end


  def test_rm_domain
    cfg = Configuration.new(TESTCONF_PATH)
    domains = [Domain.new("example.net")]

    output_buffer = StringIO.new()

    $stdout = output_buffer
    RmPlugin.run(cfg, *domains)
    $stdout = STDOUT

    actual = output_buffer.string()

    expected =
      "AgendavRm - Removed domain example.net.\n" +
      "DavicalRm - Domain example.net not found.\n" +
      "PostfixadminRm - Removed domain example.net.\n" +
      "RoundcubeRm - Removed domain example.net.\n"

    assert_equal(expected, actual)

    # Now make sure the database has what we expect.

    arm = AgendavRm.new(cfg)
    actual = arm.list_users()
    expected = []
    assert_equal(expected, actual)

    drm = DavicalRm.new(cfg)
    actual = drm.list_users()
    expected = [User.new('alice@example.com')]
    assert_equal(expected, actual)

    pfarm = PostfixadminRm.new(cfg)
    actual = pfarm.list_users()
    expected = [User.new('alice@example.com'),
                User.new('bob@example.com')]
    assert_equal(expected, actual)

    actual = pfarm.list_domains()
    expected = [Domain.new('example.com')]
    assert_equal(expected, actual)

    actual = pfarm.list_aliases()
    expected = [{'address' => 'alice@example.com',
                    'goto' => 'alice@example.com'},
                {'address' => 'bob@example.com',
                    'goto' => 'bob@example.com'}]
    assert_equal(expected, actual)

    rrm = RoundcubeRm.new(cfg)
    actual = rrm.list_users()
    expected = [User.new('alice@example.com')]
    assert_equal(expected, actual)
  end



  def teardown
    cfg = Configuration.new(TESTCONF_PATH)
    connection = connect_superuser()

    cfg.plugins.each do |plugin|
      plugin_dbname = cfg.send("#{plugin}_dbname")
      query = "DROP DATABASE #{plugin_dbname};"
      connection.query(query)
    end

    connection.close()
  end

end