102-asio_ping-pong-2-strands.cpp 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. //
  2. // Copyright (c) 2019-2020 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 "actor_test.h"
  11. #include "access.h"
  12. namespace r = rotor;
  13. namespace ra = rotor::asio;
  14. namespace rt = r::test;
  15. namespace asio = boost::asio;
  16. namespace pt = boost::posix_time;
  17. struct ping_t {};
  18. struct pong_t {};
  19. struct pinger_t : public rt::actor_test_t {
  20. std::uint32_t ping_sent = 0;
  21. std::uint32_t pong_received = 0;
  22. using rt::actor_test_t::actor_test_t;
  23. void set_ponger_addr(const r::address_ptr_t &addr) { ponger_addr = addr; }
  24. void configure(r::plugin::plugin_base_t &plugin) noexcept override {
  25. plugin.with_casted<r::plugin::starter_plugin_t>([&](auto &p) { p.subscribe_actor(&pinger_t::on_pong); });
  26. plugin.with_casted<r::plugin::link_client_plugin_t>(
  27. [&](auto &p) { p.link(ponger_addr, true, [&](auto &ec) mutable { REQUIRE(!ec); }); });
  28. }
  29. void on_start() noexcept override {
  30. r::actor_base_t::on_start();
  31. do_send_ping();
  32. }
  33. void on_pong(r::message_t<pong_t> &) noexcept {
  34. ++pong_received;
  35. supervisor->shutdown();
  36. }
  37. void do_send_ping() {
  38. ++ping_sent;
  39. send<ping_t>(ponger_addr);
  40. }
  41. r::address_ptr_t ponger_addr;
  42. };
  43. struct ponger_t : public rt::actor_test_t {
  44. std::uint32_t ping_received = 0;
  45. std::uint32_t pong_sent = 0;
  46. using rt::actor_test_t::actor_test_t;
  47. void set_pinger_addr(const r::address_ptr_t &addr) { pinger_addr = addr; }
  48. void configure(r::plugin::plugin_base_t &plugin) noexcept override {
  49. plugin.with_casted<r::plugin::starter_plugin_t>([](auto &p) { p.subscribe_actor(&ponger_t::on_ping); });
  50. }
  51. void on_ping(rotor::message_t<ping_t> &) noexcept {
  52. ++ping_received;
  53. send<pong_t>(pinger_addr);
  54. ++pong_sent;
  55. }
  56. private:
  57. r::address_ptr_t pinger_addr;
  58. };
  59. TEST_CASE("ping/pong 2 sups", "[supervisor][asio]") {
  60. asio::io_context io_context{1};
  61. auto system_context = ra::system_context_asio_t::ptr_t{new ra::system_context_asio_t(io_context)};
  62. auto strand1 = std::make_shared<asio::io_context::strand>(io_context);
  63. auto strand2 = std::make_shared<asio::io_context::strand>(io_context);
  64. auto timeout = r::pt::milliseconds{10};
  65. auto sup1 =
  66. system_context->create_supervisor<rt::supervisor_asio_test_t>().strand(strand1).timeout(timeout).finish();
  67. auto sup2 = sup1->create_actor<rt::supervisor_asio_test_t>().strand(strand2).timeout(timeout).finish();
  68. auto pinger = sup1->create_actor<pinger_t>().timeout(timeout).finish();
  69. auto ponger = sup2->create_actor<ponger_t>().timeout(timeout).finish();
  70. pinger->set_ponger_addr(static_cast<r::actor_base_t *>(ponger.get())->get_address());
  71. ponger->set_pinger_addr(static_cast<r::actor_base_t *>(pinger.get())->get_address());
  72. sup1->start();
  73. io_context.run();
  74. REQUIRE(pinger->ping_sent == 1);
  75. REQUIRE(pinger->pong_received == 1);
  76. REQUIRE(ponger->ping_received == 1);
  77. REQUIRE(ponger->pong_sent == 1);
  78. REQUIRE(static_cast<r::actor_base_t *>(sup1.get())->access<rt::to::state>() == r::state_t::SHUT_DOWN);
  79. REQUIRE(sup1->get_leader_queue().size() == 0);
  80. CHECK(rt::empty(sup1->get_subscription()));
  81. REQUIRE(static_cast<r::actor_base_t *>(sup2.get())->access<rt::to::state>() == r::state_t::SHUT_DOWN);
  82. REQUIRE(sup2->get_leader_queue().size() == 0);
  83. CHECK(rt::empty(sup2->get_subscription()));
  84. REQUIRE(pinger->get_state() == r::state_t::SHUT_DOWN);
  85. REQUIRE(ponger->get_state() == r::state_t::SHUT_DOWN);
  86. }