/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2020 Richard Hansen <rhansen@rhansen.org>
 */

#include <errno.h>
#include <libintl.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <sys/sysmacros.h>
#include <umem.h>
#include <unistd.h>
#include "test_common.h"

const char *
_umem_debug_init(void)
{
	return ("default");
}

int
main(int argc, char *argv[])
{
	int ret = 0;
	int optc;
	while ((optc = getopt(argc, argv, "df")) != -1) {
		switch (optc) {
		case 'd':
			test_set_debug();
			break;
		case 'f':
			test_set_force();
			break;
		default:
			(void) fprintf(stderr, "Usage: %s [-df]\n", argv[0]);
			exit(1);
		}
	}

	struct {
		const char *name;
		const char *dir;
		bool malloc_fail;
		const char *want;
		int want_errno;
	} test_cases[] = {
		{
			.name = "unbound query",
			.dir = NULL,
			.want = "/usr/lib/locale/",
		},
		{
			.name = "bind malloc fail",
			.dir = "/bounddir1",
			.malloc_fail = true,
			.want = NULL,
			.want_errno = EAGAIN,
		},
		{
			.name = "query after bind malloc fail",
			.dir = NULL,
			.want = "/usr/lib/locale/",
		},
		{
			.name = "normal bind",
			.dir = "/bounddir2",
			.want = "/bounddir2",
		},
		{
			.name = "query after normal bind",
			.dir = NULL,
			.want = "/bounddir2",
		},
		{
			.name = "rebind to same",
			.dir = "/bounddir2",
			.want = "/bounddir2",
		},
		{
			.name = "query after rebind to same",
			.dir = NULL,
			.want = "/bounddir2",
		},
		{
			.name = "rebind to new",
			.dir = "/bounddir3",
			.want = "/bounddir3",
		},
		{
			.name = "query after rebind to new",
			.dir = NULL,
			.want = "/bounddir3",
		},
		{
			.name = "rebind malloc fail",
			.dir = "/bounddir4",
			.malloc_fail = true,
			.want = NULL,
			.want_errno = EAGAIN,
		},
		{
			.name = "query after rebind malloc fail",
			.dir = NULL,
			.want = "/bounddir3",
		},
	}, *tc;

	for (size_t i = 0; i < ARRAY_SIZE(test_cases); ++i) {
		tc = &test_cases[i];
		test_t t = test_start(tc->name);
		umem_setmtbf((uint_t)tc->malloc_fail);
		errno = 0;
		const char *got = bindtextdomain("domain", tc->dir);
		int got_errno = errno;
		umem_setmtbf(0);
		if (((got == NULL) != (tc->want == NULL)) ||
		    ((got != NULL) && strcmp(got, tc->want))) {
			test_failed(t, "returned %s, want %s",
			    got != NULL ? got : "<NULL>",
			    tc->want != NULL ? tc->want : "<NULL>");
			ret = 1;
		}
		if (got_errno != tc->want_errno) {
			test_failed(t, "got errno %d, want %d",
			    got_errno, tc->want_errno);
			ret = 1;
		}
		test_passed(t);
	}
	test_summary();
	return (ret);
}