#ifndef C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H
#define C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H

#include "mem.h"
#include "test_util.hh"

struct Memory_Class {
    static Memory_Funcs const vtable;
    Memory const self;

    operator Memory const *() const { return &self; }

    Memory_Class(Memory_Class const &) = default;
    Memory_Class()
        : self{&vtable, this}
    {
    }

    virtual ~Memory_Class();
    virtual mem_malloc_cb malloc = 0;
    virtual mem_calloc_cb calloc = 0;
    virtual mem_realloc_cb realloc = 0;
    virtual mem_free_cb free = 0;
};

/**
 * Base test Memory class that just forwards to system_memory. Can be
 * subclassed to override individual (or all) functions.
 */
class Test_Memory : public Memory_Class {
    const Memory *mem = REQUIRE_NOT_NULL(system_memory());

    void *malloc(void *obj, uint32_t size) override;
    void *calloc(void *obj, uint32_t nmemb, uint32_t size) override;
    void *realloc(void *obj, void *ptr, uint32_t size) override;
    void free(void *obj, void *ptr) override;
};

#endif  // C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H