require "lunit"

module("tests.e2.e2features", package.seeall, lunit.testcase )

function setup()
    eressea.free_game()
    eressea.settings.set("nmr.timeout", "0")
    eressea.settings.set("rules.food.flags", "4")
    eressea.settings.set("rules.ship.storms", "0")
    eressea.settings.set("rules.encounters", "0")
    eressea.settings.set("study.produceexp", "0")
    eressea.settings.set("rules.peasants.growth.factor", "0")
end

function test_calendar()
    assert_equal("winter", get_season(1011))
    assert_equal("spring", get_season(1012))
end

function test_herbalism()
-- OBS: herbalism is currently an E2-only skill
    local r = region.create(0, 0, "plain")
    local f = faction.create("human")
    local u = unit.create(f, r, 1)

    eressea.settings.set("rules.grow.formula", 0) -- plants do not grow
    u:add_item("money", u.number * 100)
    u:set_skill("herbalism", 5)
    r:set_resource("seed", 100)
    r:set_flag(1, false) -- regular trees
    u:clear_orders()
    u:add_order("MACHE Samen")
    process_orders()
    assert_equal(1, u:get_item("seed"))
    assert_equal(99, r:get_resource("seed"))
    r:set_flag(1, true) -- mallorn
    u:clear_orders()
    u:add_order("MACHE Mallornsamen")
    process_orders()
    assert_equal(1, u:get_item("mallornseed"))
    assert_equal(98, r:get_resource("seed"))
end

function test_dwarf_bonus()
    local r = region.create(0, 0, "mountain")
    r:set_resource("iron", 100)
    local level = r:get_resourcelevel("iron")
    assert_equal(1, level)
    local u = unit.create(faction.create("dwarf"), r)
    assert_equal("dwarf", u.faction.race)
    assert_equal("dwarf", u.race)
    u.faction.name = "Zwerge"
    u.number = 10
    u:set_skill("mining", 1)
    u:add_order("MACHE EISEN")
    process_orders()
    assert_equal(30, u:get_item("iron"))
    assert_equal(82, r:get_resource("iron"))
    u.building = building.create(r, "mine")
    u.building.size = 10
    u:add_item("money", 500) -- maintenance
    assert_equal(1, u:get_skill("mining"))
    process_orders()
    assert_equal(70, u:get_item("iron"))
    assert_equal(70, r:get_resource("iron"))
end

local function one_unit(r, f)
  local u = unit.create(f, r, 1)
  u:add_item("money", u.number * 100)
  u:clear_orders()
  return u
end

local function two_factions()
  local f1 = faction.create("human", "one@eressea.de", "de")
  local f2 = faction.create("human", "two@eressea.de", "de")
  return f1, f2
end

local function two_units(r, f1, f2)
  return one_unit(r, f1), one_unit(r, f2)
end

function test_learn()
    eressea.settings.set("study.random_progress", "0")
    local r = region.create(0, 0, "plain")
    local f = faction.create("human", "noreply@eressea.de", "de")
    f.age = 20
    local u = unit.create(f, r)
    u:clear_orders()
    u:add_order("@LERNEN Reiten")
    process_orders()
    assert_equal(1, u:get_skill("riding"))
    process_orders()
    process_orders()
    assert_equal(2, u:get_skill("riding"))
    process_orders()
    process_orders()
    process_orders()
    assert_equal(3, u:get_skill("riding"))
end

function test_teach()
    eressea.settings.set("study.random_progress", "0")
    local r = region.create(0, 0, "plain")
    local f = faction.create("human", "noreply@eressea.de", "de")
    f.age = 20
    local u = unit.create(f, r, 10)
    local u2 = unit.create(f, r)
    u:clear_orders()
    u:add_order("@LERNEN reiten")
    u2:clear_orders()
    u2:add_order("LEHREN " .. itoa36(u.id))
    u2:set_skill("riding", 4)
    process_orders()
    assert_equal(1, u:get_skill("riding"))
    process_orders()
    assert_equal(2, u:get_skill("riding"))
end

function test_rename()
  local r = region.create(0, 0, "plain")
  local f = faction.create("human", "noreply@eressea.de", "de")
  local u = unit.create(f, r)
  u:add_item("aoh", 1)
  assert_equal(u:get_item("ao_healing"), 1)
