mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-07 09:58:54 +00:00
71 lines
2.1 KiB
C++
71 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2015, Wesley Shields
|
|
* 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.
|
|
*
|
|
*/
|
|
|
|
#include <cstdlib>
|
|
|
|
#include <time.h>
|
|
|
|
#include <boost/filesystem.hpp>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "osquery/core/test_util.h"
|
|
#include "osquery/database/db_handle.h"
|
|
|
|
namespace fs = boost::filesystem;
|
|
|
|
namespace osquery {
|
|
DECLARE_string(database_path);
|
|
DECLARE_string(extensions_socket);
|
|
DECLARE_string(modules_autoload);
|
|
DECLARE_string(extensions_autoload);
|
|
DECLARE_bool(disable_logging);
|
|
|
|
void initTesting() {
|
|
// Seed the random number generator, some tests generate temporary files
|
|
// ports, sockets, etc using random numbers.
|
|
std::chrono::milliseconds ms =
|
|
std::chrono::duration_cast<std::chrono::milliseconds>(
|
|
std::chrono::system_clock::now().time_since_epoch());
|
|
srand(ms.count());
|
|
|
|
// Set safe default values for path-based flags.
|
|
// Specific unittests may edit flags temporarily.
|
|
fs::remove_all(kTestWorkingDirectory);
|
|
fs::create_directories(kTestWorkingDirectory);
|
|
FLAGS_database_path = kTestWorkingDirectory + "unittests.db";
|
|
FLAGS_extensions_socket = kTestWorkingDirectory + "unittests.em";
|
|
FLAGS_extensions_autoload = kTestWorkingDirectory + "unittests-ext.load";
|
|
FLAGS_modules_autoload = kTestWorkingDirectory + "unittests-mod.load";
|
|
FLAGS_disable_logging = true;
|
|
|
|
// Create a default DBHandle instance before unittests.
|
|
(void)DBHandle::getInstance();
|
|
}
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
// Allow unit test execution from anywhere in the osquery source/build tree.
|
|
while (osquery::kTestDataPath != "/") {
|
|
if (!fs::exists(osquery::kTestDataPath)) {
|
|
osquery::kTestDataPath =
|
|
osquery::kTestDataPath.substr(3, osquery::kTestDataPath.size());
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
|
|
osquery::initTesting();
|
|
testing::InitGoogleTest(&argc, argv);
|
|
// Optionally enable Goggle Logging
|
|
// google::InitGoogleLogging(argv[0]);
|
|
return RUN_ALL_TESTS();
|
|
}
|