summaryrefslogtreecommitdiff
path: root/tests/main.cpp
blob: 664f17118c13d385ecaa8bbe854e2551b95e467b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include "test.hpp"

#include <exception>
#include <stdio.h>

#include <malloc.h>

test_runner* test_runner::_tests = 0;
size_t test_runner::_memory_fail_threshold = 0;

size_t g_memory_total_size = 0;

void* custom_allocate(size_t size)
{
	if (test_runner::_memory_fail_threshold > 0 && test_runner::_memory_fail_threshold < size)
		return 0;
	else
	{
		void* ptr = malloc(size);

		g_memory_total_size += _msize(ptr);
		
		return ptr;
	}
}

void custom_deallocate(void* ptr)
{
	if (ptr)
	{
		g_memory_total_size -= _msize(ptr);
		
		free(ptr);
	}
}

void replace_memory_management()
{
	// create some document to touch original functions
	{
		pugi::xml_document doc;
		doc.append_child().set_name("node");
	}

	// replace functions
	pugi::set_memory_management_functions(custom_allocate, custom_deallocate);
}

int main()
{
	replace_memory_management();

	unsigned int total = 0;
	unsigned int passed = 0;

	for (test_runner* test = test_runner::_tests; test; test = test->_next)
	{
		try
		{
			total++;

			g_memory_total_size = 0;
			test_runner::_memory_fail_threshold = 0;

			test->run();

			if (g_memory_total_size != 0) throw "Memory leaks found";

			passed++;
		}
		catch (const std::exception& e)
		{
			printf("Test %s failed: exception %s\n", test->_name, e.what());
		}
		catch (const char* e)
		{
			printf("Test %s failed: %s\n", test->_name, e);
		}
		catch (...)
		{
			printf("Test %s failed for unknown reason\n", test->_name);
		}
	}

	unsigned int failed = total - passed;

	if (failed != 0)
		printf("FAILURE: %u out of %u tests failed.\n", failed, total);
	else
		printf("Success: %d tests passed.\n", total);

	return failed;
}