tdlight/test/tqueue.cpp

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2020
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/db/binlog/Binlog.h"
#include "td/db/binlog/BinlogEvent.h"
#include "td/db/binlog/BinlogHelper.h"
#include "td/db/TQueue.h"

#include "td/utils/int_types.h"
#include "td/utils/misc.h"
#include "td/utils/port/path.h"
#include "td/utils/Slice.h"
#include "td/utils/Span.h"
#include "td/utils/Status.h"
#include "td/utils/tests.h"
#include "td/utils/VectorQueue.h"

#include <map>
#include <set>

namespace td {

TEST(TQueue, hands) {
  TQueue::Event events[100];
  auto events_span = MutableSpan<TQueue::Event>(events, 100);

  auto tqueue = TQueue::create();
  auto qid = 12;
  ASSERT_EQ(true, tqueue->get_head(qid).empty());
  ASSERT_EQ(true, tqueue->get_tail(qid).empty());
  tqueue->push(qid, "hello", 0);
  auto head = tqueue->get_head(qid);
  ASSERT_EQ(head.next().ok(), tqueue->get_tail(qid));
  ASSERT_EQ(1u, tqueue->get(qid, head, 0, events_span).move_as_ok());
}

class TestTQueue {
 public:
  CSlice binlog_path() {
    return "test_binlog";
  }
  TestTQueue() {
    baseline_ = TQueue::create();
    memory_ = TQueue::create();

    auto memory_storage = td::make_unique<MemoryStorage>();
    memory_storage_ = memory_storage.get();
    memory_->set_callback(std::move(memory_storage));

    binlog_ = TQueue::create();
    auto tqueue_binlog = make_unique<TQueueBinlog<Binlog>>();
    Binlog::destroy(binlog_path()).ensure();
    auto binlog = std::make_shared<Binlog>();
    binlog->init(binlog_path().str(), [&](const BinlogEvent &event) { UNREACHABLE(); }).ensure();
    tqueue_binlog->set_binlog(binlog);
    binlog_->set_callback(std::move(tqueue_binlog));
  }

  void restart(Random::Xorshift128plus &rnd) {
    baseline_->emulate_restart();

    memory_->extract_callback().release();
    auto memory_storage = unique_ptr<MemoryStorage>(memory_storage_);
    memory_ = TQueue::create();
    memory_storage->replay(*memory_);
    memory_->set_callback(std::move(memory_storage));

    if (rnd.fast(0, 100) != 0) {
      binlog_->emulate_restart();
      return;
    }

    LOG(ERROR) << "RESTART BINLOG";
    binlog_ = TQueue::create();
    auto tqueue_binlog = make_unique<TQueueBinlog<Binlog>>();
    auto binlog = std::make_shared<Binlog>();
    binlog->init(binlog_path().str(), [&](const BinlogEvent &event) { tqueue_binlog->replay(event, *binlog_); })
        .ensure();
    tqueue_binlog->set_binlog(binlog);
    binlog_->set_callback(std::move(tqueue_binlog));
  }

  TQueue::EventId push(TQueue::QueueId queue_id, string data, double expire_at,
                       TQueue::EventId new_id = TQueue::EventId()) {
    auto a_id = baseline_->push(queue_id, data, expire_at, new_id).move_as_ok();
    auto b_id = memory_->push(queue_id, data, expire_at, new_id).move_as_ok();
    auto c_id = binlog_->push(queue_id, data, expire_at, new_id).move_as_ok();
    ASSERT_EQ(a_id, b_id);
    ASSERT_EQ(a_id, c_id);
    return a_id;
  }

  void check_head_tail(TQueue::QueueId qid) {
    ASSERT_EQ(baseline_->get_head(qid), memory_->get_head(qid));
    ASSERT_EQ(baseline_->get_head(qid), binlog_->get_head(qid));
    ASSERT_EQ(baseline_->get_tail(qid), memory_->get_tail(qid));
    ASSERT_EQ(baseline_->get_tail(qid), binlog_->get_tail(qid));
  }

  void check_get(TQueue::QueueId qid, Random::Xorshift128plus &rnd) {
    TQueue::Event a[10];
    MutableSpan<TQueue::Event> a_span(a, 10);
    TQueue::Event b[10];
    MutableSpan<TQueue::Event> b_span(b, 10);
    TQueue::Event c[10];
    MutableSpan<TQueue::Event> c_span(b, 10);

    auto a_from = baseline_->get_head(qid);
    auto b_from = memory_->get_head(qid);
    auto c_from = binlog_->get_head(qid);
    ASSERT_EQ(a_from, b_from);
    ASSERT_EQ(a_from, c_from);

    auto tmp = a_from.advance(rnd.fast(-10, 10));
    if (tmp.is_ok()) {
      a_from = tmp.move_as_ok();
    }
    baseline_->get(qid, a_from, 0, a_span).move_as_ok();
    memory_->get(qid, a_from, 0, b_span).move_as_ok();
    binlog_->get(qid, a_from, 0, c_span).move_as_ok();
    ASSERT_EQ(a_span.size(), b_span.size());
    ASSERT_EQ(a_span.size(), c_span.size());
    for (size_t i = 0; i < a_span.size(); i++) {
      ASSERT_EQ(a_span[i].id, b_span[i].id);
      ASSERT_EQ(a_span[i].id, c_span[i].id);
      ASSERT_EQ(a_span[i].data, b_span[i].data);
      ASSERT_EQ(a_span[i].data, c_span[i].data);
    }
  }

 private:
  unique_ptr<TQueue> baseline_;
  unique_ptr<TQueue> memory_;
  unique_ptr<TQueue> binlog_;
  MemoryStorage *memory_storage_{nullptr};
};

TEST(TQueue, random) {
  using EventId = TQueue::EventId;
  Random::Xorshift128plus rnd(123);
  auto next_qid = [&] {
    return rnd.fast(1, 10);
  };
  auto next_first_id = [&] {
    if (rnd.fast(0, 3) == 0) {
      return EventId::from_int32(EventId::MAX_ID - 20).move_as_ok();
    }
    return EventId::from_int32(rnd.fast(1000000000, 1500000000)).move_as_ok();
  };
  TestTQueue q;
  auto push_event = [&] {
    auto data = PSTRING() << rnd();
    q.push(next_qid(), data, 0, next_first_id());
  };
  auto check_head_tail = [&] {
    q.check_head_tail(next_qid());
  };
  auto restart = [&] {
    q.restart(rnd);
  };
  auto get = [&] {
    q.check_get(next_qid(), rnd);
  };
  RandomSteps steps({{push_event, 100}, {check_head_tail, 10}, {get, 40}, {restart, 1}});
  for (int i = 0; i < 1000000; i++) {
    steps.step(rnd);
  }
}

}  // namespace td