forked from github/server
331 lines
11 KiB
C
331 lines
11 KiB
C
#ifdef _MSC_VER
|
|
#include <platform.h>
|
|
#endif
|
|
|
|
#include "automate.h"
|
|
|
|
#include "kernel/config.h"
|
|
#include "kernel/faction.h"
|
|
#include "kernel/order.h"
|
|
#include "kernel/region.h"
|
|
#include "kernel/unit.h"
|
|
|
|
#include "util/message.h"
|
|
|
|
#include "tests.h"
|
|
|
|
#include <CuTest.h>
|
|
|
|
static void test_autostudy_init(CuTest *tc) {
|
|
scholar scholars[4];
|
|
unit *u1, *u2, *u3, *u4, *u5, *ulist;
|
|
faction *f;
|
|
region *r;
|
|
message *msg;
|
|
skill_t skill = NOSKILL;
|
|
|
|
test_setup();
|
|
mt_create_error(77);
|
|
mt_create_error(771);
|
|
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
test_create_unit(f, r);
|
|
u2 = test_create_unit(f, r);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_level(u2, SK_ENTERTAINMENT, 2);
|
|
u4 = test_create_unit(f, r);
|
|
u4->thisorder = create_order(K_AUTOSTUDY, f->locale, "Dudelidu");
|
|
u3 = test_create_unit(f, r);
|
|
u3->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
u5 = test_create_unit(test_create_faction(NULL), r);
|
|
u5->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
scholars[2].u = NULL;
|
|
|
|
ulist = r->units;
|
|
CuAssertIntEquals(tc, 2, autostudy_init(scholars, 4, &ulist, &skill));
|
|
CuAssertIntEquals(tc, SK_ENTERTAINMENT, skill);
|
|
CuAssertPtrEquals(tc, u2, scholars[0].u);
|
|
CuAssertIntEquals(tc, 2, scholars[0].level);
|
|
CuAssertIntEquals(tc, 0, scholars[0].learn);
|
|
CuAssertPtrEquals(tc, u1, scholars[1].u);
|
|
CuAssertIntEquals(tc, 0, scholars[1].level);
|
|
CuAssertIntEquals(tc, 0, scholars[1].learn);
|
|
CuAssertPtrEquals(tc, NULL, scholars[2].u);
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
|
|
ulist = u3;
|
|
CuAssertIntEquals(tc, 1, autostudy_init(scholars, 4, &ulist, &skill));
|
|
CuAssertIntEquals(tc, SK_PERCEPTION, skill);
|
|
CuAssertPtrEquals(tc, u3, scholars[0].u);
|
|
CuAssertIntEquals(tc, 0, scholars[0].level);
|
|
CuAssertIntEquals(tc, 0, scholars[0].learn);
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
|
|
CuAssertPtrNotNull(tc, msg = test_find_messagetype(f->msgs, "error77"));
|
|
CuAssertPtrEquals(tc, NULL, test_find_messagetype_ex(f->msgs, "error77", msg));
|
|
test_teardown();
|
|
}
|
|
|
|
/**
|
|
* Reproduce Bug 2520
|
|
*/
|
|
static void test_autostudy_run_twoteachers(CuTest *tc) {
|
|
scholar scholars[4];
|
|
int nscholars;
|
|
unit *u1, *u2, *u3, *u4, *ulist;
|
|
faction *f;
|
|
region *r;
|
|
skill_t skill;
|
|
|
|
test_setup();
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
set_level(u1, SK_ENTERTAINMENT, 2);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
u2 = test_create_unit(f, r);
|
|
set_level(u2, SK_ENTERTAINMENT, 2);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
|
|
u3 = test_create_unit(f, r);
|
|
u3->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u3, 8);
|
|
u4 = test_create_unit(f, r);
|
|
u4->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u4, 12);
|
|
|
|
ulist = r->units;
|
|
CuAssertIntEquals(tc, 4, nscholars = autostudy_init(scholars, 4, &ulist, &skill));
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
autostudy_run(scholars, nscholars);
|
|
CuAssertIntEquals(tc, SK_ENTERTAINMENT, skill);
|
|
CuAssertIntEquals(tc, 0, scholars[0].learn);
|
|
CuAssertIntEquals(tc, 0, scholars[1].learn);
|
|
CuAssertIntEquals(tc, scholars[2].u->number * 2, scholars[2].learn);
|
|
CuAssertIntEquals(tc, scholars[3].u->number * 2, scholars[3].learn);
|
|
|
|
test_teardown();
|
|
}
|
|
|
|
static void test_autostudy_run(CuTest *tc) {
|
|
scholar scholars[4];
|
|
int nscholars;
|
|
unit *u1, *u2, *u3, *ulist;
|
|
faction *f;
|
|
region *r;
|
|
|
|
test_setup();
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u1, 2);
|
|
set_level(u1, SK_ENTERTAINMENT, 2);
|
|
u2 = test_create_unit(f, r);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u2, 10);
|
|
u3 = test_create_unit(f, r);
|
|
u3->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u3, 15);
|
|
|
|
scholars[2].u = NULL;
|
|
ulist = r->units;
|
|
CuAssertIntEquals(tc, 2, nscholars = autostudy_init(scholars, 4, &ulist, NULL));
|
|
CuAssertIntEquals(tc, UFL_MARK, u1->flags & UFL_MARK);
|
|
CuAssertIntEquals(tc, UFL_MARK, u2->flags & UFL_MARK);
|
|
CuAssertIntEquals(tc, 0, u3->flags & UFL_MARK);
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
autostudy_run(scholars, nscholars);
|
|
CuAssertIntEquals(tc, 1, scholars[0].learn);
|
|
CuAssertIntEquals(tc, 20, scholars[1].learn);
|
|
CuAssertPtrEquals(tc, NULL, scholars[2].u);
|
|
|
|
scholars[1].u = NULL;
|
|
ulist = u3;
|
|
CuAssertIntEquals(tc, 1, nscholars = autostudy_init(scholars, 4, &ulist, NULL));
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
autostudy_run(scholars, nscholars);
|
|
CuAssertIntEquals(tc, 15, scholars[0].learn);
|
|
CuAssertPtrEquals(tc, NULL, scholars[1].u);
|
|
|
|
test_teardown();
|
|
}
|
|
|
|
static void test_autostudy_run_noteachers(CuTest *tc) {
|
|
scholar scholars[4];
|
|
int nscholars;
|
|
unit *u1, *u2, *ulist;
|
|
faction *f;
|
|
region *r;
|
|
|
|
test_setup();
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u1, 5);
|
|
set_level(u1, SK_ENTERTAINMENT, 2);
|
|
|
|
u2 = test_create_unit(f, r);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u2, 7);
|
|
set_level(u2, SK_ENTERTAINMENT, 2);
|
|
|
|
scholars[2].u = NULL;
|
|
ulist = r->units;
|
|
CuAssertIntEquals(tc, 2, nscholars = autostudy_init(scholars, 4, &ulist, NULL));
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
autostudy_run(scholars, nscholars);
|
|
/* stupid qsort is unstable: */
|
|
CuAssertIntEquals(tc, 12, scholars[0].learn + scholars[1].learn);
|
|
CuAssertIntEquals(tc, 35, scholars[0].learn * scholars[1].learn);
|
|
CuAssertPtrEquals(tc, NULL, scholars[2].u);
|
|
test_teardown();
|
|
}
|
|
|
|
/**
|
|
* If a teacher unit doesn't have enough students, the remaining members study.
|
|
*/
|
|
static void test_autostudy_run_teachers_learn(CuTest *tc) {
|
|
scholar scholars[4];
|
|
int nscholars;
|
|
unit *u1, *u2, *ulist;
|
|
faction *f;
|
|
region *r;
|
|
|
|
test_setup();
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u1, 2);
|
|
set_level(u1, SK_ENTERTAINMENT, 2);
|
|
u2 = test_create_unit(f, r);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
set_number(u2, 10);
|
|
ulist = r->units;
|
|
CuAssertIntEquals(tc, 2, nscholars = autostudy_init(scholars, 4, &ulist, NULL));
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
autostudy_run(scholars, nscholars);
|
|
CuAssertIntEquals(tc, 1, scholars[0].learn);
|
|
CuAssertIntEquals(tc, 20, scholars[1].learn);
|
|
test_teardown();
|
|
}
|
|
|
|
/**
|
|
* Reproduce Bug 2514
|
|
*/
|
|
static void test_autostudy_run_skilldiff(CuTest *tc) {
|
|
scholar scholars[4];
|
|
int nscholars;
|
|
unit *u1, *u2, *u3, *ulist;
|
|
faction *f;
|
|
region *r;
|
|
|
|
test_setup();
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u1, 1);
|
|
set_level(u1, SK_PERCEPTION, 2);
|
|
u2 = test_create_unit(f, r);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u2, 10);
|
|
set_level(u2, SK_PERCEPTION, 1);
|
|
u3 = test_create_unit(f, r);
|
|
u3->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u3, 10);
|
|
scholars[3].u = NULL;
|
|
ulist = r->units;
|
|
CuAssertIntEquals(tc, 3, nscholars = autostudy_init(scholars, 4, &ulist, NULL));
|
|
CuAssertPtrEquals(tc, NULL, ulist);
|
|
autostudy_run(scholars, nscholars);
|
|
CuAssertIntEquals(tc, 0, scholars[0].learn);
|
|
CuAssertIntEquals(tc, 20, scholars[2].learn);
|
|
CuAssertIntEquals(tc, 10, scholars[1].learn);
|
|
test_teardown();
|
|
}
|
|
|
|
static void test_autostudy_batches(CuTest *tc) {
|
|
scholar scholars[2];
|
|
int nscholars;
|
|
unit *u1, *u2, *u3, *ulist;
|
|
faction *f;
|
|
region *r;
|
|
|
|
test_setup();
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u1, 1);
|
|
set_level(u1, SK_PERCEPTION, 2);
|
|
u2 = test_create_unit(f, r);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u2, 10);
|
|
u3 = test_create_unit(f, r);
|
|
u3->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u3, 10);
|
|
scholars[1].u = NULL;
|
|
ulist = r->units;
|
|
config_set("automate.batchsize", "2");
|
|
CuAssertIntEquals(tc, 2, nscholars = autostudy_init(scholars, 2, &ulist, NULL));
|
|
CuAssertPtrEquals(tc, u3, ulist);
|
|
autostudy_run(scholars, nscholars);
|
|
CuAssertIntEquals(tc, 0, scholars[0].learn);
|
|
CuAssertIntEquals(tc, 20, scholars[1].learn);
|
|
CuAssertIntEquals(tc, 1, nscholars = autostudy_init(scholars, 2, &ulist, NULL));
|
|
autostudy_run(scholars, nscholars);
|
|
CuAssertIntEquals(tc, 10, scholars[0].learn);
|
|
test_teardown();
|
|
}
|
|
|
|
static void test_do_autostudy(CuTest *tc) {
|
|
unit *u1, *u2, *u3, *u4;
|
|
faction *f;
|
|
region *r;
|
|
|
|
test_setup();
|
|
r = test_create_plain(0, 0);
|
|
f = test_create_faction(NULL);
|
|
u1 = test_create_unit(f, r);
|
|
u1->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u1, 1);
|
|
set_level(u1, SK_PERCEPTION, 2);
|
|
u2 = test_create_unit(f, r);
|
|
u2->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_PERCEPTION]);
|
|
set_number(u2, 10);
|
|
u3 = test_create_unit(f, r);
|
|
u3->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
u4 = test_create_unit(test_create_faction(NULL), r);
|
|
u4->thisorder = create_order(K_AUTOSTUDY, f->locale, skillnames[SK_ENTERTAINMENT]);
|
|
do_autostudy(r);
|
|
CuAssertIntEquals(tc, 2, get_level(u1, SK_PERCEPTION));
|
|
/* impossible to say if u2 is T1 or T2 now */
|
|
CuAssertIntEquals(tc, 1, get_level(u3, SK_ENTERTAINMENT));
|
|
CuAssertIntEquals(tc, 1, get_level(u4, SK_ENTERTAINMENT));
|
|
CuAssertIntEquals(tc, 0, u1->flags & UFL_MARK);
|
|
CuAssertIntEquals(tc, 0, u2->flags & UFL_MARK);
|
|
CuAssertIntEquals(tc, 0, u3->flags & UFL_MARK);
|
|
CuAssertIntEquals(tc, 0, u4->flags & UFL_MARK);
|
|
test_teardown();
|
|
}
|
|
|
|
CuSuite *get_automate_suite(void)
|
|
{
|
|
CuSuite *suite = CuSuiteNew();
|
|
SUITE_ADD_TEST(suite, test_autostudy_init);
|
|
SUITE_ADD_TEST(suite, test_autostudy_run);
|
|
SUITE_ADD_TEST(suite, test_do_autostudy);
|
|
SUITE_ADD_TEST(suite, test_autostudy_batches);
|
|
SUITE_ADD_TEST(suite, test_autostudy_run_noteachers);
|
|
SUITE_ADD_TEST(suite, test_autostudy_run_teachers_learn);
|
|
SUITE_ADD_TEST(suite, test_autostudy_run_twoteachers);
|
|
SUITE_ADD_TEST(suite, test_autostudy_run_skilldiff);
|
|
return suite;
|
|
}
|