#include <platform.h>

#include "xerewards.h"
#include "study.h"
#include "magic.h"

#include <kernel/faction.h>
#include <kernel/item.h>
#include <kernel/pool.h>
#include <kernel/region.h>
#include <kernel/unit.h>

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

static void test_manacrystal(CuTest *tc) {
    struct item_type *itype;
    unit *u;

    test_setup();

    u = test_create_unit(test_create_faction(NULL), test_create_plain(0, 0));
    itype = test_create_itemtype("manacrystal");
    change_resource(u, itype->rtype, 1);
    CuAssertIntEquals(tc, -1, use_manacrystal(u, itype, 1, NULL));
    CuAssertPtrNotNull(tc, test_find_messagetype(u->faction->msgs, "error295"));
    test_clear_messages(u->faction);
    create_mage(u, M_GWYRRD);
    set_level(u, SK_MAGIC, 5);
    CuAssertIntEquals(tc, 0, get_spellpoints(u));
    CuAssertIntEquals(tc, 1, use_manacrystal(u, itype, 1, NULL));
    CuAssertIntEquals(tc, 25, get_spellpoints(u));
    CuAssertPtrNotNull(tc, test_find_messagetype(u->faction->msgs, "manacrystal_use"));
    test_clear_messages(u->faction);
    set_level(u, SK_MAGIC, 8);
    CuAssertIntEquals(tc, 1, use_manacrystal(u, itype, 1, NULL));
    CuAssertPtrNotNull(tc, test_find_messagetype(u->faction->msgs, "manacrystal_use"));
    CuAssertIntEquals(tc, 25 + 33, get_spellpoints(u));
    test_teardown();
}

static void test_skillpotion(CuTest *tc) {
    unit *u;
    const struct item_type *itype;
    skill* pSkill;
    int initialWeeks_Entertainment = 0;
    int initialWeeks_Stamina = 0;
    int initialWeeks_Magic = 0;

    test_setup();
    u = test_create_unit(test_create_faction(NULL), test_create_plain(0, 0));
    itype = test_create_itemtype("skillpotion");
    change_resource(u, itype->rtype, 2);

    learn_skill(u, SK_ENTERTAINMENT, STUDYDAYS * u->number);
    pSkill = unit_skill(u, SK_ENTERTAINMENT);
    sk_set(pSkill, 5);
    initialWeeks_Entertainment = pSkill->weeks = 4;

    learn_skill(u, SK_STAMINA, STUDYDAYS * u->number);
    pSkill = unit_skill(u, SK_STAMINA);
    sk_set(pSkill, 5);
    initialWeeks_Stamina = pSkill->weeks = 4;

    learn_skill(u, SK_MAGIC, STUDYDAYS * u->number);
    pSkill = unit_skill(u, SK_MAGIC);
    sk_set(pSkill, 5);
    initialWeeks_Magic = pSkill->weeks = 4;

    CuAssertIntEquals(tc, 1, use_skillpotion(u, itype, 1, NULL));

    pSkill = unit_skill(u, SK_ENTERTAINMENT);
    CuAssertIntEquals(tc, initialWeeks_Entertainment - 3, pSkill->weeks);

    pSkill = unit_skill(u, SK_STAMINA);
    CuAssertIntEquals(tc, initialWeeks_Stamina - 3, pSkill->weeks);

    pSkill = unit_skill(u, SK_MAGIC);
    CuAssertIntEquals(tc, initialWeeks_Magic - 3, pSkill->weeks);

    test_teardown();
}

CuSuite *get_xerewards_suite(void)
{
    CuSuite *suite = CuSuiteNew();
    SUITE_ADD_TEST(suite, test_manacrystal);
    SUITE_ADD_TEST(suite, test_skillpotion);
    return suite;
}