#include "common.hpp"

namespace
{
	pugi::char_t buffer[8];
	int allocate_count = 0;
	int deallocate_count = 0;

	void* allocate(size_t size)
	{
		CHECK(size == sizeof(pugi::char_t) * 8);
		++allocate_count;
		return buffer;
	}

	void deallocate(void* ptr)
	{
		CHECK(ptr == buffer);
		++deallocate_count;
	}
}

TEST(custom_memory_management)
{
	// remember old functions
	allocation_function old_allocate = get_memory_allocation_function();
	deallocation_function old_deallocate = get_memory_deallocation_function();

	// replace functions
	set_memory_management_functions(allocate, deallocate);

	{
		// parse document
		xml_document doc;
		CHECK(doc.load(STR("<node />")));
	
		CHECK(allocate_count == 1);
		CHECK(deallocate_count == 0);
		CHECK_STRING(buffer, STR("<node"));

		// modify document
		doc.child(STR("node")).set_name(STR("foobars"));

		CHECK(allocate_count == 2);
		CHECK(deallocate_count == 0);
		CHECK_STRING(buffer, STR("foobars"));
	}

	CHECK(allocate_count == 2);
	CHECK(deallocate_count == 2);
	CHECK_STRING(buffer, STR("foobars"));

	// restore old functions
	set_memory_management_functions(old_allocate, old_deallocate);
}