#include <zmq.h>
#include "LiveRecvModule.hpp"
#include "gtest/gtest.h"
#include "buffer_config.hpp"

using namespace std;
using namespace core_buffer;


TEST(LiveRecvModule, basic_interaction) {
    auto ctx = zmq_ctx_new();
    string ipc_prefix = "ipc:///tmp/sf-live-";

    size_t n_modules = 32;
    size_t n_slots = 5;
    FastQueue<ModuleFrameBuffer> queue(MODULE_N_BYTES * n_modules, n_slots);

    void *sockets[n_modules];
    for (size_t i = 0; i < n_modules; i++) {
        sockets[i] = zmq_socket(ctx, ZMQ_PUB);

        int linger = 0;
        if (zmq_setsockopt(sockets[i], ZMQ_LINGER, &linger,
                           sizeof(linger)) != 0) {
            throw runtime_error(strerror(errno));
        }

        stringstream ipc_addr;
        ipc_addr << ipc_prefix << i;
        const auto ipc = ipc_addr.str();

        if (zmq_bind(sockets[i], ipc.c_str()) != 0) {
            throw runtime_error(strerror(errno));
        }
    }

    LiveRecvModule recv_module(queue, n_modules, ctx, ipc_prefix);

    // Nothing should be committed, queue, should be empty.
    ASSERT_EQ(queue.read(), -1);


    for (size_t i = 0; i < n_modules; i++) {
        zmq_close(sockets[i]);
    }
    zmq_ctx_destroy(ctx);
}