1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
#include "livestatus/livestatusquery.hpp"
#include "base/application.hpp"
#include "base/stdiostream.hpp"
#include "base/json.hpp"
#include <BoostTestTargetConfig.h>
using namespace icinga;
String LivestatusQueryHelper(const std::vector<String>& lines)
{
LivestatusQuery::Ptr query = new LivestatusQuery(lines, "");
std::stringstream stream;
StdioStream::Ptr sstream = new StdioStream(&stream, false);
query->Execute(new StoppableWaitGroup(), sstream);
String output;
String result;
StreamReadContext src;
for (;;) {
StreamReadStatus srs = sstream->ReadLine(&result, src);
if (srs == StatusEof)
break;
if (srs != StatusNewItem)
continue;
if (result.GetLength() > 0)
output += result + "\n";
else
break;
}
BOOST_TEST_MESSAGE("Query Result: " + output);
return output;
}
//____________________________________________________________________________//
BOOST_AUTO_TEST_SUITE(livestatus)
BOOST_AUTO_TEST_CASE(hosts)
{
BOOST_TEST_MESSAGE( "Querying Livestatus...");
std::vector<String> lines;
lines.emplace_back("GET hosts");
lines.emplace_back("Columns: host_name address check_command");
lines.emplace_back("OutputFormat: json");
lines.emplace_back("\n");
/* use our query helper */
String output = LivestatusQueryHelper(lines);
Array::Ptr query_result = JsonDecode(output);
/* the outer elements */
BOOST_CHECK(query_result->GetLength() > 1);
Array::Ptr res1 = query_result->Get(0);
Array::Ptr res2 = query_result->Get(1);
/* results are non-deterministic and not sorted by livestatus */
BOOST_CHECK(res1->Contains("test-01") || res2->Contains("test-01"));
BOOST_CHECK(res1->Contains("test-02") || res2->Contains("test-02"));
BOOST_CHECK(res1->Contains("127.0.0.1") || res2->Contains("127.0.0.1"));
BOOST_CHECK(res1->Contains("127.0.0.2") || res2->Contains("127.0.0.2"));
BOOST_TEST_MESSAGE("Done with testing livestatus hosts...");
}
BOOST_AUTO_TEST_CASE(services)
{
BOOST_TEST_MESSAGE( "Querying Livestatus...");
std::vector<String> lines;
lines.emplace_back("GET services");
lines.emplace_back("Columns: host_name service_description check_command notes");
lines.emplace_back("OutputFormat: json");
lines.emplace_back("\n");
/* use our query helper */
String output = LivestatusQueryHelper(lines);
Array::Ptr query_result = JsonDecode(output);
/* the outer elements */
BOOST_CHECK(query_result->GetLength() > 1);
Array::Ptr res1 = query_result->Get(0);
Array::Ptr res2 = query_result->Get(1);
/* results are non-deterministic and not sorted by livestatus */
BOOST_CHECK(res1->Contains("livestatus") || res2->Contains("livestatus")); //service_description
BOOST_CHECK(res1->Contains("test livestatus") || res2->Contains("test livestatus")); //notes
BOOST_TEST_MESSAGE("Done with testing livestatus services...");
}
//____________________________________________________________________________//
BOOST_AUTO_TEST_SUITE_END()
|
