#include <platform.h>
#include "market.h"
#include "tests.h"

#include <kernel/building.h>
#include <kernel/config.h>
#include <kernel/faction.h>
#include <kernel/item.h>
#include <kernel/race.h>
#include <kernel/region.h>
#include <kernel/terrain.h>
#include <kernel/unit.h>

#include <util/language.h>

#include <CuTest.h>
#include <tests.h>

#include <stdlib.h>

static void test_market_curse(CuTest * tc)
{
  region *r;
  building *b;
  unit *u;
  faction *f;
  int x, y;
  const terrain_type *terrain;
  item_type *htype, *ltype;
  luxury_type *lux;
  building_type *btype;

  free_gamedata();
  test_cleanup();
  test_create_world();

  htype = test_create_itemtype("herb");
  htype->flags |= ITF_HERB;
  htype->rtype->flags |= (RTF_ITEM | RTF_POOLED);

  ltype = test_create_itemtype("balm");
  ltype->rtype->flags |= (RTF_ITEM | RTF_POOLED);
  lux = new_luxurytype(ltype, 0);

  set_param(&global.parameters, "rules.region_owners", "1");

  btype = (building_type *)calloc(1, sizeof(building_type));
  btype->_name = "market";
  bt_register(btype);

  terrain = get_terrain("plain");

  for (x = 0; x != 3; ++x) {
    for (y = 0; y != 3; ++y) {
      r = findregion(x, y);
      if (!r) {
        r = test_create_region(x, y, terrain);
      } else {
        terraform_region(r, terrain);
      }
      rsetpeasants(r, 5000);
      r_setdemand(r, lux, 0);
      rsetherbtype(r, htype);
    }
  }
  r = findregion(1, 1);
  b = test_create_building(r, btype);
  b->flags |= BLD_WORKING;
  b->size = b->type->maxsize;

  f = test_create_faction(0);
  u = test_create_unit(f, r);
  u_set_building(u, b);

  do_markets();

  CuAssertIntEquals(tc, 70, i_get(u->items, htype));
  CuAssertIntEquals(tc, 35, i_get(u->items, ltype));
}

CuSuite *get_market_suite(void)
{
  CuSuite *suite = CuSuiteNew();
  SUITE_ADD_TEST(suite, test_market_curse);
  return suite;
}