end

function test_unit_limit_is_1500()
  local r = region.create(0,0, "plain")
  local f = faction.create("human", "noreply@eressea.de", "de")
  for i = 1,1500 do
    unit.create(f, r, 1)
  end
  local u = unit.create(f, r, 0)
  u:add_item("money", 20000)
  u:clear_orders()
  u:add_order("REKRUTIEREN 1")
  process_orders()
  assert_equal(1, u.number)
end

function test_ship_capacity()
    local r = region.create(0,0, "ocean")
    region.create(1,0, "ocean")
    local r2 = region.create(2,0, "ocean")
    local f = faction.create("human", "capacity@eressea.de", "de")

    -- u1 is at the limit and moves
    local s1 = ship.create(r, "boat")
    local u1 = unit.create(f, r, 5)
    u1.ship = s1
    u1:set_skill("sailing", 10)
    u1:clear_orders()
    u1:add_order("NACH O O")

    -- u2 has too many people
    local s2 = ship.create(r, "boat")
    local u2 = unit.create(f, r, 6)
    u2.ship = s2
    u2:set_skill("sailing", 10)
    u2:clear_orders()
    u2:add_order("NACH O O")

    -- u4 has too much stuff
    local s4 = ship.create(r, "boat")
    local u4 = unit.create(f, r, 5)
    u4.ship = s4
    u4:set_skill("sailing", 10)
    u4:add_item("sword", 1)
    u4:clear_orders()
    u4:add_order("NACH O O")

    process_orders()

--    print(s.region, u.region, r2)
    assert_equal(r2, u1.region, "boat with 5 humans did not move")
    assert_not_equal(r2, u2.region, "boat with too many people has moved")
    assert_not_equal(r2, u4.region, "boat with too much cargo has moved")
end

function test_levitate()
  local r = region.create(0,0, "plain")
  local f = faction.create("human", "noreply@eressea.de", "de")
  local u = unit.create(f, r, 2)
  local s = ship.create(r, "boat")
  u.ship = s
  u.age = 20
  u:set_skill("sailing", 5)
  u:add_item("money", 100)
  u:clear_orders()
  u:add_order("ARBEITE")
  levitate_ship(u.ship, u, 2, 1)
  assert_equal(32, u.ship.flags)
  process_orders()
  assert_equal(0, u.ship.flags)
end

function test_terrains()
  local terrains = { "hell", "wall1", "corridor1" }
  for k,v in ipairs(terrains) do
    local r = region.create(k, k, v)
    assert_not_equal(nil, r)
  end
end

function test_races()
  local races = { "wolf", "orc", "human", "demon" }
  for k,v in ipairs(races) do
    local f = faction.create("human", "noreply@eressea.de", "de")
    assert_not_equal(nil, f)
  end
end

function test_can_give_person()
  local r = region.create(0, 0, "plain")
  local f1 = faction.create("human")
  local f2 = faction.create("human")
  local u1 = unit.create(f1, r, 10)
  local u2 = unit.create(f2, r, 10)
  u1.faction.age = 10
  u2.faction.age = 10
  u1:add_item("money", 500)
  u2:add_item("money", 500)
  u2:clear_orders()
  u2:add_order("GIB ".. itoa36(u1.id) .. " 1 PERSON")
  u2:add_order("HELFE ".. itoa36(f1.id) .. " GIB")
  u1:add_order("HELFE ".. itoa36(f2.id) .. " GIB")
  u1:add_order("KONTAKTIERE ".. itoa36(u2.id))
  process_orders()
  assert_equal(9, u2.number)
  assert_equal(11, u1.number)
end

function test_no_uruk()
  local f1 = faction.create("uruk")
  assert_equal(f1.race, "orc")
end

