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#ifndef STORAGE_LEVELDB_UTIL_TESTUTIL_H_
6#define STORAGE_LEVELDB_UTIL_TESTUTIL_H_
7
8#include "leveldb/env.h"
9#include "leveldb/slice.h"
10#include "util/random.h"
11
12namespace leveldb {
13namespace test {
14
15// Store in *dst a random string of length "len" and return a Slice that
16// references the generated data.
17extern Slice RandomString(Random* rnd, int len, std::string* dst);
18
19// Return a random key with the specified length that may contain interesting
20// characters (e.g. \x00, \xff, etc.).
21extern std::string RandomKey(Random* rnd, int len);
22
23// Store in *dst a string of length "len" that will compress to
24// "N*compressed_fraction" bytes and return a Slice that references
25// the generated data.
26extern Slice CompressibleString(Random* rnd, double compressed_fraction,
27                                size_t len, std::string* dst);
28
29// A wrapper that allows injection of errors.
30class ErrorEnv : public EnvWrapper {
31 public:
32  bool writable_file_error_;
33  int num_writable_file_errors_;
34
35  ErrorEnv() : EnvWrapper(Env::Default()),
36               writable_file_error_(false),
37               num_writable_file_errors_(0) { }
38
39  virtual Status NewWritableFile(const std::string& fname,
40                                 WritableFile** result) {
41    if (writable_file_error_) {
42      ++num_writable_file_errors_;
43      *result = NULL;
44      return Status::IOError(fname, "fake error");
45    }
46    return target()->NewWritableFile(fname, result);
47  }
48};
49
50}  // namespace test
51}  // namespace leveldb
52
53#endif  // STORAGE_LEVELDB_UTIL_TESTUTIL_H_
54