#include <criterion/criterion.h>
#include "cli.h"
#include "internal/arg_handler.h"

/* Test 1: basic test */
Test(cli_handle_size, valid_number)
{
	t_ping_config	config = {0};
    const char *arg = "65527";
    int ret = cli_handle_size(arg, &config);

    cr_assert_eq(ret, CLI_SUCCESS);
    cr_assert_eq(config.packet_size, 65527);
}

/* Test 2: invalid number */
Test(cli_handle_size, invalid_number)
{
	t_ping_config	config = {0};
    const char *arg = "-42";
    int ret = cli_handle_size(arg, &config);

    cr_assert_eq(ret, CLI_ERROR);
}

/* Test 3: max test */
Test(cli_handle_size, limit_test)
{
	t_ping_config	config = {0};
    const char *arg = "65528";
    int ret = cli_handle_size(arg, &config);

    cr_assert_eq(ret, CLI_ERROR);
}