mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-08 10:23:54 +00:00
112 lines
3.9 KiB
C++
112 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2014, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include <boost/property_tree/ptree.hpp>
|
|
|
|
#include <osquery/config.h>
|
|
#include <osquery/core.h>
|
|
#include <osquery/database.h>
|
|
#include <osquery/filesystem.h>
|
|
|
|
namespace pt = boost::property_tree;
|
|
|
|
namespace osquery {
|
|
/// Init function for tests and benchmarks.
|
|
void initTesting();
|
|
|
|
/// Any SQL-dependent tests should use kTestQuery for a pre-populated example.
|
|
const std::string kTestQuery = "SELECT * FROM test_table";
|
|
|
|
extern std::string kTestDataPath;
|
|
|
|
/// Tests should limit intermediate input/output to a working directory.
|
|
/// Config data, logging results, and intermediate database/caching usage.
|
|
|
|
#ifdef DARWIN
|
|
const std::string kTestWorkingDirectory = "/private/tmp/osquery-tests";
|
|
#else
|
|
const std::string kTestWorkingDirectory = "/tmp/osquery-tests";
|
|
#endif
|
|
|
|
/// A fake directory tree should be used for filesystem iterator testing.
|
|
const std::string kFakeDirectoryName = "fstree";
|
|
extern std::string kFakeDirectory;
|
|
|
|
ScheduledQuery getOsqueryScheduledQuery();
|
|
|
|
// getTestDBExpectedResults returns the results of kTestQuery of the table that
|
|
// initially gets returned from createTestDB()
|
|
QueryData getTestDBExpectedResults();
|
|
|
|
// Starting with the dataset returned by createTestDB(), getTestDBResultStream
|
|
// returns a vector of std::pair's where pair.first is the query that would
|
|
// need to be performed on the dataset to make the results be pair.second
|
|
std::vector<std::pair<std::string, QueryData> > getTestDBResultStream();
|
|
|
|
// getSerializedRow() return an std::pair where pair->first is a string which
|
|
// should serialize to pair->second. pair->second should deserialize
|
|
// to pair->first
|
|
std::pair<pt::ptree, Row> getSerializedRow();
|
|
|
|
// getSerializedQueryData() return an std::pair where pair->first is a string
|
|
// which should serialize to pair->second. pair->second should
|
|
// deserialize to pair->first
|
|
std::pair<pt::ptree, QueryData> getSerializedQueryData();
|
|
std::pair<std::string, QueryData> getSerializedQueryDataJSON();
|
|
|
|
// getSerializedDiffResults() return an std::pair where pair->first is a string
|
|
// which should serialize to pair->second. pair->second should
|
|
// deserialize to pair->first
|
|
std::pair<pt::ptree, DiffResults> getSerializedDiffResults();
|
|
std::pair<std::string, DiffResults> getSerializedDiffResultsJSON();
|
|
|
|
// getSerializedQueryLogItem() return an std::pair where pair->first
|
|
// is a string which should serialize to pair->second. pair->second
|
|
// should deserialize to pair->first
|
|
std::pair<pt::ptree, QueryLogItem> getSerializedQueryLogItem();
|
|
std::pair<std::string, QueryLogItem> getSerializedQueryLogItemJSON();
|
|
|
|
// generate content for a PEM-encoded certificate
|
|
std::string getCACertificateContent();
|
|
|
|
// generate the content that would be found in an /etc/hosts file
|
|
std::string getEtcHostsContent();
|
|
|
|
// generate the content that would be found in an /etc/protocols file
|
|
std::string getEtcProtocolsContent();
|
|
|
|
// generate the expected data that getEtcHostsContent() should parse into
|
|
QueryData getEtcHostsExpectedResults();
|
|
|
|
// generate the expected data that getEtcProtocolsContent() should parse into
|
|
QueryData getEtcProtocolsExpectedResults();
|
|
|
|
// the three items that you need to test osquery::splitString
|
|
struct SplitStringTestData {
|
|
std::string test_string;
|
|
std::string delim;
|
|
std::vector<std::string> test_vector;
|
|
};
|
|
|
|
// generate a set of test data to test osquery::splitString
|
|
std::vector<SplitStringTestData> generateSplitStringTestData();
|
|
|
|
// generate a small directory structure for testing
|
|
void createMockFileStructure();
|
|
// remove the small directory structure used for testing
|
|
void tearDownMockFileStructure();
|
|
}
|