TEST_F(ResultsTests, test_serialize_query_data) {
auto results = getSerializedQueryData();
pt::ptree tree;
auto s = serializeQueryData(results.second, tree);
EXPECT_TRUE(s.ok());
EXPECT_EQ(s.toString(), "OK");
EXPECT_EQ(results.first, tree);
}
std::pair<pt::ptree, DiffResults> getSerializedDiffResults() {
auto qd = getSerializedQueryData();
DiffResults diff_results;
diff_results.added = qd.second;
diff_results.removed = qd.second;
pt::ptree root;
root.add_child("added", qd.first);
root.add_child("removed", qd.first);
return std::make_pair(root, diff_results);
}
std::pair<JSON, DiffResults> getSerializedDiffResults() {
auto qd = getSerializedQueryData();
DiffResults diff_results;
diff_results.added = qd.second;
diff_results.removed = qd.second;
JSON doc = JSON::newObject();
doc.add("removed", qd.first.doc());
doc.add("added", qd.first.doc());
return std::make_pair(std::move(doc), std::move(diff_results));
}
std::pair<pt::ptree, osquery::HistoricalQueryResults>
getSerializedHistoricalQueryResults() {
auto qd = getSerializedQueryData();
auto dr = getSerializedDiffResults();
HistoricalQueryResults r;
r.mostRecentResults.first = 2;
r.mostRecentResults.second = qd.second;
pt::ptree root;
pt::ptree mostRecentResults;
mostRecentResults.add_child("2", qd.first);
root.add_child("mostRecentResults", mostRecentResults);
return std::make_pair(root, r);
}
std::pair<std::string, QueryData> getSerializedQueryDataJSON() {
auto results = getSerializedQueryData();
std::ostringstream ss;
pt::write_json(ss, results.first, false);
return std::make_pair(ss.str(), results.second);
}
std::pair<std::string, QueryData> getSerializedQueryDataJSON() {
auto results = getSerializedQueryData();
std::string output;
results.first.toString(output);
return std::make_pair(output, results.second);
}