]> git.stg.codes - stg.git/blob - tests/test_reconnect_on_tariff_change.cpp
ca03d6c2f44be5abdcc8c394cc6d7cea51a30e86
[stg.git] / 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()