/*
 * This file is part of libdice.
 *
 * Copyright (C) 2018 Florian Stinglmayr <florian@n0la.org>
 *
 * libdice is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * libdice is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with libdice.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <dice.h>

#include <setjmp.h>
#include <cmocka.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

static void test_dice_simple_null(void **data)
{
    dice_t d = dice_simple(0, 0);
    assert_null(d);
}

static void test_dice_simple_one(void **data)
{
    dice_t d = dice_simple(1, 20);
    assert_non_null(d);
    dice_free(d);
}

static void test_dice_simple_roll(void **data)
{
    dice_t d = dice_simple(1, 20);
    int i = 0;

    assert_non_null(d);

    for (i = 0; i < 100000; i++) {
        int64_t res = dice_roll(d);
        assert_true(res >= 1 && res <= 20);
    }

    dice_free(d);
}

static void test_dice_simple_multi(void **data)
{
    dice_t d = dice_simple(3, 6);
    int i = 0;

    assert_non_null(d);

    for (i = 0; i < 100000; i++) {
        int64_t res = dice_roll(d);
        assert_true(res >= 3 && res <= 18);
    }

    dice_free(d);
}

static void test_dice_simple_get(void **data)
{
    dice_t d = dice_simple(8, 10);
    int i = 0;

    assert_non_null(d);

    assert_true(dice_get(d, DICEOPTION_AMOUNT, &i));
    assert_int_equal(i, 8);

    assert_true(dice_get(d, DICEOPTION_SIDES, &i));
    assert_int_equal(i, 10);

    dice_free(d);
}

static void test_dice_simple_set(void **data)
{
    dice_t d = dice_simple(1, 20);
    int i = 0;

    assert_non_null(d);

    assert_true(dice_set(d, DICEOPTION_AMOUNT, 20));
    assert_true(dice_set(d, DICEOPTION_SIDES, 4));

    assert_true(dice_get(d, DICEOPTION_AMOUNT, &i));
    assert_int_equal(i, 20);

    assert_true(dice_get(d, DICEOPTION_SIDES, &i));
    assert_int_equal(i, 4);

    dice_free(d);
}

static void test_dice_simple_invalid_getset(void **data)
{
    dice_t d = dice_simple(1, 20);
    int i = 0;

    assert_non_null(d);

    assert_false(dice_set(d, 9001, 20));
    assert_false(dice_get(d, 9001, &i));

    dice_free(d);
}

int main(int ac, char **av)
{
    const struct CMUnitTest tests[] = {
        cmocka_unit_test(test_dice_simple_null),
        cmocka_unit_test(test_dice_simple_one),
        cmocka_unit_test(test_dice_simple_roll),
        cmocka_unit_test(test_dice_simple_multi),
        cmocka_unit_test(test_dice_simple_set),
        cmocka_unit_test(test_dice_simple_get),
        cmocka_unit_test(test_dice_simple_invalid_getset),
    };

    return cmocka_run_group_tests(tests, NULL, NULL);
}