#include "contact.h"

#include "kernel/faction.h"
#include "kernel/order.h"
#include "kernel/unit.h"

#include "util/language.h"
#include "util/param.h"

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

struct region;

static void test_contact(CuTest *tc) {
    struct unit *u1, *u2, *u3;
    struct region *r;
    struct faction *f;

    test_setup();
    r = test_create_plain(0, 0);
    f = test_create_faction(NULL);
    u1 = test_create_unit(f, r);
    u2 = test_create_unit(f, r);
    u3 = test_create_unit(test_create_faction(NULL), r);
    CuAssertTrue(tc, ucontact(u1, u1));
    CuAssertTrue(tc, ucontact(u1, u2));
    CuAssertTrue(tc, !ucontact(u1, u3));
    contact_unit(u1, u3);
    CuAssertTrue(tc, ucontact(u1, u3));
    CuAssertTrue(tc, !ucontact(u2, u3));
    test_teardown();
}

static void test_contact_cmd(CuTest *tc) {
    struct unit *u, *u2;
    struct region *r;
    const struct locale *lang;
    struct order *ord;

    test_setup();
    r = test_create_plain(0, 0);
    u = test_create_unit(test_create_faction(NULL), r);
    lang = u->faction->locale;

    u2 = test_create_unit(test_create_faction(NULL), r);
    ord = create_order(K_CONTACT, u->faction->locale, "%s %i",
        LOC(lang, parameters[P_UNIT]), u2->no);
    contact_cmd(u, ord);
    CuAssertTrue(tc, ucontact(u, u2));
    free_order(ord);

    u2 = test_create_unit(test_create_faction(NULL), r);
    ord = create_order(K_CONTACT, u->faction->locale, "%s %i",
        LOC(lang, parameters[P_FACTION]), u2->faction->no);
    contact_cmd(u, ord);
    CuAssertTrue(tc, ucontact(u, u2));
    free_order(ord);

    u2 = test_create_unit(test_create_faction(NULL), r);
    ord = create_order(K_CONTACT, u->faction->locale, "%i", u2->no);
    contact_cmd(u, ord);
    CuAssertTrue(tc, ucontact(u, u2));
    free_order(ord);

    u2 = test_create_unit(test_create_faction(NULL), r);
    usetalias(u2, 42);
    ord = create_order(K_CONTACT, u->faction->locale, "%s %i",
        LOC(lang, parameters[P_TEMP]), ualias(u2));
    contact_cmd(u, ord);
    CuAssertTrue(tc, ucontact(u, u2));
    free_order(ord);

    test_teardown();
}

CuSuite *get_contact_suite(void)
{
    CuSuite *suite = CuSuiteNew();
    SUITE_ADD_TEST(suite, test_contact);
    SUITE_ADD_TEST(suite, test_contact_cmd);

    return suite;
}