server/src/gamecode/market_test.c

#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/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 char *names[4] = { "herb", "herbs", "balm", "balms" };
  const terrain_type *terrain;
  resource_type *hres, *lres;
  item_type *htype, *ltype;
  luxury_type *lux;
  building_type *btype;

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

  hres = new_resourcetype(names, 0, RTF_ITEM | RTF_POOLED);
  htype = new_itemtype(hres, ITF_HERB, 0, 0);
  lres = new_resourcetype(names + 2, 0, RTF_ITEM | RTF_POOLED);
  ltype = new_itemtype(lres, ITF_NONE, 0, 0);
  lux = new_luxurytype(ltype, 0);

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

  btype = calloc(sizeof(building_type), 1);
  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;
}