fork of https://github.com/oxigraph/rocksdb and https://github.com/facebook/rocksdb for nextgraph and oxigraph
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
3.5 KiB
99 lines
3.5 KiB
// 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>
|
|
|
|
namespace rocksdb {
|
|
|
|
/*
|
|
* If ROCKSDB_XFTEST_FORCE has a value of 1, XFUNC is forced to be defined.
|
|
* If ROCKSDB_XFTEST_FORCE has a value other than 1,
|
|
* XFUNC is forced to be undefined.
|
|
* If ROCKSDB_XFTEST_FORCE is undefined, XFUNC is defined based on NDEBUG,
|
|
* with XFUNC only being set for debug builds.
|
|
*/
|
|
#if defined(ROCKSDB_XFTEST_FORCE)
|
|
#if (ROCKSDB_XFTEST_FORCE == 1)
|
|
#define XFUNC
|
|
#endif
|
|
#elif NDEBUG
|
|
#else
|
|
#define XFUNC
|
|
#endif
|
|
|
|
#ifndef XFUNC
|
|
#define XFUNC_TEST(condition, location, lfname, fname, ...)
|
|
#else
|
|
|
|
class Options;
|
|
void GetXFTestOptions(Options* options, int skip_policy);
|
|
|
|
// This class provides the facility to run custom code to test a specific
|
|
// feature typically with all existing unit tests.
|
|
// A developer could specify cross functional test points in the codebase
|
|
// via XFUNC_TEST.
|
|
// Each xfunc test represents a position in the execution stream of a thread.
|
|
// Whenever that particular piece of code is called, the given cross-functional
|
|
// test point is executed.
|
|
// eg. on DBOpen, a particular option can be set.
|
|
// on Get, a particular option can be set, or a specific check can be invoked.
|
|
// XFUNC_TEST(TestName, location, lfname, FunctionName, Args)
|
|
// Turn on a specific cross functional test by setting the environment variable
|
|
// ROCKSDB_XFUNC_TEST
|
|
|
|
class XFuncPoint {
|
|
public:
|
|
// call once at the beginning of a test to get the test name
|
|
static void Init() {
|
|
char* s = getenv("ROCKSDB_XFUNC_TEST");
|
|
if (s == nullptr) {
|
|
xfunc_test_ = "";
|
|
enabled_ = false;
|
|
} else {
|
|
xfunc_test_ = s;
|
|
enabled_ = true;
|
|
}
|
|
initialized_ = true;
|
|
}
|
|
|
|
static bool Initialized() { return initialized_; }
|
|
|
|
static bool Check(std::string test) {
|
|
return (enabled_ &&
|
|
((test.compare("") == 0) || (test.compare(xfunc_test_) == 0)));
|
|
}
|
|
|
|
private:
|
|
static std::string xfunc_test_;
|
|
static bool initialized_;
|
|
static bool enabled_;
|
|
};
|
|
|
|
// Use XFUNC_TEST to specify cross functional test points inside the code base.
|
|
// By setting ROCKSDB_XFUNC_TEST, all XFUNC_TEST having that
|
|
// value in the condition field will be executed.
|
|
// The second argument specifies a string representing the calling location
|
|
// The third argument, lfname, is the name of the function which will be created
|
|
// and called.
|
|
// The fourth argument fname represents the function to be called
|
|
// The arguments following that are the arguments to fname
|
|
// See Options::Options in options.h for an example use case.
|
|
// XFUNC_TEST is no op in release build.
|
|
#define XFUNC_TEST(condition, location, lfname, fname, ...) \
|
|
{ \
|
|
if (!XFuncPoint::Initialized()) { \
|
|
XFuncPoint::Init(); \
|
|
} \
|
|
if (XFuncPoint::Check(condition)) { \
|
|
std::function<void()> lfname = std::bind(fname, __VA_ARGS__); \
|
|
lfname(); \
|
|
} \
|
|
}
|
|
|
|
#endif // XFUNC
|
|
|
|
enum SkipPolicy { kSkipNone = 0, kSkipNoSnapshot = 1, kSkipNoPrefix = 2 };
|
|
} // namespace rocksdb
|
|
|