/*-
 * SPDX-License-Identifier: BSD-3-Clause
 *
 * Copyright 2021 Lutz Donnerhacke
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above
 *    copyright notice, this list of conditions and the following
 *    disclaimer in the documentation and/or other materials provided
 *    with the distribution.
 * 3. Neither the name of the copyright holder nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
 * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
 * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */
#include <atf-c.h>
#include <errno.h>
#include <stdio.h>

#include "util.h"

ATF_TC(send_recv);
ATF_TC_HEAD(send_recv, conf)
{
        atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(send_recv, dummy)
{
        char            msg[] = "test";
        ng_counter_t    r;

        ng_init();
        ng_connect(".", "a", ".", "b");
        ng_register_data("b", get_data0);
        ng_send_data("a", msg, sizeof(msg));

        ng_counter_clear(r);
        ng_handle_events(50, &r);
        ATF_CHECK(r[0] == 1);
}

ATF_TC(node);
ATF_TC_HEAD(node, conf)
{
        atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(node, dummy)
{
        char            msg[] = "test";
        ng_counter_t    r;

        ng_init();
        ng_mkpeer(".", "a", "hub", "a");
        ng_name("a", "test hub");

        ng_errors(PASS);
        ng_name("a", "test hub");
        ng_errors(FAIL);
        if (errno == EADDRINUSE)
                atf_tc_expect_fail("PR241954");
        ATF_CHECK_ERRNO(0, 1);
        atf_tc_expect_pass();

        ng_connect(".", "b", "test hub:", "b");
        ng_connect(".", "c", "test hub:", "c");
        ng_register_data("a", get_data0);
        ng_register_data("b", get_data1);
        ng_register_data("c", get_data2);

        ng_counter_clear(r);
        ng_send_data("a", msg, sizeof(msg));
        ng_handle_events(50, &r);
        ATF_CHECK(r[0] == 0 && r[1] == 1 && r[2] == 1);

        ng_rmhook(".", "b");
        ng_counter_clear(r);
        ng_send_data("a", msg, sizeof(msg));
        ng_handle_events(50, &r);
        ATF_CHECK(r[0] == 0 && r[1] == 0 && r[2] == 1);

        ng_shutdown("test hub:");
}

ATF_TC(message);
ATF_TC_HEAD(message, conf)
{
        atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(message, dummy)
{
        ng_init();
        ng_mkpeer(".", "a", "hub", "a");
        ng_name("a", "test hub");

        ng_send_msg("test hub:", "setpersistent");
        ng_rmhook(".", "a");

        ng_shutdown("test hub:");
}

ATF_TC(same_name);
ATF_TC_HEAD(same_name, conf)
{
        atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(same_name, dummy)
{
        ng_init();
        ng_mkpeer(".", "a", "hub", "a");
        ng_name("a", "test");

        ng_errors(PASS);
        ng_connect(".", "a", ".", "b");
        ATF_CHECK_ERRNO(EEXIST, 1);
        ng_connect(".", "b", ".", "b");
        ATF_CHECK_ERRNO(EEXIST, 1);
        ng_name(".", "test");
        ATF_CHECK_ERRNO(EADDRINUSE, 1);

        ng_errors(FAIL);
        ng_shutdown("test:");
}

ATF_TC(queuelimit);
ATF_TC_HEAD(queuelimit, conf)
{
        atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(queuelimit, dummy)
{
        ng_counter_t    r;
        int             i;
        char            msg[] = "test";
        const int       MAX = 1000;

        ng_init();
        ng_connect(".", "a", ".", "b");
        ng_register_data("b", get_data0);

        ng_errors(PASS);
        for (i = 0; i < MAX; i++)
        {
                ng_send_data("a", msg, sizeof(msg));
                if (errno != 0)
                        break;
                /* no ng_handle_events -> messages stall */
        }
        ng_errors(FAIL);

        ng_counter_clear(r);
        ng_handle_events(50, &r);
        ATF_CHECK(r[0] > 100);
        ATF_CHECK(r[0] == i);
        atf_tc_expect_fail("Queue full (%d)", i);
        ATF_CHECK(r[0] == MAX);
        atf_tc_expect_pass();
}

ATF_TP_ADD_TCS(basic)
{
        ATF_TP_ADD_TC(basic, send_recv);
        ATF_TP_ADD_TC(basic, node);
        ATF_TP_ADD_TC(basic, message);
        ATF_TP_ADD_TC(basic, same_name);
        ATF_TP_ADD_TC(basic, queuelimit);

        return atf_no_error();
}