/**
 * @file
 * @brief Unit test for BACnetHostNPort encode and decode API
 * @author Steve Karg <skarg@users.sourceforge.net>
 * @date August 2023
 * @section LICENSE
 *
 * @copyright SPDX-License-Identifier: MIT
 */
#include <zephyr/ztest.h>
#include <bacnet/bacdcode.h>
#include <bacnet/bacdest.h>
#include <bacnet/hostnport.h>

#if defined(CONFIG_ZTEST_NEW_API)
ZTEST(host_n_port_tests, test_HostNPortMinimal_Init)
#else
static void test_HostNPortMinimal_Codec(void)
#endif
{
    uint8_t apdu[MAX_APDU] = { 0 };
    int len = 0, apdu_len = 0, null_len = 0, test_len = 0;
    BACNET_HOST_N_PORT_MINIMAL test_data_1 = { 0 }, test_data_2 = { 0 };
    BACNET_HOST_N_PORT host_n_port = { 0 };
    uint8_t address[IP6_ADDRESS_MAX] = { 1, 2,  3,  4,  5,  6,  7,  8,
                                         9, 10, 11, 12, 13, 14, 15, 16 };
    const char *hostname = "bacnet.org";
    uint16_t port = 0xBAC0;
    BACNET_ERROR_CODE error_code = ERROR_CODE_SUCCESS;

    /* test TAG for BACnet/IP address */
    host_n_port_minimal_ip_init(&test_data_1, port, address, sizeof(address));
    zassert_equal(test_data_1.tag, BACNET_HOST_ADDRESS_TAG_IP_ADDRESS, NULL);
    zassert_equal(test_data_1.port, port, NULL);
    zassert_equal(test_data_1.host.ip_address.length, sizeof(address), NULL);
    zassert_mem_equal(
        test_data_1.host.ip_address.address, address, sizeof(address), NULL);
    host_n_port_from_minimal(&host_n_port, &test_data_1);
    host_n_port_to_minimal(&test_data_2, &host_n_port);
    zassert_true(
        host_n_port_minimal_same(&test_data_1, &test_data_2),
        "test_data_1 != test_data_2");

    null_len = host_n_port_minimal_encode(NULL, &test_data_1);
    apdu_len = host_n_port_minimal_encode(apdu, &test_data_1);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len = host_n_port_minimal_decode(apdu, apdu_len, NULL, NULL);
    test_len =
        host_n_port_minimal_decode(apdu, apdu_len, &error_code, &test_data_2);
    zassert_equal(test_len, null_len, NULL);
    zassert_equal(
        apdu_len, test_len, "apdu_len=%d test_len=%d", apdu_len, test_len);
    zassert_true(
        host_n_port_minimal_same(&test_data_1, &test_data_2),
        "test_data_1 != test_data_2");
    while (test_len) {
        test_len--;
        len = host_n_port_minimal_decode(apdu, test_len, NULL, NULL);
        zassert_true(len < 0, "len=%d test_len=%d", len, test_len);
    }

    /* test TAG for BACnet/IP hostname */
    host_n_port_minimal_hostname_init(&test_data_1, port, hostname);
    zassert_equal(test_data_1.tag, BACNET_HOST_ADDRESS_TAG_NAME, NULL);
    null_len = host_n_port_minimal_encode(NULL, &test_data_1);
    apdu_len = host_n_port_minimal_encode(apdu, &test_data_1);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len = host_n_port_minimal_decode(apdu, apdu_len, NULL, NULL);
    test_len =
        host_n_port_minimal_decode(apdu, apdu_len, &error_code, &test_data_2);
    zassert_equal(test_len, null_len, NULL);
    zassert_equal(
        apdu_len, test_len, "apdu_len=%d test_len=%d", apdu_len, test_len);
    zassert_true(
        host_n_port_minimal_same(&test_data_1, &test_data_2),
        "test_data_1 != test_data_2");
    while (test_len) {
        test_len--;
        len = host_n_port_minimal_decode(apdu, test_len, NULL, NULL);
        zassert_true(len < 0, "len=%d test_len=%d", len, test_len);
    }
}

