mirror of
https://github.com/Icinga/icinga2.git
synced 2026-04-07 10:05:25 -04:00
58 lines
1.6 KiB
C++
58 lines
1.6 KiB
C++
|
// SPDX-FileCopyrightText: 2026 Icinga GmbH <https://icinga.com>
|
||
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
|
|
||
|
|
#include <BoostTestTargetConfig.h>
|
||
|
|
#include "perfdata/elasticsearchwriter.hpp"
|
||
|
|
#include "test/base-testloggerfixture.hpp"
|
||
|
|
#include "test/perfdata-perfdatawriterfixture.hpp"
|
||
|
|
#include "test/utils.hpp"
|
||
|
|
|
||
|
|
using namespace icinga;
|
||
|
|
|
||
|
|
BOOST_FIXTURE_TEST_SUITE(perfdata_elasticsearchwriter, PerfdataWriterFixture<ElasticsearchWriter>,
|
||
|
|
*boost::unit_test::label("perfdata")
|
||
|
|
*boost::unit_test::label("network")
|
||
|
|
)
|
||
|
|
|
||
|
|
BOOST_AUTO_TEST_CASE(connect)
|
||
|
|
{
|
||
|
|
ResumeWriter();
|
||
|
|
|
||
|
|
ReceiveCheckResults(1, ServiceState::ServiceCritical);
|
||
|
|
|
||
|
|
Accept();
|
||
|
|
auto resp = GetSplitDecodedRequestBody();
|
||
|
|
SendResponse();
|
||
|
|
|
||
|
|
// ElasticsearchWriter wants to send the same message twice, once for the check result
|
||
|
|
// and once for the "state change".
|
||
|
|
resp = GetSplitDecodedRequestBody();
|
||
|
|
SendResponse();
|
||
|
|
|
||
|
|
// Just some basic sanity tests. It's not important to check if everything is entirely
|
||
|
|
// correct here.
|
||
|
|
BOOST_REQUIRE_GT(resp->GetLength(), 1);
|
||
|
|
Dictionary::Ptr cr = resp->Get(1);
|
||
|
|
BOOST_CHECK(cr->Contains("@timestamp"));
|
||
|
|
BOOST_CHECK_EQUAL(cr->Get("check_command"), "dummy");
|
||
|
|
BOOST_CHECK_EQUAL(cr->Get("host"), "h1");
|
||
|
|
|
||
|
|
PauseWriter();
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOST_AUTO_TEST_CASE(pause_with_pending_work)
|
||
|
|
{
|
||
|
|
ResumeWriter();
|
||
|
|
|
||
|
|
// Process check-results until the writer is stuck.
|
||
|
|
BOOST_REQUIRE_MESSAGE(GetWriterStuck(10s), "Failed to get Writer stuck.");
|
||
|
|
|
||
|
|
// Now try to pause.
|
||
|
|
PauseWriter();
|
||
|
|
|
||
|
|
REQUIRE_LOG_MESSAGE("Connection stopped\\.", 10s);
|
||
|
|
REQUIRE_LOG_MESSAGE("'ElasticsearchWriter' paused\\.", 10s);
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOST_AUTO_TEST_SUITE_END()
|