require "lunit"

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

function setup()
    free_game()
end

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

  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