/*
 * test-alloc.c -- test allocators
 *
 * Copyright (c) 2020-2023 David Demelier <markand@malikania.fr>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <stdlib.h>

#include <mlk/core/alloc.h>

#include <dt.h>

struct point {
	int x;
	int y;
};

static struct {
	size_t alloc_count;
	size_t free_count;
} my_stats;

static void *
my_alloc(size_t n)
{
	my_stats.alloc_count += 1;

	return malloc(n);
}

static void *
my_realloc(void *ptr, size_t n)
{
	(void)ptr;
	(void)n;

	return NULL;
}

static void
my_free(void *ptr)
{
	free(ptr);
	my_stats.free_count += 1;
}

static const struct mlk_alloc_funcs my_funcs = {
	.alloc = my_alloc,
	.realloc = my_realloc,
	.free = my_free
};

static void
test_basics_resize0(void)
{
	struct point *points;

	points = mlk_alloc_new0(2, sizeof (*points));
	DT_EQ_SIZE(mlk_alloc_getn(points), 2);
	DT_EQ_SIZE(mlk_alloc_getw(points), sizeof (*points));
	DT_EQ_INT(points[0].x, 0);
	DT_EQ_INT(points[0].y, 0);
	DT_EQ_INT(points[1].x, 0);
	DT_EQ_INT(points[1].y, 0);

	points[0].x = 10;
	points[0].y = 20;
	points[1].x = 30;
	points[1].y = 40;

	points = mlk_alloc_resize0(points, 4);
	DT_EQ_SIZE(mlk_alloc_getn(points), 4);
	DT_EQ_SIZE(mlk_alloc_getw(points), sizeof (*points));

	/* Make sure previous are still correct. */
	DT_EQ_INT(points[0].x, 10);
	DT_EQ_INT(points[0].y, 20);
	DT_EQ_INT(points[1].x, 30);
	DT_EQ_INT(points[1].y, 40);

	/* Now the new items must be zero'ed. */
	DT_EQ_INT(points[2].x, 0);
	DT_EQ_INT(points[2].y, 0);
	DT_EQ_INT(points[3].x, 0);
	DT_EQ_INT(points[3].y, 0);
}

static void
test_basics_expand0(void)
{
	struct point *points;

	points = mlk_alloc_new0(2, sizeof (*points));
	DT_EQ_SIZE(mlk_alloc_getn(points), 2);
	DT_EQ_SIZE(mlk_alloc_getw(points), sizeof (*points));
	DT_EQ_INT(points[0].x, 0);
	DT_EQ_INT(points[0].y, 0);
	DT_EQ_INT(points[1].x, 0);
	DT_EQ_INT(points[1].y, 0);

	points[0].x = 10;
	points[0].y = 20;
	points[1].x = 30;
	points[1].y = 40;

	points = mlk_alloc_expand0(points, 4);
	DT_EQ_SIZE(mlk_alloc_getn(points), 6);
	DT_EQ_SIZE(mlk_alloc_getw(points), sizeof (*points));

	/* Make sure previous are still correct. */
	DT_EQ_INT(points[0].x, 10);
	DT_EQ_INT(points[0].y, 20);
	DT_EQ_INT(points[1].x, 30);
	DT_EQ_INT(points[1].y, 40);

	/* Now the new items must be zero'ed. */
	DT_EQ_INT(points[2].x, 0);
	DT_EQ_INT(points[2].y, 0);
	DT_EQ_INT(points[3].x, 0);
	DT_EQ_INT(points[3].y, 0);
	DT_EQ_INT(points[4].x, 0);
	DT_EQ_INT(points[4].y, 0);
	DT_EQ_INT(points[5].x, 0);
	DT_EQ_INT(points[5].y, 0);
}

static void
test_basics_sdupf(void)
{
	char *str = mlk_alloc_sdupf("Hello %s", "David");

	DT_EQ_STR(str, "Hello David");
	mlk_alloc_free(str);
}

static void
test_custom_count(void)
{
	mlk_alloc_set(&my_funcs);
	mlk_alloc_free(mlk_alloc_new(10, 1));
	mlk_alloc_free(mlk_alloc_new0(20, 1));
	mlk_alloc_free(mlk_alloc_sdup("malikania"));

	DT_EQ_UINT(my_stats.alloc_count, 3U);
	DT_EQ_UINT(my_stats.free_count, 3U);
}

int
main(void)
{
	DT_RUN(test_basics_resize0);
	DT_RUN(test_basics_expand0);
	DT_RUN(test_basics_sdupf);
	DT_RUN(test_custom_count);
	DT_SUMMARY();
}