diff options
Diffstat (limited to 'tests/test_memory.cpp')
-rw-r--r-- | tests/test_memory.cpp | 258 |
1 files changed, 129 insertions, 129 deletions
diff --git a/tests/test_memory.cpp b/tests/test_memory.cpp index 80e36c2..8b4b6bc 100644 --- a/tests/test_memory.cpp +++ b/tests/test_memory.cpp @@ -1,129 +1,129 @@ -#include "common.hpp"
-
-#include <string>
-
-namespace
-{
- int allocate_count = 0;
- int deallocate_count = 0;
-
- void* allocate(size_t size)
- {
- ++allocate_count;
- return new char[size];
- }
-
- void deallocate(void* ptr)
- {
- ++deallocate_count;
- delete[] reinterpret_cast<char*>(ptr);
- }
-}
-
-TEST(custom_memory_management)
-{
- allocate_count = deallocate_count = 0;
-
- // 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(allocate_count == 0 && deallocate_count == 0);
-
- CHECK(doc.load(STR("<node />")));
-
- CHECK(allocate_count == 2 && deallocate_count == 0);
-
- // modify document (no new page)
- CHECK(doc.first_child().set_name(STR("foobars")));
- CHECK(allocate_count == 2 && deallocate_count == 0);
-
- // modify document (new page)
- std::basic_string<pugi::char_t> s(65536, 'x');
-
- CHECK(doc.first_child().set_name(s.c_str()));
- CHECK(allocate_count == 3 && deallocate_count == 0);
-
- // modify document (new page, old one should die)
- s += s;
-
- CHECK(doc.first_child().set_name(s.c_str()));
- CHECK(allocate_count == 4 && deallocate_count == 1);
- }
-
- CHECK(allocate_count == 4 && deallocate_count == 4);
-
- // restore old functions
- set_memory_management_functions(old_allocate, old_deallocate);
-}
-
-TEST(large_allocations)
-{
- allocate_count = deallocate_count = 0;
-
- // 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);
-
- {
- xml_document doc;
-
- CHECK(allocate_count == 0 && deallocate_count == 0);
-
- // initial fill
- for (size_t i = 0; i < 128; ++i)
- {
- std::basic_string<pugi::char_t> s(i * 128, 'x');
-
- CHECK(doc.append_child(node_pcdata).set_value(s.c_str()));
- }
-
- CHECK(allocate_count > 0 && deallocate_count == 0);
-
- // grow-prune loop
- while (doc.first_child())
- {
- pugi::xml_node node;
-
- // grow
- for (node = doc.first_child(); node; node = node.next_sibling())
- {
- std::basic_string<pugi::char_t> s = node.value();
-
- CHECK(node.set_value((s + s).c_str()));
- }
-
- // prune
- for (node = doc.first_child(); node; )
- {
- pugi::xml_node next = node.next_sibling().next_sibling();
-
- node.parent().remove_child(node);
-
- node = next;
- }
- }
-
- CHECK(allocate_count == deallocate_count + 1); // only one live page left (it waits for new allocations)
-
- char buffer;
- CHECK(doc.load_buffer_inplace(&buffer, 0, parse_default, get_native_encoding()));
-
- CHECK(allocate_count == deallocate_count); // no live pages left
- }
-
- CHECK(allocate_count == deallocate_count); // everything is freed
-
- // restore old functions
- set_memory_management_functions(old_allocate, old_deallocate);
-}
+#include "common.hpp" + +#include <string> + +namespace +{ + int allocate_count = 0; + int deallocate_count = 0; + + void* allocate(size_t size) + { + ++allocate_count; + return new char[size]; + } + + void deallocate(void* ptr) + { + ++deallocate_count; + delete[] reinterpret_cast<char*>(ptr); + } +} + +TEST(custom_memory_management) +{ + allocate_count = deallocate_count = 0; + + // 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(allocate_count == 0 && deallocate_count == 0); + + CHECK(doc.load(STR("<node />"))); + + CHECK(allocate_count == 2 && deallocate_count == 0); + + // modify document (no new page) + CHECK(doc.first_child().set_name(STR("foobars"))); + CHECK(allocate_count == 2 && deallocate_count == 0); + + // modify document (new page) + std::basic_string<pugi::char_t> s(65536, 'x'); + + CHECK(doc.first_child().set_name(s.c_str())); + CHECK(allocate_count == 3 && deallocate_count == 0); + + // modify document (new page, old one should die) + s += s; + + CHECK(doc.first_child().set_name(s.c_str())); + CHECK(allocate_count == 4 && deallocate_count == 1); + } + + CHECK(allocate_count == 4 && deallocate_count == 4); + + // restore old functions + set_memory_management_functions(old_allocate, old_deallocate); +} + +TEST(large_allocations) +{ + allocate_count = deallocate_count = 0; + + // 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); + + { + xml_document doc; + + CHECK(allocate_count == 0 && deallocate_count == 0); + + // initial fill + for (size_t i = 0; i < 128; ++i) + { + std::basic_string<pugi::char_t> s(i * 128, 'x'); + + CHECK(doc.append_child(node_pcdata).set_value(s.c_str())); + } + + CHECK(allocate_count > 0 && deallocate_count == 0); + + // grow-prune loop + while (doc.first_child()) + { + pugi::xml_node node; + + // grow + for (node = doc.first_child(); node; node = node.next_sibling()) + { + std::basic_string<pugi::char_t> s = node.value(); + + CHECK(node.set_value((s + s).c_str())); + } + + // prune + for (node = doc.first_child(); node; ) + { + pugi::xml_node next = node.next_sibling().next_sibling(); + + node.parent().remove_child(node); + + node = next; + } + } + + CHECK(allocate_count == deallocate_count + 1); // only one live page left (it waits for new allocations) + + char buffer; + CHECK(doc.load_buffer_inplace(&buffer, 0, parse_default, get_native_encoding())); + + CHECK(allocate_count == deallocate_count); // no live pages left + } + + CHECK(allocate_count == deallocate_count); // everything is freed + + // restore old functions + set_memory_management_functions(old_allocate, old_deallocate); +} |