mirror of https://github.com/Icinga/icinga2.git
279 lines
8.7 KiB
C++
279 lines
8.7 KiB
C++
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
|
|
|
|
#include "base/utility.hpp"
|
|
#include "icinga/legacytimeperiod.hpp"
|
|
#include <boost/date_time/posix_time/posix_time.hpp>
|
|
#include <boost/date_time/posix_time/ptime.hpp>
|
|
#include <boost/date_time/posix_time/posix_time_duration.hpp>
|
|
#include <boost/date_time/gregorian/conversion.hpp>
|
|
#include <boost/date_time/date.hpp>
|
|
#include <BoostTestTargetConfig.h>
|
|
|
|
using namespace icinga;
|
|
|
|
BOOST_AUTO_TEST_SUITE(icinga_legacytimeperiod);
|
|
|
|
struct GlobalTimezoneFixture
|
|
{
|
|
char *tz;
|
|
|
|
GlobalTimezoneFixture()
|
|
{
|
|
tz = getenv("TZ");
|
|
#ifdef _WIN32
|
|
_putenv_s("TZ", "UTC");
|
|
#else
|
|
setenv("TZ", "", 1);
|
|
#endif
|
|
tzset();
|
|
}
|
|
|
|
~GlobalTimezoneFixture()
|
|
{
|
|
#ifdef _WIN32
|
|
if (tz)
|
|
_putenv_s("TZ", tz);
|
|
else
|
|
_putenv_s("TZ", "");
|
|
#else
|
|
if (tz)
|
|
setenv("TZ", tz, 1);
|
|
else
|
|
unsetenv("TZ");
|
|
#endif
|
|
tzset();
|
|
}
|
|
};
|
|
|
|
BOOST_GLOBAL_FIXTURE(GlobalTimezoneFixture);
|
|
|
|
BOOST_AUTO_TEST_CASE(simple)
|
|
{
|
|
tm tm_beg, tm_end, tm_ref;
|
|
String timestamp;
|
|
boost::posix_time::ptime begin;
|
|
boost::posix_time::ptime end;
|
|
boost::posix_time::ptime expectedBegin;
|
|
boost::posix_time::ptime expectedEnd;
|
|
|
|
//-----------------------------------------------------
|
|
// check parsing of "YYYY-MM-DD" specs
|
|
timestamp = "2016-01-01";
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2016, 1, 1), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2016, 1, 2), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
// Run test
|
|
LegacyTimePeriod::ParseTimeSpec(timestamp, &tm_beg, &tm_end, &tm_ref);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
timestamp = "2015-12-31";
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2015, 12, 31), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2016, 1, 1), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
// Run test
|
|
LegacyTimePeriod::ParseTimeSpec(timestamp, &tm_beg, &tm_end, &tm_ref);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
// Break things forcefully
|
|
BOOST_CHECK_THROW(LegacyTimePeriod::ParseTimeSpec("2015-12-32", &tm_beg, &tm_end, &tm_ref),
|
|
std::invalid_argument);
|
|
|
|
BOOST_CHECK_THROW(LegacyTimePeriod::ParseTimeSpec("2015-28-01", &tm_beg, &tm_end, &tm_ref),
|
|
std::invalid_argument);
|
|
|
|
//-----------------------------------------------------
|
|
// check parsing of "day X" and "day -X" specs
|
|
timestamp = "day 2";
|
|
tm_ref.tm_year = 2016 - 1900;
|
|
tm_ref.tm_mon = 2 - 1;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2016, 2, 2), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2016, 2, 3), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
// Run Tests
|
|
LegacyTimePeriod::ParseTimeSpec(timestamp, &tm_beg, &tm_end, &tm_ref);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
timestamp = "day 31";
|
|
tm_ref.tm_year = 2018 - 1900;
|
|
tm_ref.tm_mon = 12 - 1;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2018, 12, 31), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2019, 1, 1), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
// Run Tests
|
|
LegacyTimePeriod::ParseTimeSpec(timestamp, &tm_beg, &tm_end, &tm_ref);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
// Last day of the month
|
|
timestamp = "day -1";
|
|
tm_ref.tm_year = 2012 - 1900;
|
|
tm_ref.tm_mon = 7 - 1;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2012, 7, 31), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2012, 8, 1), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
// Run Tests
|
|
LegacyTimePeriod::ParseTimeSpec(timestamp, &tm_beg, &tm_end, &tm_ref);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
// Third last day of the month
|
|
timestamp = "day -3";
|
|
tm_ref.tm_year = 2019 - 1900;
|
|
tm_ref.tm_mon = 7 - 1;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2019, 7, 29), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2019, 7, 30), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
// Run Tests
|
|
LegacyTimePeriod::ParseTimeSpec(timestamp, &tm_beg, &tm_end, &tm_ref);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
// Leap year with the last day of the month
|
|
timestamp = "day -1";
|
|
tm_ref.tm_year = 2016 - 1900; // leap year
|
|
tm_ref.tm_mon = 2 - 1;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2016, 2, 29), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2016, 3, 1), boost::posix_time::time_duration(0, 0, 0));
|
|
|
|
// Run Tests
|
|
LegacyTimePeriod::ParseTimeSpec("day -1", &tm_beg, &tm_end, &tm_ref);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(advanced)
|
|
{
|
|
tm tm_beg, tm_end, tm_ref;
|
|
String timestamp;
|
|
boost::posix_time::ptime begin;
|
|
boost::posix_time::ptime end;
|
|
boost::posix_time::ptime expectedBegin;
|
|
boost::posix_time::ptime expectedEnd;
|
|
|
|
//-----------------------------------------------------
|
|
// 2019-05-06 where Icinga celebrates 10 years #monitoringlove
|
|
// 2019-05-06 22:00:00 - 2019-05-07 06:00:00
|
|
timestamp = "22:00-06:00";
|
|
tm_ref.tm_year = 2019 - 1900;
|
|
tm_ref.tm_mon = 5 - 1;
|
|
tm_ref.tm_mday = 6;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2019, 5, 6), boost::posix_time::time_duration(22, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2019, 5, 7), boost::posix_time::time_duration(6, 0, 0));
|
|
|
|
// Run test
|
|
LegacyTimePeriod::ProcessTimeRangeRaw(timestamp, &tm_ref, &tm_beg, &tm_end);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
// 2019-05-06 Icinga is unleashed.
|
|
// 09:00:00 - 17:00:00
|
|
timestamp = "09:00-17:00";
|
|
tm_ref.tm_year = 2009 - 1900;
|
|
tm_ref.tm_mon = 5 - 1;
|
|
tm_ref.tm_mday = 6;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2009, 5, 6), boost::posix_time::time_duration(9, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2009, 5, 6), boost::posix_time::time_duration(17, 0, 0));
|
|
|
|
// Run test
|
|
LegacyTimePeriod::ProcessTimeRangeRaw(timestamp, &tm_ref, &tm_beg, &tm_end);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
|
|
//-----------------------------------------------------
|
|
// At our first Icinga Camp in SFO 2014 at GitHub HQ, we partied all night long with an overflow.
|
|
// 2014-09-24 09:00:00 - 2014-09-25 06:00:00
|
|
timestamp = "09:00-30:00";
|
|
tm_ref.tm_year = 2014 - 1900;
|
|
tm_ref.tm_mon = 9 - 1;
|
|
tm_ref.tm_mday = 24;
|
|
|
|
expectedBegin = boost::posix_time::ptime(boost::gregorian::date(2014, 9, 24), boost::posix_time::time_duration(9, 0, 0));
|
|
|
|
expectedEnd = boost::posix_time::ptime(boost::gregorian::date(2014, 9, 25), boost::posix_time::time_duration(6, 0, 0));
|
|
|
|
// Run test
|
|
LegacyTimePeriod::ProcessTimeRangeRaw(timestamp, &tm_ref, &tm_beg, &tm_end);
|
|
|
|
// Compare times
|
|
begin = boost::posix_time::ptime_from_tm(tm_beg);
|
|
end = boost::posix_time::ptime_from_tm(tm_end);
|
|
|
|
BOOST_CHECK_EQUAL(begin, expectedBegin);
|
|
BOOST_CHECK_EQUAL(end, expectedEnd);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|