#pragma once

#include <map>
#include <string>

#include "key_value_store.h"

// In-memory KeyValueStore for tests.
struct MemoryStore : roro::KeyValueStore {
    std::map<std::string, int32_t> ints;
    std::map<std::string, std::string> strings;
    bool getInt(const char* key, int32_t& out) override {
        auto it = ints.find(key);
        if (it == ints.end()) return false;
        out = it->second;
        return true;
    }
    bool getString(const char* key, std::string& out) override {
        auto it = strings.find(key);
        if (it == strings.end()) return false;
        out = it->second;
        return true;
    }
    void putInt(const char* key, int32_t v) override { ints[key] = v; }
    void putString(const char* key, const std::string& v) override { strings[key] = v; }
};