#include "tut/tut.hpp" #include "stg/admin.h" #include "stg/user_property.h" #include "user_impl.h" #include "testsettings.h" #include "testtariffs.h" #include "teststore.h" #include "testauth.h" #include "testusers.h" #include "testservices.h" namespace { class AFTER_CONNECTED_NOTIFIER : public STG::PropertyNotifierBase { 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; }; class TEST_SETTINGS_LOCAL : public TEST_SETTINGS { public: TEST_SETTINGS_LOCAL(bool _reconnectOnTariffChange) : TEST_SETTINGS(), reconnectOnTariffChange(_reconnectOnTariffChange) {} bool GetReconnectOnTariffChange() const { return reconnectOnTariffChange; } private: bool reconnectOnTariffChange; }; } namespace tut { struct reconnect_on_tariff_change_data { }; typedef test_group tg; tg reconnect_on_tariff_change_test_group("Reconnect on tariff change tests group"); typedef tg::object testobject; template<> template<> void testobject::test<1>() { set_test_name("Check normal behaviour"); TEST_SETTINGS_LOCAL settings(false); TEST_TARIFFS tariffs; STG::Admin admin(STG::Priv(0xFFFF), {}, {}); TEST_STORE store; TEST_AUTH auth; TEST_USERS users; TEST_SERVICES services; STG::UserImpl user(&settings, &store, &tariffs, &admin, &users, services); AFTER_CONNECTED_NOTIFIER connectionNotifier; user.AddConnectedAfterNotifier(&connectionNotifier); STG::UserProperty & tariffName = user.GetProperties().tariffName; STG::UserProperty & ips = user.GetProperties().ips; ips = STG::UserIPs::parse("*"); ensure_equals("user.connected = false", user.GetConnected(), false); ensure_equals("connects = 0", connectionNotifier.GetConnects(), static_cast(0)); ensure_equals("disconnects = 0", connectionNotifier.GetDisconnects(), static_cast(0)); ensure_equals("user.tariffName == NO_TARIFF_NAME", user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME); user.Authorize(inet_strington("127.0.0.1"), 0, &auth); user.Run(); ensure_equals("user.authorised_by = true", user.IsAuthorizedBy(&auth), true); ensure_equals("user.connected = true", user.GetConnected(), true); ensure_equals("connects = 1", connectionNotifier.GetConnects(), static_cast(1)); ensure_equals("disconnects = 0", connectionNotifier.GetDisconnects(), static_cast(0)); tariffName = "test"; ensure_equals("user.tariffName == 'test'", user.GetProperties().tariffName.ConstData(), "test"); ensure_equals("user.authorised_by = true", user.IsAuthorizedBy(&auth), true); ensure_equals("user.connected = true", user.GetConnected(), true); ensure_equals("connects = 1", connectionNotifier.GetConnects(), static_cast(1)); ensure_equals("disconnects = 0", connectionNotifier.GetDisconnects(), static_cast(0)); } template<> template<> void testobject::test<2>() { set_test_name("Check reconnect on tariff change"); TEST_SETTINGS_LOCAL settings(true); TEST_SETTINGS * s1 = &settings; STG::Settings * s2 = &settings; ensure("settings.GetReconnectOnTariffChange() == true", settings.GetReconnectOnTariffChange()); ensure("s1->GetReconnectOnTariffChange() == true", s1->GetReconnectOnTariffChange()); ensure("s2->GetReconnectOnTariffChange() == true", s2->GetReconnectOnTariffChange()); TEST_TARIFFS tariffs; STG::Admin admin(STG::Priv(0xFFFF), {}, {}); TEST_STORE store; TEST_AUTH auth; TEST_USERS users; TEST_SERVICES services; STG::UserImpl user(&settings, &store, &tariffs, &admin, &users, services); AFTER_CONNECTED_NOTIFIER connectionNotifier; user.AddConnectedAfterNotifier(&connectionNotifier); STG::UserProperty & tariffName = user.GetProperties().tariffName; STG::UserProperty & ips = user.GetProperties().ips; ips = STG::UserIPs::parse("*"); ensure_equals("user.connected = false", user.GetConnected(), false); ensure_equals("connects = 0", connectionNotifier.GetConnects(), static_cast(0)); ensure_equals("disconnects = 0", connectionNotifier.GetDisconnects(), static_cast(0)); ensure_equals("user.tariffName == NO_TARIFF_NAME", user.GetProperties().tariffName.ConstData(), NO_TARIFF_NAME); user.Authorize(inet_strington("127.0.0.1"), 0, &auth); user.Run(); ensure_equals("user.authorised_by = true", user.IsAuthorizedBy(&auth), true); ensure_equals("user.connected = true", user.GetConnected(), true); ensure_equals("connects = 1", connectionNotifier.GetConnects(), static_cast(1)); ensure_equals("disconnects = 0", connectionNotifier.GetDisconnects(), static_cast(0)); tariffName = "test"; ensure_equals("user.tariffName == 'test'", user.GetProperties().tariffName.ConstData(), "test"); ensure_equals("user.authorised_by = true", user.IsAuthorizedBy(&auth), true); ensure_equals("user.connected = true", user.GetConnected(), true); ensure_equals("connects = 2", connectionNotifier.GetConnects(), static_cast(2)); ensure_equals("disconnects = 1", connectionNotifier.GetDisconnects(), static_cast(1)); } } void AFTER_CONNECTED_NOTIFIER::Notify(const bool & oldValue, const bool & newValue) { if (!oldValue && newValue) ++connects; if (oldValue && !newValue) ++disconnects; }