#include <platform.h>

#include "stealth.h"

#include <kernel/unit.h>
#include <kernel/region.h>

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

#include <stdlib.h>
#include <assert.h>

static void test_stealth(CuTest *tc) {
    unit *u;

    test_cleanup();
    u = test_create_unit(test_create_faction(test_create_race("human")), test_create_region(0, 0, 0));
    set_level(u, SK_STEALTH, 2);
    CuAssertIntEquals(tc, -1, u_geteffstealth(u));
    CuAssertIntEquals(tc, 2, eff_stealth(u, u->region));
    u_seteffstealth(u, 3);
    CuAssertIntEquals(tc, 3, u_geteffstealth(u));
    CuAssertIntEquals(tc, 2, eff_stealth(u, u->region));
    u_seteffstealth(u, 1);
    CuAssertIntEquals(tc, 1, u_geteffstealth(u));
    CuAssertIntEquals(tc, 1, eff_stealth(u, u->region));
    u_seteffstealth(u, -1);
    CuAssertIntEquals(tc, -1, u_geteffstealth(u));
    CuAssertIntEquals(tc, 2, eff_stealth(u, u->region));
    test_cleanup();
}

CuSuite *get_stealth_suite(void)
{
    CuSuite *suite = CuSuiteNew();
    SUITE_ADD_TEST(suite, test_stealth);
    return suite;
}