#include "utils.h"
#include <catch2/catch.hpp>

#include <faabric/redis/Redis.h>
#include <faabric/state/State.h>
#include <faasm/array.h>
#include <faasm/emulator.h>

namespace tests {
TEST_CASE("Test array operations", "[counter]")
{
    cleanCppTests();
    faabric::state::State& state = faabric::state::getGlobalState();
    std::string user = getEmulatorUser();

    // Create the array and check it makes a KV in the background
    REQUIRE(state.getKVCount() == 0);

    const char* key = "test_array";
    int arraySize = 10;
    faasm::AsyncArray<int> arrayA(key, arraySize);
    arrayA.zero();

    REQUIRE(state.getKVCount() == 1);

    // Check state and array point to the same thing
    auto stateKv = state.getKV(user, key, arraySize * sizeof(int));
    auto stateMem = reinterpret_cast<int*>(stateKv->get());
    REQUIRE(arrayA.data() == stateMem);

    // Check state is zeroed
    std::vector<int> actual(stateMem, stateMem + arraySize);
    std::vector<int> expected(arraySize, 0);
    REQUIRE(actual == expected);

    // Write values, check they're reflected in memory
    arrayA[5] = 55;
    arrayA[2] = 22;
    arrayA.push();

    REQUIRE(stateMem[5] == 55);
    REQUIRE(stateMem[2] == 22);
    REQUIRE(arrayA[5] == 55);
    REQUIRE(arrayA[2] == 22);

    // Create another instance and check same state is reflected
    faasm::AsyncArray<int> arrayB(key, arraySize);
    arrayB.pull();
    REQUIRE(arrayA.data() == arrayB.data());
    REQUIRE(arrayB[5] == 55);
    REQUIRE(arrayB[2] == 22);

    // Update the second instance
    arrayB[3] = 33;
    arrayB.push();

    // Check it's reflected by the first
    arrayA.pullLazy();
    REQUIRE(arrayA[5] == 55);
    REQUIRE(arrayA[2] == 22);
    REQUIRE(arrayA[3] == 33);
}
}