From cbfc3efbfd4a3a6f3b031ce8ef375d37f2c545f3 Mon Sep 17 00:00:00 2001 From: Jeremy Allison Date: Tue, 27 Aug 2013 13:09:03 -0700 Subject: Add simple limited pool tests to test_memlimit(). Signed-off-by: Jeremy Allison Reviewed-by: Simo Sorce --- lib/talloc/testsuite.c | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) (limited to 'lib/talloc') diff --git a/lib/talloc/testsuite.c b/lib/talloc/testsuite.c index d456cbb0c23..426c31a8f24 100644 --- a/lib/talloc/testsuite.c +++ b/lib/talloc/testsuite.c @@ -1359,6 +1359,8 @@ static bool test_memlimit(void) { void *root; char *l1, *l2, *l3, *l4, *l5, *t; + char *pool; + int i; printf("test: memlimit\n# MEMORY LIMITS\n"); @@ -1520,6 +1522,31 @@ static bool test_memlimit(void) talloc_report_full(root, stdout); talloc_free(root); + /* Test memlimits with pools. */ + pool = talloc_pool(NULL, 10*1024); + torture_assert("memlimit", pool != NULL, + "failed: alloc should not fail due to memory limit\n"); + talloc_set_memlimit(pool, 10*1024); + for (i = 0; i < 9; i++) { + l1 = talloc_size(pool, 1024); + torture_assert("memlimit", l1 != NULL, + "failed: alloc should not fail due to memory limit\n"); + } + /* The next alloc should fail. */ + l2 = talloc_size(pool, 1024); + torture_assert("memlimit", l2 == NULL, + "failed: alloc should fail due to memory limit\n"); + + /* Moving one of the children shouldn't change the limit, + as it's still inside the pool. */ + root = talloc_new(NULL); + talloc_steal(root, l1); + l2 = talloc_size(pool, 1024); + torture_assert("memlimit", l2 == NULL, + "failed: alloc should fail due to memory limit\n"); + + talloc_free(pool); + talloc_free(root); printf("success: memlimit\n"); return true; -- cgit