forked from github/server
221 lines
7.3 KiB
C
221 lines
7.3 KiB
C
#include <platform.h>
|
|
|
|
#include "ally.h"
|
|
#include "pool.h"
|
|
#include "magic.h"
|
|
#include "unit.h"
|
|
#include "item.h"
|
|
#include "faction.h"
|
|
#include "race.h"
|
|
#include "region.h"
|
|
#include "skill.h"
|
|
|
|
#include <CuTest.h>
|
|
#include <tests.h>
|
|
#include <assert.h>
|
|
#include <limits.h>
|
|
|
|
void test_reservation(CuTest *tc) {
|
|
unit *u;
|
|
faction *f;
|
|
region *r;
|
|
struct resource_type *rtype;
|
|
test_setup();
|
|
test_create_world();
|
|
rtype = rt_get_or_create("money");
|
|
it_get_or_create(rtype);
|
|
f = test_create_faction(NULL);
|
|
r = findregion(0, 0);
|
|
assert(r && f && rtype && rtype->itype);
|
|
u = test_create_unit(f, r);
|
|
assert(u);
|
|
i_change(&u->items, rtype->itype, 100);
|
|
CuAssertIntEquals(tc, 100, get_resource(u, rtype));
|
|
CuAssertIntEquals(tc, 0, get_reservation(u, rtype->itype));
|
|
CuAssertIntEquals(tc, 50, change_reservation(u, rtype->itype, 50));
|
|
CuAssertIntEquals(tc, 100, change_reservation(u, rtype->itype, 50));
|
|
CuAssertIntEquals(tc, 50, set_resvalue(u, rtype->itype, 50));
|
|
CuAssertIntEquals(tc, 100, get_resource(u, rtype));
|
|
CuAssertIntEquals(tc, 200, change_resource(u, rtype, 100));
|
|
CuAssertIntEquals(tc, 200, get_resource(u, rtype));
|
|
CuAssertIntEquals(tc, 200, i_get(u->items, rtype->itype));
|
|
|
|
test_teardown();
|
|
}
|
|
|
|
void test_pool_get_item(CuTest *tc) {
|
|
unit *u1, *u2;
|
|
faction *f;
|
|
region *r;
|
|
race *rc;
|
|
struct resource_type *rtype;
|
|
|
|
test_setup();
|
|
rtype = rt_get_or_create("money");
|
|
rtype->flags |= RTF_POOLED;
|
|
it_get_or_create(rtype);
|
|
f = test_create_faction(NULL);
|
|
r = test_create_plain(0, 0);
|
|
u1 = test_create_unit(f, r);
|
|
u_setrace(u1, rc = test_create_race("undead"));
|
|
rc->ec_flags &= ~ECF_GETITEM;
|
|
u2 = test_create_unit(f, r);
|
|
i_change(&u2->items, rtype->itype, 2);
|
|
CuAssertIntEquals(tc, 0, get_pooled(u1, rtype, GET_DEFAULT, 1));
|
|
CuAssertIntEquals(tc, 0, i_get(u1->items, rtype->itype));
|
|
CuAssertIntEquals(tc, 2, i_get(u2->items, rtype->itype));
|
|
test_teardown();
|
|
}
|
|
|
|
void test_pool(CuTest *tc) {
|
|
unit *u1, *u2, *u3;
|
|
faction *f;
|
|
region *r;
|
|
struct resource_type *rtype;
|
|
|
|
test_setup();
|
|
rtype = rt_get_or_create("money");
|
|
rtype->flags |= RTF_POOLED;
|
|
it_get_or_create(rtype);
|
|
f = test_create_faction(NULL);
|
|
r = test_create_plain(0, 0);
|
|
assert(r && f && rtype && rtype->itype);
|
|
u1 = test_create_unit(f, r);
|
|
u2 = test_create_unit(f, r);
|
|
u3 = test_create_unit(test_create_faction(NULL), r);
|
|
assert(u1 && u2);
|
|
i_change(&u1->items, rtype->itype, 100);
|
|
set_resvalue(u1, rtype->itype, 50);
|
|
i_change(&u2->items, rtype->itype, 200);
|
|
set_resvalue(u2, rtype->itype, 100);
|
|
i_change(&u3->items, rtype->itype, 400);
|
|
set_resvalue(u3, rtype->itype, 200);
|
|
|
|
CuAssertIntEquals(tc, 50, get_pooled(u1, rtype, GET_SLACK, 40));
|
|
CuAssertIntEquals(tc, 50, get_pooled(u1, rtype, GET_SLACK, INT_MAX));
|
|
CuAssertIntEquals(tc, 100, get_pooled(u1, rtype, GET_SLACK | GET_RESERVE, INT_MAX));
|
|
CuAssertIntEquals(tc, 150, get_pooled(u1, rtype, GET_SLACK | GET_POOLED_SLACK, INT_MAX));
|
|
CuAssertIntEquals(tc, 100, get_pooled(u1, rtype, GET_POOLED_SLACK, INT_MAX));
|
|
CuAssertIntEquals(tc, 200, get_pooled(u1, rtype, GET_POOLED_SLACK | GET_POOLED_RESERVE, INT_MAX));
|
|
|
|
ally_set(&u3->faction->allies, f, HELP_GUARD);
|
|
CuAssertIntEquals(tc, 0, get_pooled(u1, rtype, GET_ALLIED_SLACK | GET_ALLIED_RESERVE, INT_MAX));
|
|
ally_set(&u3->faction->allies, f, HELP_MONEY);
|
|
CuAssertIntEquals(tc, 200, get_pooled(u1, rtype, GET_ALLIED_SLACK, INT_MAX));
|
|
CuAssertIntEquals(tc, 200, get_pooled(u1, rtype, GET_ALLIED_RESERVE, INT_MAX));
|
|
CuAssertIntEquals(tc, 400, get_pooled(u1, rtype, GET_ALLIED_SLACK | GET_ALLIED_RESERVE, INT_MAX));
|
|
|
|
CuAssertIntEquals(tc, 100, get_pooled(u1, rtype, GET_ALL, 50));
|
|
CuAssertIntEquals(tc, 300, get_pooled(u1, rtype, GET_ALL, 150));
|
|
CuAssertIntEquals(tc, 300, get_pooled(u1, rtype, GET_ALL, INT_MAX));
|
|
|
|
test_teardown();
|
|
}
|
|
|
|
void test_pool_bug_2042(CuTest *tc) {
|
|
unit *u1, *u2;
|
|
faction *f;
|
|
region *r;
|
|
struct resource_type *rtype;
|
|
|
|
test_setup();
|
|
test_create_world();
|
|
rtype = rt_get_or_create("money");
|
|
it_get_or_create(rtype);
|
|
f = test_create_faction(NULL);
|
|
r = findregion(0, 0);
|
|
assert(r && f && rtype && rtype->itype);
|
|
u1 = test_create_unit(f, r);
|
|
u2 = test_create_unit(f, r);
|
|
assert(u1 && u2);
|
|
i_change(&u2->items, rtype->itype, 100);
|
|
|
|
CuAssertIntEquals(tc, 100, get_pooled(u1, rtype, GET_SLACK | GET_POOLED_SLACK, 100));
|
|
|
|
test_teardown();
|
|
}
|
|
|
|
void test_pool_use(CuTest *tc) {
|
|
unit *u1, *u2, *u3;
|
|
faction *f;
|
|
region *r;
|
|
struct item_type *itype;
|
|
|
|
test_setup();
|
|
test_create_world();
|
|
itype = it_get_or_create(rt_get_or_create("money"));
|
|
f = test_create_faction(NULL);
|
|
r = findregion(0, 0);
|
|
assert(r && f && itype);
|
|
u1 = test_create_unit(f, r);
|
|
u2 = test_create_unit(f, r);
|
|
u3 = test_create_unit(test_create_faction(NULL), r);
|
|
assert(u1 && u2);
|
|
i_change(&u1->items, itype, 100);
|
|
set_resvalue(u1, itype, 50);
|
|
i_change(&u2->items, itype, 200);
|
|
set_resvalue(u2, itype, 100);
|
|
i_change(&u3->items, itype, 400);
|
|
set_resvalue(u3, itype, 200);
|
|
ally_set(&u3->faction->allies, f, HELP_MONEY);
|
|
|
|
CuAssertIntEquals(tc, 10, use_pooled(u1, itype->rtype, GET_SLACK, 10));
|
|
CuAssertIntEquals(tc, 40, use_pooled(u1, itype->rtype, GET_SLACK, 50));
|
|
CuAssertIntEquals(tc, 50, i_get(u1->items, itype));
|
|
|
|
CuAssertIntEquals(tc, 50, get_reservation(u1, itype));
|
|
CuAssertIntEquals(tc, 10, use_pooled(u1, itype->rtype, GET_RESERVE, 10));
|
|
CuAssertIntEquals(tc, 40, i_get(u1->items, itype));
|
|
CuAssertIntEquals(tc, 40, get_reservation(u1, itype));
|
|
CuAssertIntEquals(tc, 40, use_pooled(u1, itype->rtype, GET_RESERVE, 50));
|
|
|
|
CuAssertIntEquals(tc, 10, use_pooled(u1, itype->rtype, GET_POOLED_SLACK, 10));
|
|
CuAssertIntEquals(tc, 90, use_pooled(u1, itype->rtype, GET_POOLED_SLACK, 100));
|
|
CuAssertIntEquals(tc, 100, i_get(u2->items, itype));
|
|
CuAssertIntEquals(tc, 10, use_pooled(u1, itype->rtype, GET_POOLED_RESERVE, 10));
|
|
CuAssertIntEquals(tc, 90, get_reservation(u2, itype));
|
|
CuAssertIntEquals(tc, 90, use_pooled(u1, itype->rtype, GET_POOLED_RESERVE, 100));
|
|
CuAssertIntEquals(tc, 0, i_get(u2->items, itype));
|
|
|
|
test_teardown();
|
|
}
|
|
|
|
void test_change_resource(CuTest * tc)
|
|
{
|
|
struct unit * u;
|
|
struct faction * f;
|
|
struct region * r;
|
|
const char * names[] = { "money", "aura", "permaura", "horse", "hp", 0 };
|
|
int i;
|
|
|
|
test_setup();
|
|
test_create_world();
|
|
enable_skill(SK_MAGIC, true);
|
|
|
|
r = findregion(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u = test_create_unit(f, r);
|
|
CuAssertPtrNotNull(tc, u);
|
|
set_level(u, SK_MAGIC, 5);
|
|
create_mage(u, M_DRAIG);
|
|
|
|
for (i = 0; names[i]; ++i) {
|
|
const struct resource_type *rtype = rt_find(names[i]);
|
|
int have = get_resource(u, rtype);
|
|
CuAssertIntEquals(tc, have + 1, change_resource(u, rtype, 1));
|
|
CuAssertIntEquals(tc, have + 1, get_resource(u, rtype));
|
|
}
|
|
test_teardown();
|
|
}
|
|
|
|
CuSuite *get_pool_suite(void)
|
|
{
|
|
CuSuite *suite = CuSuiteNew();
|
|
SUITE_ADD_TEST(suite, test_reservation);
|
|
SUITE_ADD_TEST(suite, test_pool);
|
|
SUITE_ADD_TEST(suite, test_pool_get_item);
|
|
SUITE_ADD_TEST(suite, test_pool_bug_2042);
|
|
SUITE_ADD_TEST(suite, test_pool_use);
|
|
SUITE_ADD_TEST(suite, test_change_resource);
|
|
return suite;
|
|
}
|