check_asset_load_write.cpp

Go to the documentation of this file.

#include <stdlib.h>
#include <string.h>
#include <check.h>
#include "global.h"
#include <toolkit_common.h>
#include <math.h>
#include "Archetypes.h"
#include "Treasures.h"
#include "TreasureLoader.h"
#include "TreasureWriter.h"
#include "AnimationWriter.h"
#include "FaceLoader.h"
#include "FaceWriter.h"
#include "Faces.h"
#include "Animations.h"
#include "assets.h"
#include "Quests.h"
#include "QuestLoader.h"
#include "QuestWriter.h"
 
static void setup(void) {
    init_hash_table();
    assets_init();
}
 
static void teardown(void) {
    assets_free();
}
 
char *generate_name(size_t max_length) {
    static char buf[250];
    size_t len = rndm(5, MIN(max_length, sizeof(buf) - 1));
    buf[len] = 0;
    for (size_t c = 0; c < len; c++) {
        buf[c] = 'a' + rndm(0, 25);
    }
    return buf;
}
 
template<class T>
static void write_load(T *asset, AssetWriter<T> &writer, AssetLoader &loader) {
    StringBuffer *sb = stringbuffer_new();
 
    writer.write(asset, sb);
 
    auto length = stringbuffer_length(sb);
    auto data = stringbuffer_finish(sb);
    BufferReader *br = bufferreader_init_from_memory(NULL, data, length);
 
    loader.load(br, "dummy");
 
    bufferreader_destroy(br);
    free(data);
}
 
treasure *generate_random_treasure(Archetypes &arches, int depth, bool single) {
    treasure *trs = static_cast<treasure *>(calloc(1, sizeof(treasure)));
    trs->chance = rndm(0, 15);
    if (rndm(0, 1)) {
        trs->name = add_string(generate_name(50));
        trs->list_magic_adjustment = rndm(-3, 8);
        trs->list_magic_value = rndm(0, 10);
    } else {
        trs->item = arches.get(generate_name(50));
        trs->artifact = add_string(generate_name(49));
    }
    trs->magic = rndm(0, 6);
    trs->nrof = rndm(0, 5);
    if (rndm(0, 5 - depth) != 0) {
        trs->next = generate_random_treasure(arches, depth + 1, single);
    }
    if (!single && rndm(0, 2) == 0) {
        trs->next_no = generate_random_treasure(arches, depth + 1, single);
    }
    if (!single && rndm(0, 2) == 0) {
        trs->next_yes = generate_random_treasure(arches, depth + 1, single);
    }
    return trs;
}
 
treasurelist *generate_random_list(Archetypes &arches) {
    treasurelist *list = static_cast<treasurelist *>(calloc(1, sizeof(treasurelist)));
    list->name = add_string(generate_name(50));
    bool single = rndm(0, 1) == 0;
    list->items = generate_random_treasure(arches, 0, single);
    if (single) {
        auto i = list->items;
        while (i) {
            list->total_chance += i->chance;
            i = i->next;
        }
    }
    return list;
}
 
bool equal(const treasure *left, treasure *right) {
    if (left == nullptr) {
        fail_unless(right == nullptr, "left treasure == null but right treasure != null");
        return true;
    }
    fail_unless(right != nullptr, "left treasure != null but right treasure == null");
    fail_unless(left->item == right->item, "different arch");
    fail_unless(left->artifact == right->artifact, "different artifacts");
    fail_unless(left->name == right->name, "different list");
    fail_unless(left->chance == right->chance, "different chance %d %d", left->chance, right->chance);
    fail_unless(left->magic == right->magic, "different magic %d %d", left->magic, right->magic);
    fail_unless(left->nrof == right->nrof, "different nrof %d %d", left->nrof, right->nrof);
    fail_unless(left->list_magic_value == right->list_magic_value, "different list_magic_value %d %d", left->list_magic_value, right->list_magic_value);
    fail_unless(left->list_magic_adjustment == right->list_magic_adjustment, "different list_magic_adjustment %d %d", left->list_magic_adjustment, right->list_magic_adjustment);
    return true;
}
 
bool equal(const treasurelist *left, const treasurelist *right) {
    if (left == nullptr) {
        fail_unless(right == nullptr, "left list == null but right list != null");
        return true;
    }
    fail_unless(right != nullptr, "left list != null but right list == null");
    fail_unless(strcmp(left->name, right->name) == 0, "left list name %s != right list name %s", left->name, right->name);
    fail_unless(equal(left->items, right->items), "left list items != right list items");
    fail_unless(left->total_chance == right->total_chance, "left list total_chance %d != right list total_chance %d", left->total_chance, right->total_chance);
    return true;
}
 
