summaryrefslogtreecommitdiffstats
path: root/tests/util/str/oom.c
blob: 810c1f25a7a1bd360535e49ba7626f54913b9bfc (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
#include "clar_libgit2.h"
#include "clar_libgit2_alloc.h"

/* Override default allocators with ones that will fail predictably. */

static git_allocator std_alloc;
static git_allocator oom_alloc;

static void *oom_malloc(size_t n, const char *file, int line)
{
	/* Reject any allocation of more than 100 bytes */
	return (n > 100) ? NULL : std_alloc.gmalloc(n, file, line);
}

static void *oom_realloc(void *p, size_t n, const char *file, int line)
{
	/* Reject any allocation of more than 100 bytes */
	return (n > 100) ? NULL : std_alloc.grealloc(p, n, file, line);
}

void test_str_oom__initialize(void)
{
	git_stdalloc_init_allocator(&std_alloc);
	git_stdalloc_init_allocator(&oom_alloc);

	oom_alloc.gmalloc = oom_malloc;
	oom_alloc.grealloc = oom_realloc;

	cl_git_pass(git_allocator_setup(&oom_alloc));
}

void test_str_oom__cleanup(void)
{
	cl_git_pass(git_allocator_setup(NULL));
}

void test_str_oom__grow(void)
{
	git_str buf = GIT_STR_INIT;

	cl_git_pass(git_str_grow(&buf, 42));
	cl_assert(!git_str_oom(&buf));

	cl_assert(git_str_grow(&buf, 101) == -1);
	cl_assert(git_str_oom(&buf));

	git_str_dispose(&buf);
}

void test_str_oom__grow_by(void)
{
	git_str buf = GIT_STR_INIT;

	cl_git_pass(git_str_grow_by(&buf, 42));
	cl_assert(!git_str_oom(&buf));

	cl_assert(git_str_grow_by(&buf, 101) == -1);
	cl_assert(git_str_oom(&buf));
}

void test_str_oom__allocation_failure(void)
{
	git_str buf = GIT_STR_INIT;

	cl_alloc_limit(10);

	cl_git_pass(git_str_puts(&buf, "foobar"));
	cl_git_fail(git_str_puts(&buf, "foobar"));

	cl_alloc_reset();
}