mirror of
https://github.com/Icinga/icinga2.git
synced 2026-02-03 20:40:17 -05:00
c0bf592ec9
There's a set of two tests for each perfdatawriter, just to make sure they can connect and send data that looks reasonably correct, and to make sure pausing actually works while the connection is stuck. Then there's a more in-depth suite of tests for PerfdataWriterConnection itself, to verify that connection handling works well in all types of scenarios.
81 lines
2.1 KiB
C++
81 lines
2.1 KiB
C++
/* Icinga 2 | (c) 2025 Icinga GmbH | GPLv2+ */
|
|
|
|
#include "utils.hpp"
|
|
#include <cstring>
|
|
#include <iomanip>
|
|
#include <sstream>
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <boost/random.hpp>
|
|
|
|
tm make_tm(std::string s)
|
|
{
|
|
int dst = -1;
|
|
size_t l = strlen("YYYY-MM-DD HH:MM:SS");
|
|
if (s.size() > l) {
|
|
std::string zone = s.substr(l);
|
|
if (zone == " PST") {
|
|
dst = 0;
|
|
} else if (zone == " PDT") {
|
|
dst = 1;
|
|
} else {
|
|
// tests should only use PST/PDT (for now)
|
|
BOOST_CHECK_MESSAGE(false, "invalid or unknown time time: " << zone);
|
|
}
|
|
}
|
|
|
|
std::tm t = {};
|
|
std::istringstream stream(s);
|
|
stream >> std::get_time(&t, "%Y-%m-%d %H:%M:%S");
|
|
t.tm_isdst = dst;
|
|
|
|
return t;
|
|
}
|
|
|
|
#ifndef _WIN32
|
|
const char *GlobalTimezoneFixture::TestTimezoneWithDST = "America/Los_Angeles";
|
|
#else /* _WIN32 */
|
|
// Tests are using pacific time because Windows only really supports timezones following US DST rules with the TZ
|
|
// environment variable. Format is "[Standard TZ][negative UTC offset][DST TZ]".
|
|
// https://docs.microsoft.com/en-us/cpp/c-runtime-library/reference/tzset?view=msvc-160#remarks
|
|
const char *GlobalTimezoneFixture::TestTimezoneWithDST = "PST8PDT";
|
|
#endif /* _WIN32 */
|
|
|
|
GlobalTimezoneFixture::GlobalTimezoneFixture(const char *fixed_tz)
|
|
{
|
|
tz = getenv("TZ");
|
|
#ifdef _WIN32
|
|
_putenv_s("TZ", fixed_tz == "" ? "UTC" : fixed_tz);
|
|
#else
|
|
setenv("TZ", fixed_tz, 1);
|
|
#endif
|
|
tzset();
|
|
}
|
|
|
|
GlobalTimezoneFixture::~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();
|
|
}
|
|
|
|
std::string GetRandomString(std::string prefix, std::size_t length)
|
|
{
|
|
std::string alphabet = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
|
|
boost::random::mt19937 generator;
|
|
boost::random::uniform_int_distribution<> distribution(0, alphabet.size() - 1);
|
|
|
|
for (auto i = 0U; i < length; i++) {
|
|
prefix += alphabet[distribution(generator)];
|
|
}
|
|
|
|
return prefix;
|
|
}
|