tests/test_reconnect_on_tariff_change.cpp

   1 #define BOOST_TEST_MODULE STGReconnectOnTariffChange
   2
   3 #include "stg/admin.h"
   4 #include "stg/user_property.h"
   5 #include "user_impl.h"
   6
   7 #include "testsettings.h"
   8 #include "testtariffs.h"
   9 #include "teststore.h"
  10 #include "testauth.h"
  11 #include "testusers.h"
  12 #include "testservices.h"
  13
  14 #pragma GCC diagnostic push
  15 #pragma GCC diagnostic ignored "-Wold-style-cast"
  16 #pragma GCC diagnostic ignored "-Wunused-parameter"
  17 #pragma GCC diagnostic ignored "-Wsign-compare"
  18 #pragma GCC diagnostic ignored "-Wparentheses"
  19 #include <boost/test/unit_test.hpp>
  20 #pragma GCC diagnostic pop
  21
  22 volatile time_t stgTime = 0;
  23
  24 namespace
  25 {
  26
  27 class ConnectCtr
  28 {
  29     public:
  30         ConnectCtr()
  31             : m_connects(0),
  32               m_disconnects(0)
  33         {}
  34
  35         void update(bool isConnect)
  36         {
  37             if (isConnect)
  38                 ++m_connects;
  39             else
  40                 ++m_disconnects;
  41         }
  42
  43         size_t connects() const { return m_connects; }
  44         size_t disconnects() const { return m_disconnects; }
  45
  46     private:
  47         size_t m_connects;
  48         size_t m_disconnects;
  49 };
  50
  51 class Settings : public TestSettings
  52 {
  53     public:
  54         Settings(bool reconnectOnTariffChange)
  55             : m_reconnectOnTariffChange(reconnectOnTariffChange)
  56         {}
  57
  58         bool GetReconnectOnTariffChange() const { return m_reconnectOnTariffChange; }
  59
  60     private:
  61         bool m_reconnectOnTariffChange;
  62 };
  63
  64 }
  65
  66 BOOST_AUTO_TEST_SUITE(ReconnectOnTariffChange)
  67
  68 BOOST_AUTO_TEST_CASE(NormalBehavior)
  69 {
  70     Settings settings(false);
  71     TestTariffs tariffs;
  72     tariffs.ReadTariffs();
  73     STG::Admin admin(STG::Priv(0xFFFF), {}, {});
  74     TestStore store;
  75     TestAuth auth;
  76     TestUsers users;
  77     TestServices services;
  78     STG::UserImpl user(&settings, &store, &tariffs, &admin, &users, services);
  79
  80     ConnectCtr ctr;
  81     STG::ScopedConnection conn = user.afterConnectedChange([&ctr](auto, auto newVal){ ctr.update(newVal); });
  82
  83     STG::UserProperty<std::string> & tariffName = user.GetProperties().tariffName;
  84     STG::UserProperty<STG::UserIPs> & ips = user.GetProperties().ips;
  85
  86     ips = STG::UserIPs::parse("*");
  87
  88     BOOST_CHECK_EQUAL(user.GetConnected(), false);
  89     BOOST_CHECK_EQUAL(ctr.connects(), static_cast<size_t>(0));
  90     BOOST_CHECK_EQUAL(ctr.disconnects(), static_cast<size_t>(0));
  91
  92     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME);
  93
  94     user.Authorize(inet_strington("127.0.0.1"), 0, &auth);
  95     user.Run();
  96
  97     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
  98
  99     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 100     BOOST_CHECK_EQUAL(ctr.connects(), static_cast<size_t>(1));
 101     BOOST_CHECK_EQUAL(ctr.disconnects(), static_cast<size_t>(0));
 102
 103     tariffName = "test";
 104     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), "test");
 105
 106     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
 107
 108     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 109     BOOST_CHECK_EQUAL(ctr.connects(), static_cast<size_t>(1));
 110     BOOST_CHECK_EQUAL(ctr.disconnects(), static_cast<size_t>(0));
 111 }
 112
 113 BOOST_AUTO_TEST_CASE(Reconnect)
 114 {
 115     Settings settings(true);
 116
 117     TestSettings * s1 = &settings;
 118     STG::Settings * s2 = &settings;
 119
 120     BOOST_CHECK(settings.GetReconnectOnTariffChange());
 121     BOOST_CHECK(s1->GetReconnectOnTariffChange());
 122     BOOST_CHECK(s2->GetReconnectOnTariffChange());
 123
 124     TestTariffs tariffs;
 125     STG::Admin admin(STG::Priv(0xFFFF), {}, {});
 126     TestStore store;
 127     TestAuth auth;
 128     TestUsers users;
 129     TestServices services;
 130     STG::UserImpl user(&settings, &store, &tariffs, &admin, &users, services);
 131
 132     ConnectCtr ctr;
 133     STG::ScopedConnection conn = user.afterConnectedChange([&ctr](auto, auto newVal){ ctr.update(newVal); });
 134
 135     STG::UserProperty<std::string> & tariffName = user.GetProperties().tariffName;
 136     STG::UserProperty<STG::UserIPs> & ips = user.GetProperties().ips;
 137
 138     ips = STG::UserIPs::parse("*");
 139
 140     BOOST_CHECK_EQUAL(user.GetConnected(), false);
 141     BOOST_CHECK_EQUAL(ctr.connects(), static_cast<size_t>(0));
 142     BOOST_CHECK_EQUAL(ctr.disconnects(), static_cast<size_t>(0));
 143
 144     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME);
 145
 146     user.Authorize(inet_strington("127.0.0.1"), 0, &auth);
 147     user.Run();
 148
 149     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
 150
 151     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 152     BOOST_CHECK_EQUAL(ctr.connects(), static_cast<size_t>(1));
 153     BOOST_CHECK_EQUAL(ctr.disconnects(), static_cast<size_t>(0));
 154
 155     tariffName = "test";
 156     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), "test");
 157
 158     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
 159
 160     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 161     BOOST_CHECK_EQUAL(ctr.connects(), static_cast<size_t>(2));
 162     BOOST_CHECK_EQUAL(ctr.disconnects(), static_cast<size_t>(1));
 163 }
 164
 165 BOOST_AUTO_TEST_SUITE_END()