Inside Bitcoin

1// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file. See the AUTHORS file for names of contributors.
4
5#include "util/testharness.h"
6
7#include <stdlib.h>
8#include <sys/stat.h>
9#include <sys/types.h>
10
11#include <string>
12#include <vector>
13
14#include "leveldb/env.h"
15
16namespace leveldb {
17namespace test {
18
19namespace {
20struct Test {
21  const char* base;
22  const char* name;
23  void (*func)();
24};
25std::vector<Test>* tests;
26}  // namespace
27
28bool RegisterTest(const char* base, const char* name, void (*func)()) {
29  if (tests == nullptr) {
30    tests = new std::vector<Test>;
31  }
32  Test t;
33  t.base = base;
34  t.name = name;
35  t.func = func;
36  tests->push_back(t);
37  return true;
38}
39
40int RunAllTests() {
41  const char* matcher = getenv("LEVELDB_TESTS");
42
43  int num = 0;
44  if (tests != nullptr) {
45    for (size_t i = 0; i < tests->size(); i++) {
46      const Test& t = (*tests)[i];
47      if (matcher != nullptr) {
48        std::string name = t.base;
49        name.push_back('.');
50        name.append(t.name);
51        if (strstr(name.c_str(), matcher) == nullptr) {
52          continue;
53        }
54      }
55      fprintf(stderr, "==== Test %s.%s\n", t.base, t.name);
56      (*t.func)();
57      ++num;
58    }
59  }
60  fprintf(stderr, "==== PASSED %d tests\n", num);
61  return 0;
62}
63
64std::string TmpDir() {
65  std::string dir;
66  Status s = Env::Default()->GetTestDirectory(&dir);
67  ASSERT_TRUE(s.ok()) << s.ToString();
68  return dir;
69}
70
71int RandomSeed() {
72  const char* env = getenv("TEST_RANDOM_SEED");
73  int result = (env != nullptr ? atoi(env) : 301);
74  if (result <= 0) {
75    result = 301;
76  }
77  return result;
78}
79
80}  // namespace test
81}  // namespace leveldb
82