/*
 * Runner for littlefs tests
 *
 * Copyright (c) 2022, The littlefs authors.
 * SPDX-License-Identifier: BSD-3-Clause
 */
#ifndef TEST_RUNNER_H
#define TEST_RUNNER_H


// override LFS3_TRACE
void test_trace(const char *fmt, ...);

#define LFS3_TRACE_(fmt, ...) \
    test_trace("%s:%d:trace: " fmt "%s\n", \
        __FILE__, \
        __LINE__, \
        __VA_ARGS__)
#define LFS3_TRACE(...) LFS3_TRACE_(__VA_ARGS__, "")
#define LFS3_EMUBD_TRACE(...) LFS3_TRACE_(__VA_ARGS__, "")


// note these are indirectly included in any generated files
#include "bd/lfs3_emubd.h"
#include <stdio.h>

// give source a chance to define feature macros
#undef _FEATURES_H
#undef _STDIO_H


// generated test configurations
struct lfs3_cfg;

enum test_flags {
    TEST_INTERNAL  = 0x1,
    TEST_REENTRANT = 0x2,
    TEST_FUZZ      = 0x4,
};
typedef uint8_t test_flags_t;

typedef struct test_define {
    const char *name;
    intmax_t *define;
    intmax_t (*cb)(void *data, size_t i);
    void *data;
    size_t permutations;
} test_define_t;

struct test_case {
    const char *name;
    const char *path;
    test_flags_t flags;

    const test_define_t *defines;
    size_t permutations;

    bool (*if_)(void);
    void (*run)(struct lfs3_cfg *cfg);
};

struct test_suite {
    const char *name;
    const char *path;
    test_flags_t flags;

    const test_define_t *defines;
    size_t define_count;

    const struct test_case *cases;
    size_t case_count;
};

extern const struct test_suite *const test_suites[];
extern const size_t test_suite_count;


// this variable tracks the number of powerlosses triggered during the
// current test permutation, this is useful for both tests and debugging
extern volatile size_t TEST_PLS;

// deterministic prng for pseudo-randomness in tests
uint32_t test_prng(uint32_t *state);

#define TEST_PRNG(state) test_prng(state)

// generation of specific permutations of an array for exhaustive testing
size_t test_factorial(size_t x);
void test_permutation(size_t i, uint32_t *buffer, size_t size);

#define TEST_FACTORIAL(x) test_factorial(x)
#define TEST_PERMUTATION(i, buffer, size) test_permutation(i, buffer, size)


// a few preconfigured defines that control how tests run
#define TEST_IMPLICIT_DEFINES \
    /*          name                    value (overridable)                 */ \
    TEST_DEFINE(READ_SIZE,              1                                    ) \
    TEST_DEFINE(PROG_SIZE,              1                                    ) \
    TEST_DEFINE(BLOCK_SIZE,             4096                                 ) \
    TEST_DEFINE(BLOCK_COUNT,            DISK_SIZE/BLOCK_SIZE                 ) \
    TEST_DEFINE(DISK_SIZE,              1024*1024                            ) \
    TEST_DEFINE(BLOCK_RECYCLES,         -1                                   ) \
    TEST_DEFINE(RCACHE_SIZE,            LFS3_MAX(16, READ_SIZE)              ) \
    TEST_DEFINE(PCACHE_SIZE,            LFS3_MAX(16, PROG_SIZE)              ) \
    TEST_DEFINE(FILE_CACHE_SIZE,        16                                   ) \
    TEST_DEFINE(LOOKAHEAD_SIZE,         16                                   ) \
    TEST_DEFINE(GC_FLAGS,               0                                    ) \
    TEST_DEFINE(GC_STEPS,               0                                    ) \
    TEST_DEFINE(GC_COMPACT_THRESH,      0                                    ) \
    TEST_DEFINE(INLINE_SIZE,            BLOCK_SIZE/4                         ) \
    TEST_DEFINE(FRAGMENT_SIZE,          LFS3_MIN(BLOCK_SIZE/8, 512)          ) \
    TEST_DEFINE(CRYSTAL_THRESH,         BLOCK_SIZE/8                         ) \
    TEST_DEFINE(GBMAP_REPOP_THRESH,     BLOCK_COUNT/4                        ) \
    TEST_DEFINE(ERASE_VALUE,            0xff                                 ) \
    TEST_DEFINE(ERASE_CYCLES,           0                                    ) \
    TEST_DEFINE(BADBLOCK_BEHAVIOR,      LFS3_EMUBD_BADBLOCK_PROGERROR        ) \
    TEST_DEFINE(POWERLOSS_BEHAVIOR,     LFS3_EMUBD_POWERLOSS_ATOMIC          ) \
    TEST_DEFINE(EMUBD_SEED,             0                                    )

// declare defines as global intmax_ts
#define TEST_DEFINE(k, v) \
    extern intmax_t k;

    TEST_IMPLICIT_DEFINES
#undef TEST_DEFINE

// map defines to cfg struct fields
#define TEST_CFG \
    .read_size              = READ_SIZE,            \
    .prog_size              = PROG_SIZE,            \
    .block_size             = BLOCK_SIZE,           \
    .block_count            = BLOCK_COUNT,          \
    .block_recycles         = BLOCK_RECYCLES,       \
    .rcache_size            = RCACHE_SIZE,          \
    .pcache_size            = PCACHE_SIZE,          \
    .file_cache_size        = FILE_CACHE_SIZE,      \
    .lookahead_size         = LOOKAHEAD_SIZE,       \
    TEST_GBMAP_CFG                                  \
    TEST_GC_CFG                                     \
    .gc_compact_thresh      = GC_COMPACT_THRESH,    \
    .inline_size            = INLINE_SIZE,          \
    .fragment_size          = FRAGMENT_SIZE,        \
    .crystal_thresh         = CRYSTAL_THRESH,

#ifdef LFS3_GBMAP
#define TEST_GBMAP_CFG \
    .gbmap_repop_thresh     = GBMAP_REPOP_THRESH,
#else
#define TEST_GBMAP_CFG
#endif

#ifdef LFS3_GC
#define TEST_GC_CFG \
    .gc_flags               = GC_FLAGS,             \
    .gc_steps               = GC_STEPS,
#else
#define TEST_GC_CFG
#endif

#define TEST_BDCFG \
    .erase_value            = ERASE_VALUE,          \
    .erase_cycles           = ERASE_CYCLES,         \
    .badblock_behavior      = BADBLOCK_BEHAVIOR,    \
    .powerloss_behavior     = POWERLOSS_BEHAVIOR,   \
    .seed                   = EMUBD_SEED,


#endif