START_TEST(test_treasure) {
    Archetypes arch;
    Treasures trs;
    TreasureLoader loader(&trs, &arch, nullptr);
    TreasureWriter writer;
 
    auto generated = generate_random_list(arch);
 
    write_load(generated, writer, loader);
 
    auto loaded = trs.find(generated->name);
    fail_unless(loaded, "Treasurelist should have been loaded!");
    fail_unless(equal(generated, loaded), "loaded treasurelist isn't the same as generated one");
 
    trs.define(generated->name, generated);
    trs.clear();
    arch.clear();
}
END_TEST
 
static Animations *generate_animation(Faces &faces) {
    auto anim = static_cast<Animations *>(calloc(1, sizeof(Animations)));
    anim->name = generate_name(50);
    anim->facings = pow(2, rndm(0, 3));
    anim->num_animations = rndm(1, 16);
    anim->faces = static_cast<const Face **>(calloc(anim->num_animations, sizeof(const Face *)));
    for (uint8_t f = 0; f < anim->num_animations; f++) {
        anim->faces[f] = faces.get(generate_name(25));
    }
    return anim;
}
 
START_TEST(test_animation) {
    Faces faces;
    AllAnimations anims;
    Animations *anim = generate_animation(faces);
 
    AnimationWriter writer;
    FaceLoader loader(&faces, &anims, nullptr);
 
    write_load(anim, writer, loader);
 
    auto loaded = anims.find(anim->name);
    fail_unless(loaded, "face should have been loaded");
    ck_assert_str_eq(anim->name, loaded->name);
    ck_assert_int_eq(anim->num_animations, loaded->num_animations);
    for (uint8_t f = 0; f < anim->num_animations; f++) {
        ck_assert_str_eq(anim->faces[f]->name, loaded->faces[f]->name);
    }
    ck_assert_int_eq(anim->facings, loaded->facings);
 
    anims.define(anim->name, anim);
    faces.clear();
    anims.clear();
}
END_TEST
 
static Face *generate_face(Faces &faces) {
    Face *face = static_cast<Face *>(calloc(1, sizeof(Face)));
    face->name = generate_name(50);
    if (rndm(0, 10) == 0) {
        face->smoothface = faces.get(generate_name(50));
        return face;
    }
 
    face->magicmap = rndm(0, 10);
    if (rndm(0, 3) == 0) {
        face->magicmap |= FACE_FLOOR;
    }
    face->visibility = rndm(0, 50);
    if (rndm(0, 3) == 0) {
        face->smoothface = faces.get(generate_name(50));
    }
    return face;
}
 
START_TEST(test_face) {
    Faces faces;
    AllAnimations anims;
    FaceWriter writer;
    FaceLoader loader(&faces, &anims, nullptr);
 
    Face *face = generate_face(faces);
    write_load(face, writer, loader);
 
    auto loaded = faces.find(face->name);
    ck_assert_int_eq(face->magicmap, loaded->magicmap);
    ck_assert_int_eq(face->visibility, loaded->visibility);
    if (face->smoothface) {
        ck_assert(loaded->smoothface);
        ck_assert_str_eq(face->smoothface->name, loaded->smoothface->name);
    } else {
        ck_assert(loaded->smoothface == nullptr);
    }
 
    faces.define(face->name, face);
    faces.clear();
    anims.clear();
}
END_TEST
 
quest_condition *generate_condition() {
    auto c = static_cast<quest_condition *>(calloc(1, sizeof(quest_condition)));
    c->quest_code = generate_name(33);
    if (rndm(0, 3) == 0) {
        c->minstep = -1;
        c->maxstep = -1;
    } else if (rndm(0, 2) == 0) {
        c->minstep = rndm(4, 15);
        c->maxstep = c->minstep;
    } else if (rndm(0, 1)) {
        c->maxstep = rndm(4, 15);
    } else {
        c->minstep = rndm(0, 14);
        c->maxstep = c->minstep + rndm(1, 10);
    }
    return c;
}
 
quest_step_definition *generate_step() {
    auto s = static_cast<quest_step_definition *>(calloc(1, sizeof(quest_step_definition)));
    s->is_completion_step = rndm(0, 1);
    s->step = rndm(1, 50);
    s->step_description = rndm(0, 1) ? generate_name(40) : nullptr;
    for (int i = rndm(0, 2); i != 0; i--) {
        auto c = generate_condition();
        s->conditions.push_back(c);
    }
    return s;
}
 
