#include <platform.h>
#include <kernel/types.h>

#include "ally.h"
#include "pool.h"
#include "magic.h"
#include "unit.h"
#include "item.h"
#include "faction.h"
#include "region.h"
#include "skill.h"

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

void test_pool(CuTest *tc) {
    unit *u1, *u2, *u3;
    faction *f;
    region *r;
    struct resource_type *rtype;

    test_cleanup();
    test_create_world();
    rtype = rt_get_or_create("money");
    it_get_or_create(rtype);
    f = test_create_faction(0);
    r = findregion(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(0), r);
    assert(u1 && u2);
    i_change(&u1->items, rtype->itype, 100);
    set_resvalue(u1, rtype, 50);
    i_change(&u2->items, rtype->itype, 200);
    set_resvalue(u2, rtype, 100);
    i_change(&u3->items, rtype->itype, 400);
    set_resvalue(u3, rtype, 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));

    u3->faction->allies = calloc(1, sizeof(ally));
    u3->faction->allies->faction = f;
    u3->faction->allies->status = HELP_GUARD;
    CuAssertIntEquals(tc, 0, get_pooled(u1, rtype, GET_ALLIED_SLACK | GET_ALLIED_RESERVE, INT_MAX));
    u3->faction->allies->status = 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));
}

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_cleanup();
    test_create_world();
    enable_skill(SK_MAGIC, true);

    r = findregion(0, 0);
    f = test_create_faction(0);
    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));
    }
}

CuSuite *get_pool_suite(void)
{
    CuSuite *suite = CuSuiteNew();
    SUITE_ADD_TEST(suite, test_pool);
    SUITE_ADD_TEST(suite, test_change_resource);
    return suite;
}