static void test_HostNPortMinimal_Copy(BACNET_HOST_N_PORT *data)
{
    BACNET_HOST_N_PORT_MINIMAL test_data_1 = { 0 }, test_data_2 = { 0 };

    host_n_port_to_minimal(&test_data_1, data);
    host_n_port_minimal_copy(&test_data_2, &test_data_1);
    zassert_true(
        host_n_port_minimal_same(&test_data_1, &test_data_2),
        "test_data_1 != test_data_2");
}

static void test_HostNPortCodec(BACNET_HOST_N_PORT *data)
{
    uint8_t apdu[MAX_APDU] = { 0 };
    BACNET_HOST_N_PORT test_data = { 0 };
    BACNET_ERROR_CODE error_code = ERROR_CODE_SUCCESS;
    int len = 0, apdu_len = 0, null_len = 0, test_len = 0;
    uint8_t tag_number = 0;
    bool status = false;

    null_len = host_n_port_encode(NULL, data);
    apdu_len = host_n_port_encode(apdu, data);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len = host_n_port_decode(apdu, apdu_len, NULL, NULL);
    test_len = host_n_port_decode(apdu, apdu_len, &error_code, &test_data);
    zassert_equal(test_len, null_len, NULL);
    zassert_equal(
        apdu_len, test_len, "apdu_len=%d test_len=%d", apdu_len, test_len);
    while (test_len) {
        test_len--;
        len = host_n_port_decode(apdu, test_len, NULL, NULL);
        zassert_true(len < 0, "len=%d test_len=%d", len, test_len);
    }

    null_len = host_n_port_context_encode(NULL, tag_number, data);
    apdu_len = host_n_port_context_encode(apdu, tag_number, data);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);

    null_len = host_n_port_context_decode(
        apdu, apdu_len, tag_number, &error_code, NULL);
    test_len = host_n_port_context_decode(
        apdu, apdu_len, tag_number, &error_code, &test_data);
    zassert_equal(test_len, null_len, NULL);
    zassert_true(test_len > 0, "test_len=%d", len);
    status = host_n_port_same(&test_data, data);
    zassert_true(status, NULL);

    status = host_n_port_copy(&test_data, data);
    zassert_true(status, NULL);
    status = host_n_port_same(&test_data, data);
    zassert_true(status, NULL);
}

#if defined(CONFIG_ZTEST_NEW_API)
ZTEST(host_n_port_tests, test_HostNPort)
#else
static void test_HostNPort(void)
#endif
{
    BACNET_HOST_N_PORT data = { 0 };
    bool status = false;
    const char *dotted_ip = "192.168.1.1";
    const char *dotted_ip_port = "192.168.1.1:47808";

    /* none */
    test_HostNPortCodec(&data);
    /* IP Address */
    octetstring_init_ascii_hex(&data.host.ip_address, "c0a80101");
    data.host_ip_address = true;
    data.host_name = false;
    data.port = 0xBAC0;
    test_HostNPortCodec(&data);
    test_HostNPortMinimal_Copy(&data);
    /* Host Name */
    characterstring_init_ansi(&data.host.name, "bacnet.org");
    data.host_ip_address = false;
    data.host_name = true;
    data.port = 0xBAC0;
    test_HostNPortCodec(&data);
    test_HostNPortMinimal_Copy(&data);

    status = host_n_port_from_ascii(NULL, NULL);
    zassert_false(status, NULL);
    status = host_n_port_from_ascii(NULL, dotted_ip);
    zassert_false(status, NULL);
    status = host_n_port_from_ascii(&data, NULL);
    zassert_false(status, NULL);
    status = host_n_port_from_ascii(&data, dotted_ip);
    zassert_true(status, NULL);
    status = host_n_port_from_ascii(&data, dotted_ip_port);
    zassert_true(status, NULL);
    test_HostNPortCodec(&data);
    test_HostNPortMinimal_Copy(&data);
}

