/*
  Copyright (c) 2015 Red Hat, Inc. <http://www.redhat.com>
  This file is part of GlusterFS.

  This file is licensed to you under your choice of the GNU Lesser
  General Public License, version 3 or any later version (LGPLv3 or
  later), or the GNU General Public License, version 2 (GPLv2), in all
  cases as published by the Free Software Foundation.
*/

#ifndef _GF_UNITTEST_H_
#define _GF_UNITTEST_H_

#ifdef UNIT_TESTING
#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka_pbc.h>
#include <cmocka.h>

extern void
mock_assert(const int result, const char *const expression,
            const char *const file, const int line);

// Change GF_CALLOC and GF_FREE to use
// cmocka memory allocation versions
#ifdef UNIT_TESTING
#undef GF_CALLOC
#define GF_CALLOC(n, s, t) test_calloc(n, s)
#undef GF_FREE
#define GF_FREE test_free

/* Catch intended assert()'s while unit-testing */
extern void
mock_assert(const int result, const char *const expression,
            const char *const file, const int line);

#undef assert
#define assert(expression)                                                     \
    mock_assert((int)(expression), #expression, __FILE__, __LINE__);
#endif
#else
#define REQUIRE(p) /**/
#define ENSURE(p)  /**/
#endif

#endif /* _GF_UNITTEST */