/*
 * Copyright 2022-2024 the Pacemaker project contributors
 *
 * The version control history for this file may have further details.
 *
 * This source code is licensed under the GNU General Public License version 2
 * or later (GPLv2+) WITHOUT ANY WARRANTY.
 */

#include <crm_internal.h>

#include <crm/common/unittest_internal.h>

#include <glib.h>

#include "crmcommon_private.h"  // pcmk__xe_set_score()

/*!
 * \internal
 * \brief Update an XML attribute value and check it against a reference value
 *
 * The attribute name is hard-coded as \c "X".
 *
 * \param[in] initial        Initial value
 * \param[in] new            Value to set
 * \param[in] reference_val  Expected attribute value after update
 * \param[in] reference_rc   Expected return code from \c pcmk__xe_set_score()
 */
static void
assert_set_score(const char *initial, const char *new,
                 const char *reference_val, int reference_rc)
{
    const char *name = "X";
    xmlNode *test_xml = pcmk__xe_create(NULL, "test_xml");

    crm_xml_add(test_xml, name, initial);
    assert_int_equal(pcmk__xe_set_score(test_xml, name, new), reference_rc);
    assert_string_equal(crm_element_value(test_xml, name), reference_val);

    free_xml(test_xml);
}

static void
value_is_name_plus_plus(void **state)
{
    assert_set_score("5", "X++", "6", pcmk_rc_ok);
}

static void
value_is_name_plus_equals_integer(void **state)
{
    assert_set_score("5", "X+=2", "7", pcmk_rc_ok);
}

// NULL input

static void
target_is_NULL(void **state)
{
    // Dumps core via CRM_CHECK()
    assert_int_equal(pcmk__xe_set_score(NULL, "X", "X++"), EINVAL);
}

static void
name_is_NULL(void **state)
{
    xmlNode *test_xml = pcmk__xe_create(NULL, "test_xml");

    crm_xml_add(test_xml, "X", "5");

    // Dumps core via CRM_CHECK()
    assert_int_equal(pcmk__xe_set_score(test_xml, NULL, "X++"), EINVAL);
    assert_string_equal(crm_element_value(test_xml, "X"), "5");

    free_xml(test_xml);
}

static void
value_is_NULL(void **state)
{
    assert_set_score("5", NULL, "5", pcmk_rc_ok);
}

// the value input doesn't start with the name input

static void
value_is_wrong_name(void **state)
{
    assert_set_score("5", "Y++", "Y++", pcmk_rc_ok);
}

static void
value_is_only_an_integer(void **state)
{
    assert_set_score("5", "2", "2", pcmk_rc_ok);
}

// non-integers

static void
variable_is_initialized_to_be_non_numeric(void **state)
{
    assert_set_score("hello", "X++", "1", pcmk_rc_ok);
}

static void
variable_is_initialized_to_be_non_numeric_2(void **state)
{
    assert_set_score("hello", "X+=2", "2", pcmk_rc_ok);
}

static void
variable_is_initialized_to_be_numeric_and_decimal_point_containing(void **state)
{
    assert_set_score("5.01", "X++", "6", pcmk_rc_ok);
}

static void
variable_is_initialized_to_be_numeric_and_decimal_point_containing_2(void **state)
{
    assert_set_score("5.50", "X++", "6", pcmk_rc_ok);
}

static void
variable_is_initialized_to_be_numeric_and_decimal_point_containing_3(void **state)
{
    assert_set_score("5.99", "X++", "6", pcmk_rc_ok);
}

static void
value_is_non_numeric(void **state)
{
    assert_set_score("5", "X+=hello", "5", pcmk_rc_ok);
}

static void
value_is_numeric_and_decimal_point_containing(void **state)
{
    assert_set_score("5", "X+=2.01", "7", pcmk_rc_ok);
}

static void
value_is_numeric_and_decimal_point_containing_2(void **state)
{
    assert_set_score("5", "X+=1.50", "6", pcmk_rc_ok);
}

static void
value_is_numeric_and_decimal_point_containing_3(void **state)
{
    assert_set_score("5", "X+=1.99", "6", pcmk_rc_ok);
}

// undefined input

static void
name_is_undefined(void **state)
{
    assert_set_score(NULL, "X++", "X++", pcmk_rc_ok);
}

// large input

static void
assignment_result_is_too_large(void **state)
{
    assert_set_score("5", "X+=100000000000", "1000000", pcmk_rc_ok);
}

PCMK__UNIT_TEST(pcmk__xml_test_setup_group, NULL,
                cmocka_unit_test(value_is_name_plus_plus),
                cmocka_unit_test(value_is_name_plus_equals_integer),
                cmocka_unit_test(target_is_NULL),
                cmocka_unit_test(name_is_NULL),
                cmocka_unit_test(value_is_NULL),
                cmocka_unit_test(value_is_wrong_name),
                cmocka_unit_test(value_is_only_an_integer),
                cmocka_unit_test(variable_is_initialized_to_be_non_numeric),
                cmocka_unit_test(variable_is_initialized_to_be_non_numeric_2),
                cmocka_unit_test(variable_is_initialized_to_be_numeric_and_decimal_point_containing),
                cmocka_unit_test(variable_is_initialized_to_be_numeric_and_decimal_point_containing_2),
                cmocka_unit_test(variable_is_initialized_to_be_numeric_and_decimal_point_containing_3),
                cmocka_unit_test(value_is_non_numeric),
                cmocka_unit_test(value_is_numeric_and_decimal_point_containing),
                cmocka_unit_test(value_is_numeric_and_decimal_point_containing_2),
                cmocka_unit_test(value_is_numeric_and_decimal_point_containing_3),
                cmocka_unit_test(name_is_undefined),
                cmocka_unit_test(assignment_result_is_too_large))