#if defined(CONFIG_ZTEST_NEW_API)
ZTEST(host_n_port_tests, test_is_valid_hostname)
#else
static void test_is_valid_hostname(void)
#endif
{
    bool status = false;
    BACNET_CHARACTER_STRING hostname;

    status = characterstring_init_ansi(&hostname, "valid.host.name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_true(status, NULL);

    status = characterstring_init_ansi(&hostname, "invalid..host.name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_false(status, NULL);

    status = characterstring_init_ansi(&hostname, "-invalid.host.name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_false(status, NULL);

    status = characterstring_init_ansi(&hostname, "valid.host-name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_true(status, NULL);

    status = characterstring_init_ansi(&hostname, "invalid.host--name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_false(status, NULL);

    status = characterstring_init_ansi(&hostname, "invalid.host name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_false(status, NULL);

    status = characterstring_init_ansi(&hostname, "valid.host.123.name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_true(status, NULL);

    {
        char too_long_label[65] = { 0 };
        memset(too_long_label, 'l', sizeof(too_long_label) - 1);
        status = characterstring_init_ansi(&hostname, too_long_label);
        zassert_true(status, NULL);
        status = bacnet_is_valid_hostname(&hostname);
        zassert_false(status, NULL);
    }

    {
        char max_length_label[64] = { 0 };
        memset(max_length_label, 'l', sizeof(max_length_label) - 1);
        status = characterstring_init_ansi(&hostname, max_length_label);
        zassert_true(status, NULL);
        status = bacnet_is_valid_hostname(&hostname);
        zassert_true(status, NULL);
    }

    status = characterstring_init_ansi(&hostname, "non-alpanumer!c.host name");
    zassert_true(status, NULL);
    status = bacnet_is_valid_hostname(&hostname);
    zassert_false(status, NULL);
}

#if defined(CONFIG_ZTEST_NEW_API)
ZTEST(host_n_port_tests, test_fdt_entry)
#else
static void test_fdt_entry(void)
#endif
{
    bool status = false;
    const char *fdt_string = "1.2.3.4:47808,60,30";
    const char *ipv6_fdt_string =
        "fe80:0000:0000:0000:020c:29ff:fe50:745b:47808,60,30";
    const char *bdt_string = "1.2.3.4:47808";
    const char *ipv6_bdt_string =
        "fe48:0000:000d:0000:0002:0c29:fffe:5074:47808";
    char str[128] = { 0 };

    BACNET_FDT_ENTRY fdt_entry = { 0 }, test_fdt_entry = { 0 };
    BACNET_BDT_ENTRY bdt_entry = { 0 }, test_bdt_entry = { 0 };
    uint8_t apdu[MAX_APDU] = { 0 };
    int apdu_len = 0, null_len = 0, test_len = 0;
    BACNET_ERROR_CODE error_code = ERROR_CODE_SUCCESS;

    status = bacnet_fdt_entry_from_ascii(NULL, NULL);
    zassert_false(status, NULL);
    status = bacnet_fdt_entry_from_ascii(NULL, fdt_string);
    zassert_false(status, NULL);
    status = bacnet_fdt_entry_from_ascii(&fdt_entry, NULL);
    zassert_false(status, NULL);
    status = bacnet_fdt_entry_from_ascii(&fdt_entry, fdt_string);
    zassert_true(status, NULL);
    status = bacnet_fdt_entry_copy(&test_fdt_entry, &fdt_entry);
    zassert_true(status, NULL);
    status = bacnet_fdt_entry_same(&test_fdt_entry, &fdt_entry);
    zassert_true(status, NULL);

    null_len = bacnet_fdt_entry_context_encode(NULL, 0, &fdt_entry);
    apdu_len = bacnet_fdt_entry_context_encode(apdu, 0, &fdt_entry);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len =
        bacnet_fdt_entry_context_decode(apdu, apdu_len, 0, &error_code, NULL);
    test_len = bacnet_fdt_entry_context_decode(
        apdu, apdu_len, 0, &error_code, &test_fdt_entry);
    zassert_equal(test_len, null_len, NULL);
    test_len = bacnet_fdt_entry_to_ascii(str, sizeof(str), &test_fdt_entry);
    zassert_true(test_len > 0, "%s", str);

    status = bacnet_fdt_entry_from_ascii(&fdt_entry, ipv6_fdt_string);
    zassert_true(status, NULL);
    null_len = bacnet_fdt_entry_context_encode(NULL, 0, &fdt_entry);
    apdu_len = bacnet_fdt_entry_context_encode(apdu, 0, &fdt_entry);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len =
        bacnet_fdt_entry_context_decode(apdu, apdu_len, 0, &error_code, NULL);
    test_len = bacnet_fdt_entry_context_decode(
        apdu, apdu_len, 0, &error_code, &test_fdt_entry);
    zassert_equal(test_len, null_len, NULL);
    test_len = bacnet_fdt_entry_to_ascii(str, sizeof(str), &test_fdt_entry);
    zassert_true(test_len > 0, "%s", str);

    status = bacnet_bdt_entry_from_ascii(NULL, NULL);
    zassert_false(status, NULL);
    status = bacnet_bdt_entry_from_ascii(NULL, bdt_string);
    zassert_false(status, NULL);
    status = bacnet_bdt_entry_from_ascii(&bdt_entry, NULL);
    zassert_false(status, NULL);
    status = bacnet_bdt_entry_from_ascii(&bdt_entry, bdt_string);
    zassert_true(status, NULL);
    status = bacnet_bdt_entry_copy(&test_bdt_entry, &bdt_entry);
    zassert_true(status, NULL);
    status = bacnet_bdt_entry_same(&test_bdt_entry, &bdt_entry);
    zassert_true(status, NULL);

    null_len = bacnet_bdt_entry_encode(NULL, &bdt_entry);
    apdu_len = bacnet_bdt_entry_encode(apdu, &bdt_entry);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len = bacnet_bdt_entry_decode(apdu, apdu_len, &error_code, NULL);
    test_len =
        bacnet_bdt_entry_decode(apdu, apdu_len, &error_code, &test_bdt_entry);
    zassert_equal(test_len, null_len, NULL);
    test_len = bacnet_bdt_entry_to_ascii(str, sizeof(str), &test_bdt_entry);
    zassert_true(test_len > 0, "%s", str);

    status = bacnet_bdt_entry_from_ascii(&bdt_entry, ipv6_bdt_string);
    zassert_true(status, NULL);
    null_len = bacnet_bdt_entry_encode(NULL, &bdt_entry);
    apdu_len = bacnet_bdt_entry_encode(apdu, &bdt_entry);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len = bacnet_bdt_entry_decode(apdu, apdu_len, &error_code, NULL);
    test_len =
        bacnet_bdt_entry_decode(apdu, apdu_len, &error_code, &test_bdt_entry);
    zassert_equal(test_len, null_len, NULL);
    test_len = bacnet_bdt_entry_to_ascii(str, sizeof(str), &test_bdt_entry);
    zassert_true(test_len > 0, "%s", str);

    null_len = bacnet_bdt_entry_context_encode(NULL, 0, &bdt_entry);
    apdu_len = bacnet_bdt_entry_context_encode(apdu, 0, &bdt_entry);
    zassert_equal(apdu_len, null_len, NULL);
    zassert_true(apdu_len != BACNET_STATUS_ERROR, NULL);
    null_len =
        bacnet_bdt_entry_context_decode(apdu, apdu_len, 0, &error_code, NULL);
    test_len = bacnet_bdt_entry_context_decode(
        apdu, apdu_len, 0, &error_code, &test_bdt_entry);
    zassert_equal(test_len, null_len, NULL);
}

#if defined(CONFIG_ZTEST_NEW_API)
ZTEST_SUITE(host_n_port_tests, NULL, NULL, NULL, NULL, NULL);
#else
void test_main(void)
{
    ztest_test_suite(
        host_n_port_tests, ztest_unit_test(test_HostNPort),
        ztest_unit_test(test_HostNPortMinimal_Codec),
        ztest_unit_test(test_is_valid_hostname),
        ztest_unit_test(test_fdt_entry));

    ztest_run_test_suite(host_n_port_tests);
}
#endif