#include "test/jemalloc_test.h"

#include "jemalloc/internal/log.h"

static void

expect_no_logging(const char *names) {

	log_var_t log_l1 = LOG_VAR_INIT("l1");

	log_var_t log_l2 = LOG_VAR_INIT("l2");

	log_var_t log_l2_a = LOG_VAR_INIT("l2.a");

	strcpy(log_var_names, names);

	int count = 0;

	for (int i = 0; i < 10; i++) {

		log_do_begin(log_l1)

			count++;

		log_do_end(log_l1)

		log_do_begin(log_l2)

			count++;

		log_do_end(log_l2)

		log_do_begin(log_l2_a)

			count++;

		log_do_end(log_l2_a)

	}

	assert_d_eq(count, 0, "Disabled logging not ignored!");

}

TEST_BEGIN(test_log_disabled) {

	test_skip_if(!config_log);

	atomic_store_b(&log_init_done, true, ATOMIC_RELAXED);

	expect_no_logging("");

	expect_no_logging("abc");

	expect_no_logging("a.b.c");

	expect_no_logging("l12");

	expect_no_logging("l123|a456|b789");

	expect_no_logging("|||");

}

TEST_END

TEST_BEGIN(test_log_enabled_direct) {

	test_skip_if(!config_log);

	atomic_store_b(&log_init_done, true, ATOMIC_RELAXED);

	log_var_t log_l1 = LOG_VAR_INIT("l1");

	log_var_t log_l1_a = LOG_VAR_INIT("l1.a");

	log_var_t log_l2 = LOG_VAR_INIT("l2");

	int count;

	count = 0;

	strcpy(log_var_names, "l1");

	for (int i = 0; i < 10; i++) {

		log_do_begin(log_l1)

			count++;

		log_do_end(log_l1)

	}

	assert_d_eq(count, 10, "Mis-logged!");

	count = 0;

	strcpy(log_var_names, "l1.a");

	for (int i = 0; i < 10; i++) {

		log_do_begin(log_l1_a)

			count++;

		log_do_end(log_l1_a)

	}

	assert_d_eq(count, 10, "Mis-logged!");

	count = 0;

	strcpy(log_var_names, "l1.a|abc|l2|def");

	for (int i = 0; i < 10; i++) {

		log_do_begin(log_l1_a)

			count++;

		log_do_end(log_l1_a)

		log_do_begin(log_l2)

			count++;

		log_do_end(log_l2)

	}

	assert_d_eq(count, 20, "Mis-logged!");

}

TEST_END

TEST_BEGIN(test_log_enabled_indirect) {

	test_skip_if(!config_log);

	atomic_store_b(&log_init_done, true, ATOMIC_RELAXED);

	strcpy(log_var_names, "l0|l1|abc|l2.b|def");

	/* On. */

	log_var_t log_l1 = LOG_VAR_INIT("l1");

	/* Off. */

	log_var_t log_l1a = LOG_VAR_INIT("l1a");

	/* On. */

	log_var_t log_l1_a = LOG_VAR_INIT("l1.a");

	/* Off. */

	log_var_t log_l2_a = LOG_VAR_INIT("l2.a");

	/* On. */

	log_var_t log_l2_b_a = LOG_VAR_INIT("l2.b.a");

	/* On. */

	log_var_t log_l2_b_b = LOG_VAR_INIT("l2.b.b");

	/* 4 are on total, so should sum to 40. */

	int count = 0;

	for (int i = 0; i < 10; i++) {

		log_do_begin(log_l1)

			count++;

		log_do_end(log_l1)

		log_do_begin(log_l1a)

			count++;

		log_do_end(log_l1a)

		log_do_begin(log_l1_a)

			count++;

		log_do_end(log_l1_a)

		log_do_begin(log_l2_a)

			count++;

		log_do_end(log_l2_a)

		log_do_begin(log_l2_b_a)

			count++;

		log_do_end(log_l2_b_a)

		log_do_begin(log_l2_b_b)

			count++;

		log_do_end(log_l2_b_b)

	}

	assert_d_eq(count, 40, "Mis-logged!");

}

TEST_END

TEST_BEGIN(test_log_enabled_global) {

	test_skip_if(!config_log);

	atomic_store_b(&log_init_done, true, ATOMIC_RELAXED);

	strcpy(log_var_names, "abc|.|def");

	log_var_t log_l1 = LOG_VAR_INIT("l1");

	log_var_t log_l2_a_a = LOG_VAR_INIT("l2.a.a");

	int count = 0;

	for (int i = 0; i < 10; i++) {

		log_do_begin(log_l1)

		    count++;

		log_do_end(log_l1)

		log_do_begin(log_l2_a_a)

		    count++;

		log_do_end(log_l2_a_a)

	}

	assert_d_eq(count, 20, "Mis-logged!");

}

TEST_END

TEST_BEGIN(test_logs_if_no_init) {

	test_skip_if(!config_log);

	atomic_store_b(&log_init_done, false, ATOMIC_RELAXED);

	log_var_t l = LOG_VAR_INIT("definitely.not.enabled");

	int count = 0;

	for (int i = 0; i < 10; i++) {

		log_do_begin(l)

			count++;

		log_do_end(l)

	}

	assert_d_eq(count, 0, "Logging shouldn't happen if not initialized.");

}

TEST_END

/*

 * This really just checks to make sure that this usage compiles; we don't have

 * any test code to run.

 */

TEST_BEGIN(test_log_only_format_string) {

	if (false) {

		LOG("log_str", "No arguments follow this format string.");

	}

}

TEST_END

int

main(void) {

	return test(

	    test_log_disabled,

	    test_log_enabled_direct,

	    test_log_enabled_indirect,

	    test_log_enabled_global,

	    test_logs_if_no_init,

	    test_log_only_format_string);

}