// Copyright 2022 Memgraph Ltd.
//
// Use of this software is governed by the Business Source License
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
// License, and you may not use this file except in compliance with the Business Source License.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0, included in the file
// licenses/APL.txt.

//#include <gtest/gtest.h>

#include <string>
#include <vector>

#include "io/v3/simulator.hpp"
#include "utils/terminate_handler.hpp"

struct RequestMsg {
  std::string data;

  std::vector<uint8_t> Serialize() { return std::vector<uint8_t>(); }

  static RequestMsg Deserialize(uint8_t *ptr, size_t len) { return RequestMsg{}; }
};

struct ResponseMsg {
  std::string data;

  std::vector<uint8_t> Serialize() { return std::vector<uint8_t>(); }

  static ResponseMsg Deserialize(uint8_t *ptr, size_t len) { return ResponseMsg{}; }
};

int main() {
  auto simulator = Simulator();
  auto cli_addr = Address::TestAddress(1);
  auto srv_addr = Address::TestAddress(2);

  Io<SimulatorTransport> cli_io = simulator.Register(cli_addr, false);
  Io<SimulatorTransport> srv_io = simulator.Register(srv_addr, false);

  // send request
  RequestMsg cli_req;
  cli_req.data = "hello";
  ResponseFuture<ResponseMsg> response_future = cli_io.Request<RequestMsg, ResponseMsg>(srv_addr, cli_req);

  // receive request
  RequestResult<RequestMsg> request_result = srv_io.Receive<RequestMsg>();
  auto request_envelope = request_result.GetValue();
  RequestMsg req = std::get<RequestMsg>(request_envelope.message);

  auto srv_res = ResponseMsg{req.data};

  // send response
  // srv_io.Send(request_envelope.from_address, request_envelope.request_id, srv_res);
  request_envelope.Reply(srv_res, srv_io);

  // receive response
  auto response_result = response_future.Wait();
  auto response_envelope = response_result.GetValue();

  MG_ASSERT(response_envelope.message.data == "hello");

  std::cout << "IT WORKED :)" << std::endl;

  return 0;
}