function test_block_movement()
  eressea.settings.set("rules.guard.base_stop_prob", "0.3")
  eressea.settings.set("rules.guard.amulet_stop_prob", "0.0")
  eressea.settings.set("rules.guard.skill_stop_prob", "0.1")

  local r0 = region.create(0, 0, "plain")
  local r1 = region.create(1, 0, "plain")
  local r2 = region.create(2, 0, "plain")
  local f1, f2 = two_factions()
  f1.age=20
  f2.age=20

  local u11 = one_unit(r1, f1)
  local u2 = { }
  for i = 1, 20 do
    u2[i] = one_unit(r0, f2)
  end
  
  u11:add_item("sword", 1)
  u11:add_item("money", 1)
  u11:set_skill("melee", 1)
  u11:set_skill("perception", 7)
  u11:clear_orders()
  u11:add_order("BEWACHEN")
  
  process_orders()
  
  for i, u in ipairs(u2) do
    u:add_item("horse", 1)
  	u:set_skill("riding", 1)
  	u:clear_orders()
  	u:add_order("NACH o o")
  end
  
  u2[1]:set_skill("stealth", 8)
  
  process_orders()

  assert_equal(r2, u2[1].region, "nobody should see me")
  for i, u in ipairs(u2) do
    if i > 1 then
      assert_equal(r1, u.region, "perception +7 should always stop me")
    end
  end
end

function test_block_movement_aots()
  eressea.settings.set("rules.guard.base_stop_prob", "0.0")
  eressea.settings.set("rules.guard.skill_stop_prob", "1.0")
  eressea.settings.set("rules.guard.amulet_stop_prob", "1.0")

  local r0 = region.create(0, 0, "plain")
  local r1 = region.create(1, 0, "plain")
  local r2 = region.create(2, 0, "plain")
  local f1, f2 = two_factions()
  f1.age=20
  f2.age=20

  local u11, u12 = two_units(r1, f1, f1)
  local u21, u22 = two_units(r0, f2, f2)
  
  for i, u in ipairs ({ u11, u12 }) do
    u:add_item("sword", 1)
  	u:add_item("money", 1)
  	u:set_skill("melee", 1)
  	u:clear_orders()
  	u:add_order("BEWACHEN")
  end
  
  process_orders()
  
  for i, u in ipairs ({ u21, u22 }) do
    u:add_item("horse", 1)
  	u:set_skill("riding", 1)
  	u:clear_orders()
  	u:add_order("NACH o o")
  end
  
  u12:add_item("aots", 1)
  u22:set_skill("stealth", 1)

  process_orders()

  assert_equal(r1, u21.region, "unit with amulet should stop me")
  assert_equal(r2, u22.region, "nobody should see me")
end

function test_stonegolems()
  local r0 = region.create(0, 0, "plain")
  local f1 = faction.create("stonegolem")
  local u1 = unit.create(f1, r0, 1)
  local u2 = unit.create(f1, r0, 2)
  local c1 = building.create(r0, "castle")

  c1.size = 226

  u1:set_skill("building", 1)
  u2:set_skill("building", 1)

-- test that no server crash occurs
  u1:clear_orders()
  u1:add_order("Mache Burg")
  process_orders()
  assert_equal(0, u1.number, "There should be no more stone golems")
-- end test server crash

-- test that Stone Golems build for four stones
  u2:clear_orders()
  u2:add_order("MACHE 4 BURG " .. itoa36(c1.id))
  process_orders()
  assert_equal(230, c1.size, "resulting size should be 230")
  assert_equal(1, u2.number, "There should be one stone golem")
-- end test Stone Golems four stones
end


function test_birthdaycake()
  r = region.create(0,0, "plain")
  f = faction.create("human")
  u = unit.create(f, r, 1)
  u:add_item("birthdaycake", 1)
  u:clear_orders()
  u:add_order("ZEIGE Geburtstagstorte")
  process_orders()
end

function test_demonstealth()
  local desc, r, f, u
  r = region.create(0, 0, "plain")
  f = faction.create("demon")
  u = unit.create(f, r, 1)

  u:clear_orders()
  u:add_order("TARNE Zwerg")
  process_orders()
  desc = u:show()
  assert_not_nil(string.find(desc, "Zwerg"))

  u:clear_orders()
  u:add_order("TARNE Drache")
  process_orders()
  desc = u:show()
  assert_equal(nil, string.find(desc, "Drache"))
end

function test_calendar_season_2328()
    assert_equal("fall", get_season(1026))
