47 lines
777 B
C
47 lines
777 B
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
int call_count = 0;
|
|
void *
|
|
bad_malloc(size_t sz)
|
|
{
|
|
call_count++;
|
|
if(call_count != 2) {
|
|
return malloc(sz);
|
|
}
|
|
return NULL;
|
|
}
|
|
#define NIARENA_MALLOC(sz) bad_malloc(sz)
|
|
|
|
void
|
|
mock_free(void *p)
|
|
{
|
|
free(p);
|
|
}
|
|
#define NIARENA_FREE(p) mock_free(p)
|
|
|
|
#define N_INTS_TEST 4
|
|
#define NIARENA_ARENA_SIZE (N_INTS_TEST * sizeof(int))
|
|
#define NIARENA_IMPLEMENTATION
|
|
|
|
#include "../niarena.h"
|
|
#include <assert.h>
|
|
#include <string.h>
|
|
|
|
void
|
|
niarena_test_malloc_fail_buffer()
|
|
{
|
|
NIArena *arena = niarena_new();
|
|
assert(arena == NULL);
|
|
const char *err = niarena_get_error();
|
|
assert(strcmp(err, "niarena_new: out of memory allocating buffer") == 0);
|
|
}
|
|
|
|
int
|
|
main()
|
|
{
|
|
niarena_test_malloc_fail_buffer();
|
|
printf("== TESTS SUCCESSFUL ==\n");
|
|
return 0;
|
|
}
|