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 AfterConnectedNotifier : public STG::PropertyNotifierBase<bool>
  28 {
  29     public:
  30         AfterConnectedNotifier()
  31             : m_connects(0),
  32               m_disconnects(0)
  33         {}
  34
  35         void notify(const bool& oldValue, const bool& newValue) override
  36         {
  37             if (!oldValue && newValue)
  38                 ++m_connects;
  39             if (oldValue && !newValue)
  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     AfterConnectedNotifier connectionNotifier;
  81
  82     user.AddConnectedAfterNotifier(&connectionNotifier);
  83
  84     STG::UserProperty<std::string> & tariffName = user.GetProperties().tariffName;
  85     STG::UserProperty<STG::UserIPs> & ips = user.GetProperties().ips;
  86
  87     ips = STG::UserIPs::parse("*");
  88
  89     BOOST_CHECK_EQUAL(user.GetConnected(), false);
  90     BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(0));
  91     BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
  92
  93     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME);
  94
  95     user.Authorize(inet_strington("127.0.0.1"), 0, &auth);
  96     user.Run();
  97
  98     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
  99
 100     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 101     BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(1));
 102     BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
 103
 104     tariffName = "test";
 105     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), "test");
 106
 107     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
 108
 109     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 110     BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(1));
 111     BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
 112 }
 113
 114 BOOST_AUTO_TEST_CASE(Reconnect)
 115 {
 116     Settings settings(true);
 117
 118     TestSettings * s1 = &settings;
 119     STG::Settings * s2 = &settings;
 120
 121     BOOST_CHECK(settings.GetReconnectOnTariffChange());
 122     BOOST_CHECK(s1->GetReconnectOnTariffChange());
 123     BOOST_CHECK(s2->GetReconnectOnTariffChange());
 124
 125     TestTariffs tariffs;
 126     STG::Admin admin(STG::Priv(0xFFFF), {}, {});
 127     TestStore store;
 128     TestAuth auth;
 129     TestUsers users;
 130     TestServices services;
 131     STG::UserImpl user(&settings, &store, &tariffs, &admin, &users, services);
 132
 133     AfterConnectedNotifier connectionNotifier;
 134
 135     user.AddConnectedAfterNotifier(&connectionNotifier);
 136
 137     STG::UserProperty<std::string> & tariffName = user.GetProperties().tariffName;
 138     STG::UserProperty<STG::UserIPs> & ips = user.GetProperties().ips;
 139
 140     ips = STG::UserIPs::parse("*");
 141
 142     BOOST_CHECK_EQUAL(user.GetConnected(), false);
 143     BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(0));
 144     BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
 145
 146     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME);
 147
 148     user.Authorize(inet_strington("127.0.0.1"), 0, &auth);
 149     user.Run();
 150
 151     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
 152
 153     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 154     BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(1));
 155     BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
 156
 157     tariffName = "test";
 158     BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), "test");
 159
 160     BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
 161
 162     BOOST_CHECK_EQUAL(user.GetConnected(), true);
 163     BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(2));
 164     BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(1));
 165 }
 166
 167 BOOST_AUTO_TEST_SUITE_END()