104-asio_timer.cpp 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. //
  2. // Copyright (c) 2019-2021 Ivan Baidakou (basiliscos) (the dot dmol at gmail dot com)
  3. //
  4. // Distributed under the MIT Software License
  5. //
  6. #include "catch.hpp"
  7. #include "rotor.hpp"
  8. #include "rotor/asio.hpp"
  9. #include "supervisor_asio_test.h"
  10. #include "access.h"
  11. namespace r = rotor;
  12. namespace ra = rotor::asio;
  13. namespace rt = r::test;
  14. namespace asio = boost::asio;
  15. namespace pt = boost::posix_time;
  16. struct sample_res_t {};
  17. struct sample_req_t {
  18. using response_t = sample_res_t;
  19. };
  20. using traits_t = r::request_traits_t<sample_req_t>;
  21. struct bad_actor_t : public r::actor_base_t {
  22. using r::actor_base_t::actor_base_t;
  23. r::extended_error_ptr_t ee;
  24. void configure(r::plugin::plugin_base_t &plugin) noexcept override {
  25. r::actor_base_t::configure(plugin);
  26. plugin.with_casted<r::plugin::starter_plugin_t>([](auto &p) { p.subscribe_actor(&bad_actor_t::on_response); });
  27. }
  28. void on_start() noexcept override {
  29. r::actor_base_t::on_start();
  30. start_timer(r::pt::milliseconds(1), *this, &bad_actor_t::delayed_start);
  31. start_timer(r::pt::minutes(1), *this, &bad_actor_t::delayed_start); // to be cancelled
  32. }
  33. void delayed_start(r::request_id_t, bool) noexcept {
  34. request<traits_t::request::type>(address).send(r::pt::milliseconds(1));
  35. }
  36. void on_response(traits_t::response::message_t &msg) noexcept {
  37. ee = msg.payload.ee;
  38. supervisor->do_shutdown();
  39. }
  40. };
  41. TEST_CASE("timer", "[supervisor][asio]") {
  42. asio::io_context io_context{1};
  43. auto timeout = r::pt::milliseconds{10};
  44. auto system_context = ra::system_context_asio_t::ptr_t{new ra::system_context_asio_t(io_context)};
  45. auto strand = std::make_shared<asio::io_context::strand>(io_context);
  46. auto sup = system_context->create_supervisor<rt::supervisor_asio_test_t>().strand(strand).timeout(timeout).finish();
  47. auto actor = sup->create_actor<bad_actor_t>().timeout(timeout).finish();
  48. sup->start();
  49. io_context.run();
  50. REQUIRE(actor->ee->ec == r::error_code_t::request_timeout);
  51. REQUIRE(static_cast<r::actor_base_t *>(sup.get())->access<rt::to::state>() == r::state_t::SHUT_DOWN);
  52. CHECK(rt::empty(sup->get_subscription()));
  53. REQUIRE(sup->get_state() == r::state_t::SHUT_DOWN);
  54. REQUIRE(sup->get_leader_queue().size() == 0);
  55. CHECK(rt::empty(sup->get_subscription()));
  56. }