#ifndef C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H
#define C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H

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

struct Memory_Class {
    static Tox_Memory_Funcs const vtable;
    Tox_Memory const self;

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

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

    virtual ~Memory_Class();
    virtual tox_memory_malloc_cb malloc = 0;
    virtual tox_memory_realloc_cb realloc = 0;
    virtual tox_memory_dealloc_cb dealloc = 0;
};

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

    void *malloc(void *obj, uint32_t size) override;
    void *realloc(void *obj, void *ptr, uint32_t size) override;
    void dealloc(void *obj, void *ptr) override;
};

#endif  // C_TOXCORE_TOXCORE_MEM_TEST_UTIL_H