// This file is part of CAF, the C++ Actor Framework. See the file LICENSE in
// the main distribution directory for license terms and copyright or visit
// https://github.com/actor-framework/actor-framework/blob/master/LICENSE.

#define CAF_SUITE simple_timeout

#include "caf/after.hpp"
#include "caf/all.hpp"

#include "core-test.hpp"

#include <chrono>
#include <memory>

using namespace caf;

namespace {

using ms = std::chrono::milliseconds;

using timer = typed_actor<result<void>(reset_atom)>;

struct timer_state {
  bool had_reset = false;
};

timer::behavior_type timer_impl(timer::stateful_pointer<timer_state> self) {
  self->delayed_send(self, ms(100), reset_atom_v);
  return {
    [=](reset_atom) {
      MESSAGE("timer reset");
      self->state.had_reset = true;
    },
    after(ms(600)) >>
      [=] {
        MESSAGE("timer expired");
        CAF_REQUIRE(self->state.had_reset);
        self->quit();
      },
  };
}

timer::behavior_type timer_impl2(timer::pointer self) {
  auto had_reset = std::make_shared<bool>(false);
  delayed_anon_send(self, ms(100), reset_atom_v);
  return {
    [=](reset_atom) {
      MESSAGE("timer reset");
      *had_reset = true;
    },
    after(ms(600)) >>
      [=] {
        MESSAGE("timer expired");
        CAF_REQUIRE(*had_reset);
        self->quit();
      },
  };
}

} // namespace

BEGIN_FIXTURE_SCOPE(test_coordinator_fixture<>)

CAF_TEST(single_timeout) {
  sys.spawn(timer_impl);
}

CAF_TEST(single_anon_timeout) {
  sys.spawn(timer_impl2);
}

END_FIXTURE_SCOPE()
