require "lunit"

module("tests.e3.buildings", package.seeall, lunit.testcase )

function setup()
    eressea.game.reset()
    eressea.settings.set("rules.food.flags", "4")
end

function teardown()
    eressea.settings.set("rules.food.flags", "0")
end

function test_castle_names()
    local r = region.create(0, 0, "plain")
    local b = building.create(r, "castle")

    assert_equal("site", b:get_typename(1))
    assert_equal("site", b:get_typename(9))
    assert_equal("fortification", b:get_typename(10))
    assert_equal("fortification", b:get_typename(49))
    assert_equal("tower", b:get_typename(50))
    assert_equal("tower", b:get_typename(249))
    assert_equal("castle", b:get_typename(250))
    assert_equal("castle", b:get_typename(1249))
    assert_equal("fortress", b:get_typename(1250))
    assert_equal("fortress", b:get_typename(6249))
    assert_equal("citadel", b:get_typename(6250))
end

function test_build_watch()
    local r = region.create(0, 0, "plain")
    local f = faction.create("human", "e3build@eressea.de", "de")
    local u = unit.create(f, r, 1)

    u.number = 20
    u:add_item("log", 30)
    u.id = 42

    u:set_skill("building", 1)
    u:add_order("MACHE Wache")
    process_orders()
    assert_not_nil(u.building)
    -- stage two needs skill 2, this unit can only build a first stage:
    assert_equal(5, u.building.size)

    u:set_skill("building", 2)
    u:clear_orders()
    u:add_order("MACHE Wache " .. itoa36(u.building.id))
    process_orders()
    assert_not_nil(u.building)
    assert_equal(10, u.building.size)
end

function test_watch_names()
    local r = region.create(0, 0, "plain")
    local b = building.create(r, "watch")

    assert_equal("scaffolding", b:get_typename(1))
    assert_equal("scaffolding", b:get_typename(4))
    assert_equal("guardhouse", b:get_typename(5))
    assert_equal("guardhouse", b:get_typename(9))
    assert_equal("guardtower", b:get_typename(10))
end

function test_small_castles()
    local r = region.create(0, 0, "plain")
    local f1 = faction.create("human", "noreply@eressea.de", "de")
    local u1 = unit.create(f1, r, 1)
    local f2 = faction.create("halfling", "noreply@eressea.de", "de")
    local u2 = unit.create(f2, r, 1)

    local b = building.create(r, "castle")
    u2.building = b
    u1.building = b

    b.owner = u2
    assert_equal("site", b:get_typename(7))
    assert_equal("fortification", b:get_typename(8))
    b.owner = u1
    assert_equal("site", b:get_typename(9))
    assert_equal("fortification", b:get_typename(10))
end

function test_build_normal()
    local r = region.create(0, 0, "plain")
    local f = faction.create("human", "noreply@eressea.de", "de")
    local u = unit.create(f, r, 1)
    u:clear_orders()
    u:add_item("stone", 10)
    u:set_skill("building", 10)
    u:add_order("MACHE BURG")
    process_orders()
    assert_not_nil(u.building)
    assert_equal(10, u.building.size)
end

function test_build_packice()
    local r = region.create(0, 0, "packice")
    local f = faction.create("human", "packice@eressea.de", "de")
    local u = unit.create(f, r, 1)
    u:clear_orders()
    u:add_item("stone", 10)
    u:set_skill("building", 10)
    u:add_order("MACHE BURG")
    process_orders()
    assert_equal(nil, u.building)
end

function test_build_castle_stages()
    local r = region.create(0,0, "plain")
    local f = faction.create("human", "castle@eressea.de", "de")
    local u = unit.create(f, r, 1000)
    local b = building.create(r, "castle")

    u:add_item("stone", 1000)

    u:set_skill("building", 1)
    u:clear_orders()

    u:add_order("MACHE BURG " .. itoa36(b.id))
    process_orders()
    assert_equal(10, b.size)
    
    u:set_skill("building", 3)
    u:clear_orders()
    
    u:add_order("MACHE BURG " .. itoa36(b.id))
    process_orders()
    assert_equal(250, b.size)
end