quest_definition *generate_quest(Quests &quests, Faces &faces) {
    auto q = static_cast<quest_definition *>(calloc(1, sizeof(quest_definition)));
    q->quest_code = add_string(generate_name(30));
    q->quest_description = add_string(generate_name(100));
    q->quest_is_system = rndm(0, 1);
    q->quest_restart = rndm(0, 1);
    q->quest_title = add_string(generate_name(100));
    q->quest_comment = rndm(0, 1) ? add_string(generate_name(90)) : nullptr;
    if (rndm(0, 1)) {
        q->parent = quests.get(generate_name(25));
    }
    if (rndm(0, 1)) {
        q->face = faces.get(generate_name(25));
    } else {
        q->face = faces.get("quest_generic.111");
    }
    for (int i = rndm(0, 5); i != 0; i--) {
        q->steps.push_back(generate_step());
    }
    return q;
}
 
void check_conditions(const quest_condition *lc, const quest_condition *rc) {
    if (!lc) {
        ck_assert(rc == nullptr);
        return;
    }
    ck_assert(rc);
    ck_assert_str_eq(lc->quest_code, rc->quest_code);
    ck_assert_int_eq(lc->minstep, rc->minstep);
    ck_assert_int_eq(lc->maxstep, rc->maxstep);
}
 
void check_step(const quest_step_definition *ls, const quest_step_definition *rs) {
    if (!ls) {
        ck_assert(rs == nullptr);
        return;
    }
    ck_assert(rs);
    if (ls->step_description)
        ck_assert_str_eq(ls->step_description, rs->step_description);
    else
        ck_assert(rs->step_description == nullptr);
    ck_assert_int_eq(ls->is_completion_step, rs->is_completion_step);
    ck_assert_int_eq(ls->step, rs->step);
 
    ck_assert_int_eq(ls->conditions.size(), rs->conditions.size());
    auto li = ls->conditions.cbegin(), ri = rs->conditions.cbegin();
    while (li != ls->conditions.cend()) {
        check_conditions(*li, *ri);
        ++li;
        ++ri;
    }
}
 
void check_steps(const std::vector<quest_step_definition *> &ls, const std::vector<quest_step_definition *> &rs) {
    ck_assert(ls.size() == rs.size());
    for (size_t i = 0; i < ls.size(); i++) {
        check_step(ls[i], rs[i]);
    }
}
 
START_TEST(test_quest) {
    Faces faces;
    Quests quests;
    QuestWriter writer;
    QuestLoader loader(&quests, &faces, nullptr);
 
    auto quest = generate_quest(quests, faces);
    write_load(quest, writer, loader);
 
    auto loaded = quests.find(quest->quest_code);
    ck_assert_str_eq(quest->quest_description, loaded->quest_description);
    ck_assert_str_eq(quest->quest_title, loaded->quest_title);
    if (quest->quest_comment)
        ck_assert_str_eq(quest->quest_comment, loaded->quest_comment);
    else
        ck_assert(loaded->quest_comment == nullptr);
    ck_assert_int_eq(quest->quest_restart, loaded->quest_restart);
    ck_assert_int_eq(quest->quest_is_system, loaded->quest_is_system);
    ck_assert(quest->face == loaded->face);
    ck_assert(quest->parent == loaded->parent);
    check_steps(quest->steps, loaded->steps);
 
    quests.define(quest->quest_code, quest);
    faces.clear();
    quests.clear();
}
END_TEST
 
static Suite *shstr_suite(void) {
    Suite *s = suite_create("asset_load_write");
    TCase *tc_core = tcase_create("Core");
 
    /*setup and teardown will be called before each test in testcase 'tc_core' */
    tcase_add_checked_fixture(tc_core, setup, teardown);
 
    suite_add_tcase(s, tc_core);
    tcase_add_loop_test(tc_core, test_treasure, 0, 25);
    tcase_add_loop_test(tc_core, test_animation, 0, 25);
    tcase_add_loop_test(tc_core, test_face, 0, 25);
    tcase_add_loop_test(tc_core, test_quest, 0, 25);
 
    return s;
}
 
int main(void) {
    int nf;
    Suite *s = shstr_suite();
    SRunner *sr = srunner_create(s);
 
    srunner_set_xml(sr, LOGDIR "/unit/common/asset_load_write.xml");
    srunner_set_log(sr, LOGDIR "/unit/common/asset_load_write.out");
    srunner_set_fork_status(sr, CK_NOFORK);
    srunner_run_all(sr, CK_ENV); /*verbosity from env variable*/
    nf = srunner_ntests_failed(sr);
    srunner_free(sr);
    return (nf == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
}