-#include "tut/tut.hpp"
+#define BOOST_TEST_MODULE STGReconnectOnTariffChange
+#include "stg/admin.h"
#include "stg/user_property.h"
#include "user_impl.h"
#include "testsettings.h"
#include "testtariffs.h"
-#include "testadmin.h"
#include "teststore.h"
#include "testauth.h"
+#include "testusers.h"
+#include "testservices.h"
-class AFTER_CONNECTED_NOTIFIER : public PROPERTY_NOTIFIER_BASE<bool>,
- private NONCOPYABLE {
-public:
- AFTER_CONNECTED_NOTIFIER()
- : connects(0),
- disconnects(0)
- {}
- void Notify(const bool & oldValue, const bool & newValue);
-
- size_t GetConnects() const { return connects; }
- size_t GetDisconnects() const { return disconnects; }
-
-private:
- size_t connects;
- size_t disconnects;
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wold-style-cast"
+#pragma GCC diagnostic ignored "-Wunused-parameter"
+#pragma GCC diagnostic ignored "-Wsign-compare"
+#pragma GCC diagnostic ignored "-Wparentheses"
+#include <boost/test/unit_test.hpp>
+#pragma GCC diagnostic pop
+
+volatile time_t stgTime = 0;
+
+namespace
+{
+
+class AfterConnectedNotifier : public STG::PropertyNotifierBase<bool>
+{
+ public:
+ AfterConnectedNotifier()
+ : m_connects(0),
+ m_disconnects(0)
+ {}
+
+ void Notify(const bool& oldValue, const bool& newValue) override
+ {
+ if (!oldValue && newValue)
+ ++m_connects;
+ if (oldValue && !newValue)
+ ++m_disconnects;
+ }
+
+ size_t connects() const { return m_connects; }
+ size_t disconnects() const { return m_disconnects; }
+
+ private:
+ size_t m_connects;
+ size_t m_disconnects;
};
-namespace tut
+class Settings : public TestSettings
{
- struct reconnect_on_tariff_change_data {
- };
+ public:
+ Settings(bool reconnectOnTariffChange)
+ : m_reconnectOnTariffChange(reconnectOnTariffChange)
+ {}
- typedef test_group<reconnect_on_tariff_change_data> tg;
- tg reconnect_on_tariff_change_test_group("Reconnect on tariff change tests group");
+ bool GetReconnectOnTariffChange() const { return m_reconnectOnTariffChange; }
- typedef tg::object testobject;
+ private:
+ bool m_reconnectOnTariffChange;
+};
- class TEST_SETTINGS_LOCAL : public TEST_SETTINGS {
- public:
- TEST_SETTINGS_LOCAL(bool _reconnectOnTariffChange)
- : reconnectOnTariffChange(_reconnectOnTariffChange)
- {}
+}
- bool GetReconnectOnTariffChange() const { return reconnectOnTariffChange; }
+BOOST_AUTO_TEST_SUITE(ReconnectOnTariffChange)
- private:
- bool reconnectOnTariffChange;
- };
+BOOST_AUTO_TEST_CASE(NormalBehavior)
+{
+ Settings settings(false);
+ TestTariffs tariffs;
+ tariffs.ReadTariffs();
+ STG::Admin admin(STG::Priv(0xFFFF), {}, {});
+ TestStore store;
+ TestAuth auth;
+ TestUsers users;
+ TestServices services;
+ STG::UserImpl user(&settings, &store, &tariffs, &admin, &users, services);
+
+ AfterConnectedNotifier connectionNotifier;
- template<>
- template<>
- void testobject::test<1>()
- {
- set_test_name("Check normal behaviour");
+ user.AddConnectedAfterNotifier(&connectionNotifier);
- TEST_SETTINGS_LOCAL settings(false);
- TEST_TARIFFS tariffs;
- TEST_ADMIN admin;
- TEST_STORE store;
- TEST_AUTH auth;
- USER_IMPL user(&settings, &store, &tariffs, &admin, NULL);
+ STG::UserProperty<std::string> & tariffName = user.GetProperties().tariffName;
+ STG::UserProperty<STG::UserIPs> & ips = user.GetProperties().ips;
- AFTER_CONNECTED_NOTIFIER connectionNotifier;
+ ips = STG::UserIPs::parse("*");
- user.AddConnectedAfterNotifier(&connectionNotifier);
+ BOOST_CHECK_EQUAL(user.GetConnected(), false);
+ BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(0));
+ BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
- USER_PROPERTY<double> & cash(user.GetProperty().cash);
- USER_PROPERTY<std::string> & tariffName(user.GetProperty().tariffName);
- USER_PROPERTY<USER_IPS> & ips(user.GetProperty().ips);
+ BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME);
- ips = StrToIPS("*");
+ user.Authorize(inet_strington("127.0.0.1"), 0, &auth);
+ user.Run();
- ensure_equals("user.connected = false", user.GetConnected(), false);
- ensure_equals("connects = 0", connectionNotifier.GetConnects(), 0);
- ensure_equals("disconnects = 0", connectionNotifier.GetDisconnects(), 0);
+ BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
- ensure_equals("user.tariffName == NO_TARIFF_NAME", user.GetProperty().tariffName.ConstData(), NO_TARIFF_NAME);
+ BOOST_CHECK_EQUAL(user.GetConnected(), true);
+ BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(1));
+ BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
- tariffName = "test";
- ensure_equals("user.tariffName == 'test'", user.GetProperty().tariffName.ConstData(), "test");
+ tariffName = "test";
+ BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), "test");
- user.Authorize(inet_strington("127.0.0.1"), 0, &auth);
+ BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
- ensure_equals("user.connected = true", user.GetConnected(), true);
- ensure_equals("connects = 1", connectionNotifier.GetConnects(), 1);
- ensure_equals("disconnects = 0", connectionNotifier.GetDisconnects(), 0);
- }
+ BOOST_CHECK_EQUAL(user.GetConnected(), true);
+ BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(1));
+ BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
}
-void AFTER_CONNECTED_NOTIFIER::Notify(const bool & oldValue, const bool & newValue)
+BOOST_AUTO_TEST_CASE(Reconnect)
{
- if (!oldValue && newValue)
- ++connects;
- if (oldValue && !newValue)
- ++disconnects;
+ Settings settings(true);
+
+ TestSettings * s1 = &settings;
+ STG::Settings * s2 = &settings;
+
+ BOOST_CHECK(settings.GetReconnectOnTariffChange());
+ BOOST_CHECK(s1->GetReconnectOnTariffChange());
+ BOOST_CHECK(s2->GetReconnectOnTariffChange());
+
+ TestTariffs tariffs;
+ STG::Admin admin(STG::Priv(0xFFFF), {}, {});
+ TestStore store;
+ TestAuth auth;
+ TestUsers users;
+ TestServices services;
+ STG::UserImpl user(&settings, &store, &tariffs, &admin, &users, services);
+
+ AfterConnectedNotifier connectionNotifier;
+
+ user.AddConnectedAfterNotifier(&connectionNotifier);
+
+ STG::UserProperty<std::string> & tariffName = user.GetProperties().tariffName;
+ STG::UserProperty<STG::UserIPs> & ips = user.GetProperties().ips;
+
+ ips = STG::UserIPs::parse("*");
+
+ BOOST_CHECK_EQUAL(user.GetConnected(), false);
+ BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(0));
+ BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
+
+ BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME);
+
+ user.Authorize(inet_strington("127.0.0.1"), 0, &auth);
+ user.Run();
+
+ BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
+
+ BOOST_CHECK_EQUAL(user.GetConnected(), true);
+ BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(1));
+ BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(0));
+
+ tariffName = "test";
+ BOOST_CHECK_EQUAL(user.GetProperties().tariffName.ConstData(), "test");
+
+ BOOST_CHECK_EQUAL(user.IsAuthorizedBy(&auth), true);
+
+ BOOST_CHECK_EQUAL(user.GetConnected(), true);
+ BOOST_CHECK_EQUAL(connectionNotifier.connects(), static_cast<size_t>(2));
+ BOOST_CHECK_EQUAL(connectionNotifier.disconnects(), static_cast<size_t>(1));
}
+
+BOOST_AUTO_TEST_SUITE_END()
git.stg.codes / stg.git / blobdiff