end

function test_give_to_other_okay()
    -- can give a person to another faction
    eressea.settings.set("GiveRestriction", "0")
    local r = region.create(0, 0, "plain")
    local f1 = faction.create("human")
    local f2 = faction.create("human")

    local u1 = unit.create(f1, r, 2, "human")
    local u2 = unit.create(f2, r, 1, "human")
    u2:add_order("KONTAKTIERE " .. itoa36(u1.id))
    u1:add_order("GIB " .. itoa36(u2.id) .. " 1 PERSON")
    process_orders()
    assert_equal(1, u1.number)
    assert_equal(2, u2.number)
end

local function get_name(x)
    return x.name .. " (" .. itoa36(x.id) .. ")"
end

function test_faction_stealth()
    local f = faction.create('human')
    local f2 = faction.create('human')
    local r = region.create(0, 0, 'plain')
    local u = unit.create(f, r)
    local u2 = unit.create(f2, r)
    assert_equal(get_name(u) .. ", 1 Mensch, aggressiv.", u:show(f))
    assert_equal(get_name(u) .. ", " .. get_name(f) .. ", 1 Mensch.", u:show(f2))
    u:add_order("TARNE PARTEI NUMMER " .. itoa36(f2.id))
    process_orders()
    assert_equal(get_name(u) .. ", " .. get_name(f2) .. ", 1 Mensch, aggressiv.", u:show(f))
    assert_equal(get_name(u) .. ", " .. get_name(f2) .. ", 1 Mensch.", u:show(f2))
    u:clear_orders()
    u:add_order("TARNE PARTEI NUMMER")
    process_orders()
    assert_equal(get_name(u) .. ", 1 Mensch, aggressiv.", u:show(f))
    assert_equal(get_name(u) .. ", " .. get_name(f) .. ", 1 Mensch.", u:show(f2))
end

function test_faction_anonymous()
    local f = faction.create('human')
    local f2 = faction.create('human')
    local r = region.create(0, 0, 'plain')
    local u = unit.create(f, r)
    local u2 = unit.create(f2, r)
    assert_equal(get_name(u) .. ", 1 Mensch, aggressiv.", u:show(f))
    assert_equal(get_name(u) .. ", " .. get_name(f) .. ", 1 Mensch.", u:show(f2))
    u:add_order("TARNE PARTEI")
    process_orders()
    assert_equal(get_name(u) .. ", anonym, 1 Mensch, aggressiv.", u:show(f))
    assert_equal(get_name(u) .. ", anonym, 1 Mensch.", u:show(f2))
    u:clear_orders()
    u:add_order("TARNE PARTEI NICHT")
    process_orders()
    assert_equal(get_name(u) .. ", 1 Mensch, aggressiv.", u:show(f))
    assert_equal(get_name(u) .. ", " .. get_name(f) .. ", 1 Mensch.", u:show(f2))
end

function test_new_orc_has_skills()
    local f = faction.create('orc')
    local r = region.create(0, 0, 'plain')
    local u = unit.create(f, r)
    u:add_item('money', 400)
    u.number = 0
    u:add_order("REKRUTIEREN 2")
    process_orders()
    assert_equal(2, u.number)
    assert_equal(1, u:get_skill('polearm'))
    assert_equal(1, u:get_skill('melee'))
end

function test_new_dracoid()
    local f = faction.create('human')
    local r = region.create(0, 0, 'plain')
    local u = unit.create(f, r, 'dracoid')
    u.number = 2
    u:equip("new_dracoid")
    assert_equal(2, u.number)
    item = nil
    if u:get_skill('bow') > 1 then
        item = 'bow'
    elseif u:get_skill('polearm') > 1 then
        item = 'spear'
    elseif u:get_skill('melee') > 1 then
        item = 'sword'
    end
    assert_not_nil(item)
    assert_equal(u.number, u:get_item(item))
end

function test_rising_undead()
    local f = faction.create('human')
    local r = region.create(0, 0, 'plain')
    local u = unit.create(f, r, 'undead')
    u.number = 2
    u:equip("rising_undead")
    assert_equal(2, u.number)
    assert_equal(u.number, u:get_item('rustysword'))
end