From 1d7c193de6f5b1e0fd4ed655ae01fc20da2aa9b6 Mon Sep 17 00:00:00 2001
From: Enno Rehling <enno.rehling@gmail.com>
Date: Fri, 31 Oct 2014 19:01:13 +0100
Subject: [PATCH] more test coverage for resource pooling.

---
 src/kernel/pool.test.c | 76 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)

diff --git a/src/kernel/pool.test.c b/src/kernel/pool.test.c
index 7a45475db..e6594bfac 100644
--- a/src/kernel/pool.test.c
+++ b/src/kernel/pool.test.c
@@ -15,6 +15,34 @@
 #include <assert.h>
 #include <limits.h>
 
+void test_reservation(CuTest *tc) {
+    unit *u;
+    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);
+    u = test_create_unit(f, r);
+    assert(u);
+    i_change(&u->items, rtype->itype, 100);
+    CuAssertIntEquals(tc, 100, get_resource(u, rtype));
+    CuAssertIntEquals(tc, 0, get_reservation(u, rtype));
+    CuAssertIntEquals(tc, 50, change_reservation(u, rtype, 50));
+    CuAssertIntEquals(tc, 100, change_reservation(u, rtype, 50));
+    CuAssertIntEquals(tc, 50, set_resvalue(u, rtype, 50));
+    CuAssertIntEquals(tc, 100, get_resource(u, rtype));
+    CuAssertIntEquals(tc, 200, change_resource(u, rtype, 100));
+    CuAssertIntEquals(tc, 200, get_resource(u, rtype));
+    CuAssertIntEquals(tc, 200, i_get(u->items, rtype->itype));
+
+    test_cleanup();
+}
+
 void test_pool(CuTest *tc) {
     unit *u1, *u2, *u3;
     faction *f;
@@ -60,6 +88,52 @@ void test_pool(CuTest *tc) {
     CuAssertIntEquals(tc, 300, get_pooled(u1, rtype, GET_ALL, INT_MAX));
 }
 
+void test_pool_use(CuTest *tc) {
+    unit *u1, *u2, *u3;
+    faction *f;
+    region *r;
+    struct resource_type *rtype;
+    ally *al;
+
+    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);
+    al = ally_add(&u3->faction->allies, f);
+    al->status = HELP_MONEY;
+
+    CuAssertIntEquals(tc, 10, use_pooled(u1, rtype, GET_SLACK, 10));
+    CuAssertIntEquals(tc, 40, use_pooled(u1, rtype, GET_SLACK, 50));
+    CuAssertIntEquals(tc, 50, i_get(u1->items, rtype->itype));
+
+    CuAssertIntEquals(tc, 50, get_reservation(u1, rtype));
+    CuAssertIntEquals(tc, 10, use_pooled(u1, rtype, GET_RESERVE, 10));
+    CuAssertIntEquals(tc, 40, i_get(u1->items, rtype->itype));
+    CuAssertIntEquals(tc, 40, get_reservation(u1, rtype));
+    CuAssertIntEquals(tc, 40, use_pooled(u1, rtype, GET_RESERVE, 50));
+
+    CuAssertIntEquals(tc, 10, use_pooled(u1, rtype, GET_POOLED_SLACK, 10));
+    CuAssertIntEquals(tc, 90, use_pooled(u1, rtype, GET_POOLED_SLACK, 100));
+    CuAssertIntEquals(tc, 100, i_get(u2->items, rtype->itype));
+    CuAssertIntEquals(tc, 10, use_pooled(u1, rtype, GET_POOLED_RESERVE, 10));
+    CuAssertIntEquals(tc, 90, get_reservation(u2, rtype));
+    CuAssertIntEquals(tc, 90, use_pooled(u1, rtype, GET_POOLED_RESERVE, 100));
+    CuAssertIntEquals(tc, 0, i_get(u2->items, rtype->itype));
+}
+
 void test_change_resource(CuTest * tc)
 {
     struct unit * u;
@@ -90,7 +164,9 @@ void test_change_resource(CuTest * tc)
 CuSuite *get_pool_suite(void)
 {
     CuSuite *suite = CuSuiteNew();
+    SUITE_ADD_TEST(suite, test_reservation);
     SUITE_ADD_TEST(suite, test_pool);
+    SUITE_ADD_TEST(suite, test_pool_use);
     SUITE_ADD_TEST(suite, test_change_resource);
     return suite;
 }