132-ev_timer.cpp 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  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/ev.hpp"
  9. #include <ev.h>
  10. #include "access.h"
  11. namespace r = rotor;
  12. namespace re = rotor::ev;
  13. namespace pt = boost::posix_time;
  14. namespace rt = r::test;
  15. struct sample_res_t {};
  16. struct sample_req_t {
  17. using response_t = sample_res_t;
  18. };
  19. using traits_t = r::request_traits_t<sample_req_t>;
  20. struct bad_actor_t : public r::actor_base_t {
  21. using r::actor_base_t::actor_base_t;
  22. r::extended_error_ptr_t ee;
  23. void configure(r::plugin::plugin_base_t &plugin) noexcept override {
  24. r::actor_base_t::configure(plugin);
  25. plugin.with_casted<r::plugin::starter_plugin_t>([](auto &p) { p.subscribe_actor(&bad_actor_t::on_response); });
  26. }
  27. void on_start() noexcept override {
  28. r::actor_base_t::on_start();
  29. start_timer(r::pt::milliseconds(1), *this, &bad_actor_t::delayed_start);
  30. start_timer(r::pt::minutes(1), *this, &bad_actor_t::delayed_start); // to be cancelled
  31. }
  32. void delayed_start(r::request_id_t, bool) noexcept {
  33. request<traits_t::request::type>(address).send(r::pt::milliseconds(1));
  34. }
  35. void on_response(traits_t::response::message_t &msg) noexcept {
  36. ee = msg.payload.ee;
  37. supervisor->do_shutdown();
  38. }
  39. };
  40. TEST_CASE("timer", "[supervisor][ev]") {
  41. auto *loop = ev_loop_new(0);
  42. auto system_context = r::intrusive_ptr_t<re::system_context_ev_t>{new re::system_context_ev_t()};
  43. auto timeout = r::pt::milliseconds{10};
  44. auto sup = system_context->create_supervisor<re::supervisor_ev_t>()
  45. .loop(loop)
  46. .timeout(timeout)
  47. .loop_ownership(true)
  48. .finish();
  49. auto actor = sup->create_actor<bad_actor_t>().timeout(timeout).finish();
  50. sup->start();
  51. ev_run(loop);
  52. REQUIRE(actor->ee->ec == r::error_code_t::request_timeout);
  53. REQUIRE(static_cast<r::actor_base_t *>(sup.get())->access<rt::to::state>() == r::state_t::SHUT_DOWN);
  54. }